feat: use struct instead const

cloudwego · May 10, 2024 · 515fb17 · 515fb17
1 parent 8d7b453
commit 515fb17
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/network/netpoll/transport.go b/pkg/network/netpoll/transport.go
@@ -36,11 +36,11 @@ func init() {
 	netpoll.SetLoggerOutput(io.Discard)
 }
 
-const ctxCancelKey = "ctxCancelKey"
+type ctxCancelKey struct{}
 
 func cancelContext(ctx context.Context) context.Context {
 	ctx, cancel := context.WithCancel(ctx)
-	ctx = context.WithValue(ctx, ctxCancelKey, cancel)
+	ctx = context.WithValue(ctx, ctxCancelKey{}, cancel)
 	return ctx
 }
 
@@ -117,7 +117,7 @@ func (t *transporter) ListenAndServe(onReq network.OnData) (err error) {
 
 	if t.senseClientDisconnection {
 		opts = append(opts, netpoll.WithOnDisconnect(func(ctx context.Context, connection netpoll.Connection) {
-			cancelFunc, ok := ctx.Value(ctxCancelKey).(context.CancelFunc)
+			cancelFunc, ok := ctx.Value(ctxCancelKey{}).(context.CancelFunc)
 			if cancelFunc != nil && ok {
 				cancelFunc()
 			}