分享一个项目：go `file_line`，在编译期得到源码行号，减少运行期runtime消耗

作者:张富春(ahfuzhang)，转载时请注明作者和引用链接，谢谢！

• cnblogs博客
• zhihu
• Github
• 公众号:一本正经的瞎扯
  

---

file_line

https://github.com/ahfuzhang/file_line

Like __FILE__/__LINE__ of C: use go generate to get source code line number at compile time.
像 C 语言里面的 __FILE__/__LINE__ 宏一样：在编译期，通过 go generate来得到源码行号。

我通常使用下面这个函数来获取源码的行号：

func SourceCodeLoc(callDepth int) string { 	_, file, line, ok := runtime.Caller(callDepth) 	if !ok { 		return "" 	} 	file = strings.ReplaceAll(file, "\", "/") 	arr := strings.Split(file, "/") 	if len(arr) > 3 { 		file = strings.Join(arr[len(arr)-3:], "/") 	} 	return fmt.Sprintf("%s:%d", file, line) }  func example(){      Mylogger.Infof("[%s]something happens here", SourceCodeLoc(1)) } 

这里的 runtime.Caller() 实在程序运行期间去计算程序对应的源码行的，必然会带来性能损耗。
这种需求完全可以在编译期间实现，最终我发现使用 go ast 库能够简单的达成这一功能。

How to use

1. 安装：

go install github.com/ahfuzhang/file_line@latest 

2. 编写代码，在需要使用行号的地方使用这样的place holder:

func myCode(){     Mylogger.Infof(“%s: something happens here”, “[file.go:123]") } 

3. 在程序的入口出加上 go generate指令：

//go:generate file_line -src=./  func main() { 	fmt.Println("use a place holder:", "[file.go:123]") } 

4. 在编译前执行 go generate

• 所有的处于函数调用参数位置的 place holder 会被替换为正确的文件名和行号
• 也可以直接在命令行执行 file_line -src=./

5. 执行 go build。

Have Fun. 😃