本文将详细介绍如何在 Go 语言的 main 函数中设置程序退出码，类似于 C 语言中的 exit() 函数。通过 os.Exit() 函数可以方便地设置退出码，而 log.Fatal() 系列函数则可以在输出错误信息的同时设置非零退出码，从而实现更友好的错误处理。
在 Go 语言中，main 函数的退出方式与 C 语言有所不同。Go 语言没有像 C 语言那样直接使用 return 语句来返回退出码。取而代之的是使用 os.Exit() 函数。
os.Exit() 函数接受一个整数作为参数，该整数将作为程序的退出码。例如，要以错误码 1 退出程序，可以这样做：
package main
import (
"fmt"
"os"
)
func main() {
if len(os.Args) != 2 {
fmt.Println("Usage: myprogram ")
os.Exit(1) // Exit with error code 1
}
// Process the argument here...
fmt.Println("Argument:", os.Args[1])
} 在这个例子中，如果命令行参数的数量不是 2 (程序名 + 一个参数)，程序将打印用法信息并以退出码 1 退出。
使用 log.Fatal() 系列函数
除了 os.Exit()，Go 语言还提供了 log.Fatal()、log.Fatalf() 和 log.Fatalln() 函数，它们位于 log 包中。这些函数不仅可以输出错误信息，还会自动调用 os.Exit(1)，从而以非零退出码退出程序。
package main
import (
"log"
"os"
)
func main() {
if len(os.Args) != 2 {
log.Fatal("Usage: myprogram ") // Prints error and exits with code 1
}
// Process the argument here...
log.Println("Argument:", os.Args[1])
} 在这个例子中，如果命令行参数的数量不正确，log.Fatal() 将打印错误信息到标准错误输出，并以退出码 1 退出程序。
log.Fatalf() 允许使用格式化字符串：
package main
import (
"log"
"os"
)
func main() {
if len(os.Args) != 2 {
log.Fatalf("Usage: %s ", os.Args[0]) // Prints formatted error and exits with code 1
}
// Process the argument here...
log.Println("Argument:", os.Args[1])
} log.Fatalln() 类似于 log.Fatal()，但会在输出的信息后添加换行符：
package main
import (
"log"
"os"
)
func main() {
if len(os.Args) != 2 {
log.Fatalln("Usage: myprogram ") // Prints error with newline and exits with code 1
}
// Process the argument here...
log.Println("Argument:", os.Args[1])
} 注意事项
- os.Exit() 会立即终止程序，不会执行任何延迟函数 (使用 defer 声明的函数)。
- log.Fatal() 系列函数也会立即终止程序，同样不会执行延迟函数。
- 通常，0 表示程序成功执行，非零值表示程序执行过程中遇到了错误。
- 选择合适的退出码可以帮助其他程序或脚本判断程序的执行状态。
总结
在 Go 语言中，可以使用 os.Exit() 函数来设置程序的退出码。log.Fatal() 系列函数则提供了更方便的方式，可以在输出错误信息的同时设置非零退出码。 根据实际需求选择合适的函数，可以编写出更健壮和易于维护的 Go 程序。
