modify prompt

Author: sylvia7788

go.sum

@@ -11,7 +11,7 @@ func NewContextCheck() *goanalysis.Linter {
 	analyzer := contextcheck.NewAnalyzer()
 	return goanalysis.NewLinter(
 		"contextcheck",
-		"check for using context.Background() and context.TODO() directly",
+		"check the function whether use a non-inherited context",
 		[]*analysis.Analyzer{analyzer},
 		nil,
 	).WithLoadMode(goanalysis.LoadModeTypesInfo)

pkg/golinters/contextcheck.go

@@ -21,15 +21,21 @@ func contextcheckCase2(ctx context.Context) {
 		funcWithCtx(ctx)
 	}(ctx)
 
-	funcWithCtx(context.Background()) // ERROR `The context param may be context.TODO\(\) or context.Background\(\), please replace it with another way, such as context.WithValue\(ctx, key, val\)`
+	funcWithCtx(context.Background()) // ERROR "Non-inherited new context, use function like `context.WithXXX` instead"
 }
 
 func contextcheckCase3(ctx context.Context) {
 	func() {
 		funcWithCtx(ctx)
 	}()
 
-	ctx = context.Background() // ERROR `Invalid call to get new context, please replace it with another way, such as context.WithValue\(ctx, key, val\)`
+	ctx = context.Background() // ERROR "Non-inherited new context, use function like `context.WithXXX` instead"
+	funcWithCtx(ctx)
+}
+
+func contextcheckCase4(ctx context.Context) {
+	ctx, cancel := getNewCtx(ctx)
+	defer cancel()
 	funcWithCtx(ctx)
 }
 
@@ -38,3 +44,7 @@ func funcWithCtx(ctx context.Context) {}
 func funcWithoutCtx() {
 	funcWithCtx(context.TODO())
 }
+
+func getNewCtx(ctx context.Context) (newCtx context.Context, cancel context.CancelFunc) {
+	return context.WithCancel(ctx)
+}