如何使用Go Analyze包判断函数参数的作用域层级？

先看示例代码example.go：

package main

import "log"

var myglobalvar = "hello from global"

func myLog(format string, args ...interface{}) {
    const prefix = "[my] "
    mylocalvar := "hello from local"
    log.Printf(prefix+myglobalvar+mylocalvar+format, args...)
}

func main() {
    mystring := "HELLO FROM MAIN"
    myLog(mystring)
    myLog(myglobalvar)
}

以下是修正后的完整分析器实现analyze.go，可以准确判断参数的作用域：

package main

import (
	"go/ast"
	"go/analysis"
	"go/analysis/passes/inspect"
	"go/ast/inspector"
)

var Analyzer = &analysis.Analyzer{
	Name: "scopelint",
	Doc:  "检测函数调用参数是全局变量还是局部变量",
	Run:  run,
	Requires: []*analysis.Analyzer{
		inspect.Analyzer,
	},
}

func run(pass *analysis.Pass) (interface{}, error) {
	inspect := pass.ResultOf[inspect.Analyzer].(*inspector.Inspector)
	nodeFilter := []ast.Node{(*ast.CallExpr)(nil)}

	inspect.Preorder(nodeFilter, func(node ast.Node) {
		callExpr := node.(*ast.CallExpr)

		// 过滤出目标函数myLog的调用
		funcIdent, ok := callExpr.Fun.(*ast.Ident)
		if !ok || funcIdent.Name != "myLog" {
			return
		}

		// 遍历每个参数
		for _, arg := range callExpr.Args {
			// 仅处理直接变量引用的参数
			argIdent, ok := arg.(*ast.Ident)
			if !ok {
				pass.Reportf(arg.Pos(), "参数不是直接变量引用")
				continue
			}

			obj := argIdent.Obj
			if obj == nil {
				pass.Reportf(arg.Pos(), "无法解析变量对象")
				continue
			}

			// 只处理变量类型的标识符
			if obj.Kind != ast.Var {
				pass.Reportf(arg.Pos(), "参数不是变量（可能是常量/函数等）")
				continue
			}

			// 根据声明类型区分作用域
			switch decl := obj.Decl.(type) {
			case *ast.ValueSpec:
				// 判断是否为包级全局变量：声明位置在函数体外
				isGlobal := false
				// 遍历父节点，检查是否在函数体之外
				parent := pass.TypesInfo.ObjectOf(argIdent).Parent()
				if parent == nil || parent.Pkg() == pass.Pkg {
					// 包级声明的变量
					isGlobal = true
				}
				if isGlobal {
					pass.Reportf(arg.Pos(), "参数是全局变量: %s", obj.Name)
				} else {
					pass.Reportf(arg.Pos(), "参数是函数内局部变量（var声明）: %s", obj.Name)
				}
			case *ast.AssignStmt:
				// 短变量声明（x := ...）或赋值语句，属于局部作用域
				pass.Reportf(arg.Pos(), "参数是局部变量（短声明/赋值）: %s", obj.Name)
			case *ast.Field:
				// 函数参数，属于局部作用域
				pass.Reportf(arg.Pos(), "参数是函数参数: %s", obj.Name)
			default:
				pass.Reportf(arg.Pos(), "未知的变量声明类型: %T", decl)
			}
		}
	})

	return nil, nil
}

核心实现逻辑 #

1. 定位目标函数调用：通过AST遍历筛选出myLog函数的调用节点*ast.CallExpr。
2. 解析参数标识符：仅处理直接的变量引用（*ast.Ident类型），跳过表达式类参数。
3. 获取变量元信息：通过argIdent.Obj获取变量的声明信息，需处理Obj为nil的异常情况。
4. 区分作用域：
   • *ast.ValueSpec：对应包级var声明（全局）或函数内var声明（局部），通过变量的父级包信息判断是否为全局。
   • *ast.AssignStmt：对应短变量声明或赋值操作，均属于局部作用域。
   • *ast.Field：对应函数参数，属于局部作用域。

内容的提问来源于stack exchange，提问作者Libraraimos