ruleguard: assign the current func decl (#328)

quasilyte · Dec 28, 2021 · 9ecd049 · 9ecd049
1 parent fd27736
commit 9ecd049
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ruleguard/ast_walker.go b/ruleguard/ast_walker.go
@@ -303,10 +303,11 @@ func (w *astWalker) walk(n ast.Node) {
 		}
 
 	case *ast.FuncDecl:
+		prevFunc := w.filterParams.currentFunc
+		w.filterParams.currentFunc = n
 		if n.Doc != nil {
 			w.walk(n.Doc)
 		}
-		prevFunc := w.filterParams.currentFunc
 		if n.Recv != nil {
 			w.walk(n.Recv)
 		}