Merge pull request #6 from Buzzvil/add_recover_flag

Add recover flag

Author: dc7303

README.md

@@ -12,7 +12,12 @@ go install github.com/Buzzvil/recovergoroutine
 
 ## Usage
 ```bash
-recovergoroutine ./...
+recovergoroutine -recover="" ./...
+
+# -recover string
+#         Custom recover method name. Currently, it is difficult to determine
+#         if a CustomRecover function declared in another package is valid,
+#         so this option can be used to resolve it.
 ```
 
 Check out the test cases for validation [examples](./test/src/faildata/failcode.go).

recovergoroutine/recovergoroutine.go

@@ -1,6 +1,7 @@
 package recovergoroutine
 
 import (
+	"flag"
 	"fmt"
 	"go/ast"
 	"go/parser"
@@ -10,13 +11,24 @@ import (
 	"golang.org/x/tools/go/analysis"
 )
 
+var customRecover string
+
 func NewAnalyzer() *analysis.Analyzer {
 	analyzer := &analysis.Analyzer{
 		Name: "recovergoroutine",
 		Doc:  "finds goroutine code without recover",
 		Run:  run,
 	}
 
+	analyzer.Flags.Init("recovergoroutine", flag.ExitOnError)
+	analyzer.Flags.StringVar(
+		&customRecover,
+		"recover",
+		"",
+		"It is difficult to determine if a CustomRecover function declared in another package is valid,"+
+			" so this option can be used to resolve it.",
+	)
+
 	return analyzer
 }
 
@@ -122,7 +134,7 @@ func hasRecover(expr ast.Node, pass *analysis.Pass) (bool, error) {
 				return false
 			}
 
-			if ok {
+			if ok || n.Sel.Name == customRecover {
 				result = true
 				return false
 			}