最近在使用go写一些windows端的exe,需要定义退出码,竟然能发现go的main函数不能定义返回类型,查了一些相关资料,总结出如下方法:
func main(){
code := 0
defer func() {
fmt.Printf("exit code:%d\n", code)
os.Exit(code)
}()
defer func() {
if r := recover(); r != nil {
fmt.Println("Recovered from panic:", r)
code = 2
}
}()
// 下面两段代码二选一
// test panic
{
a := 100
b := 0
fmt.Printf("a/b=%d", a/b)
}
// test non 0 exit
{
if true {
code = 3
return
}
}
code = 0
}
解释如下:
1,使用两个延时函数来处理最终的返回值,第一个延时函数
defer func() {
if r := recover(); r != nil {
fmt.Println("Recovered from panic:", r)
code = 2
}
}()
用来处理panic的情况,这样做的目的是防止程序异常退出,没办法正确返回我期望的退出码
2,如果在出错的分支直接调用os.Exit()会导致资源释放不充分,所以使用这个延时函数
defer func() {
fmt.Printf(“exit code:%d\n”, code)
os.Exit(code)
}()
在main函数退出之前调用,确保我的进程能按照预期返回正确的退出码