Go语言系统编程与命令行工具

Go语言系统编程与命令行工具一、命令行参数解析Go语言提供了多个标准库来处理命令行参数包括flag包和os包。使用flag包package main import ( flag fmt ) func main() { // 定义命令行参数 name : flag.String(name, Guest, User name) age : flag.Int(age, 18, User age) verbose : flag.Bool(verbose, false, Enable verbose mode) // 解析参数 flag.Parse() fmt.Printf(Name: %s\n, *name) fmt.Printf(Age: %d\n, *age) fmt.Printf(Verbose: %v\n, *verbose) // 解析剩余的非flag参数 fmt.Printf(Args: %v\n, flag.Args()) }使用pflag包go get github.com/spf13/pflagpackage main import ( fmt github.com/spf13/pflag ) func main() { name : pflag.StringP(name, n, Guest, User name) age : pflag.IntP(age, a, 18, User age) verbose : pflag.BoolP(verbose, v, false, Enable verbose mode) pflag.Parse() fmt.Printf(Name: %s\n, *name) fmt.Printf(Age: %d\n, *age) fmt.Printf(Verbose: %v\n, *verbose) }二、文件操作文件读写package main import ( fmt io/ioutil os ) func main() { // 写入文件 err : ioutil.WriteFile(example.txt, []byte(Hello, World!), 0644) if err ! nil { fmt.Printf(Write error: %v\n, err) return } // 读取文件 content, err : ioutil.ReadFile(example.txt) if err ! nil { fmt.Printf(Read error: %v\n, err) return } fmt.Printf(File content: %s\n, content) }文件操作进阶package main import ( bufio fmt os ) func main() { // 打开文件 file, err : os.OpenFile(example.txt, os.O_RDWR|os.O_CREATE, 0644) if err ! nil { fmt.Printf(Open error: %v\n, err) return } defer file.Close() // 写入数据 writer : bufio.NewWriter(file) _, err writer.WriteString(Hello, Go!\n) if err ! nil { fmt.Printf(Write error: %v\n, err) return } writer.Flush() // 读取数据 file.Seek(0, 0) scanner : bufio.NewScanner(file) for scanner.Scan() { fmt.Println(scanner.Text()) } if err : scanner.Err(); err ! nil { fmt.Printf(Scan error: %v\n, err) } }三、进程管理启动子进程package main import ( fmt os/exec ) func main() { // 执行命令 cmd : exec.Command(echo, Hello, World!) output, err : cmd.CombinedOutput() if err ! nil { fmt.Printf(Command error: %v\n, err) return } fmt.Printf(Output: %s\n, output) }带管道的进程package main import ( fmt os/exec io ) func main() { // 创建两个命令 cmd1 : exec.Command(echo, Hello, Go!) cmd2 : exec.Command(tr, a-z, A-Z) // 连接管道 pipe, err : cmd1.StdoutPipe() if err ! nil { fmt.Printf(Pipe error: %v\n, err) return } cmd2.Stdin pipe // 启动命令 err cmd1.Start() if err ! nil { fmt.Printf(Start cmd1 error: %v\n, err) return } output, err : cmd2.CombinedOutput() if err ! nil { fmt.Printf(Cmd2 error: %v\n, err) return } cmd1.Wait() fmt.Printf(Output: %s\n, output) }四、信号处理package main import ( fmt os os/signal syscall time ) func main() { // 创建信号通道 sigChan : make(chan os.Signal, 1) signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM) // 启动业务逻辑 go func() { for { fmt.Println(Working...) time.Sleep(1 * time.Second) } }() // 等待信号 sig : -sigChan fmt.Printf(\nReceived signal: %v\n, sig) // 清理资源 fmt.Println(Cleaning up...) fmt.Println(Exiting...) }五、环境变量package main import ( fmt os ) func main() { // 获取环境变量 home : os.Getenv(HOME) fmt.Printf(HOME: %s\n, home) // 设置环境变量 err : os.Setenv(MY_APP_ENV, production) if err ! nil { fmt.Printf(Setenv error: %v\n, err) return } // 获取所有环境变量 env : os.Environ() for _, e : range env { fmt.Println(e) } // 检查环境变量是否存在 _, exists : os.LookupEnv(MY_APP_ENV) fmt.Printf(MY_APP_ENV exists: %v\n, exists) }六、系统信息package main import ( fmt os runtime ) func main() { // 获取操作系统信息 fmt.Printf(OS: %s\n, runtime.GOOS) fmt.Printf(Arch: %s\n, runtime.GOARCH) fmt.Printf(CPU cores: %d\n, runtime.NumCPU()) // 获取进程信息 fmt.Printf(Process ID: %d\n, os.Getpid()) fmt.Printf(Parent PID: %d\n, os.Getppid()) // 获取当前工作目录 cwd, err : os.Getwd() if err ! nil { fmt.Printf(Getwd error: %v\n, err) return } fmt.Printf(Current directory: %s\n, cwd) }七、命令行工具框架使用Cobrago get github.com/spf13/cobralatestpackage main import ( fmt os github.com/spf13/cobra ) func main() { var rootCmd cobra.Command{ Use: myapp, Short: A simple command line tool, Long: MyApp is a demonstration of Cobra commands, Run: func(cmd *cobra.Command, args []string) { fmt.Println(Hello from MyApp!) }, } var echoCmd cobra.Command{ Use: echo [message], Short: Echo the message, Args: cobra.MinimumNArgs(1), Run: func(cmd *cobra.Command, args []string) { upper, _ : cmd.Flags().GetBool(upper) message : args[0] if upper { message strings.ToUpper(message) } fmt.Println(message) }, } echoCmd.Flags().BoolP(upper, u, false, Convert to uppercase) rootCmd.AddCommand(echoCmd) if err : rootCmd.Execute(); err ! nil { fmt.Fprintln(os.Stderr, err) os.Exit(1) } }八、日志工具使用logrusgo get github.com/sirupsen/logruspackage main import ( github.com/sirupsen/logrus ) func main() { log : logrus.New() // 设置日志级别 log.SetLevel(logrus.DebugLevel) // 设置输出格式 log.SetFormatter(logrus.JSONFormatter{}) // 不同级别的日志 log.Debug(Debug message) log.Info(Info message) log.Warn(Warn message) log.Error(Error message) // 带字段的日志 log.WithFields(logrus.Fields{ user: john, id: 42, }).Info(User logged in) }九、配置文件处理使用toml配置go get github.com/BurntSushi/tomlpackage main import ( fmt os github.com/BurntSushi/toml ) type Config struct { Database struct { Host string toml:host Port int toml:port Database string toml:database } toml:database Server struct { Port int toml:port } toml:server } func main() { config : Config{} file, err : os.Open(config.toml) if err ! nil { fmt.Printf(Open config error: %v\n, err) return } defer file.Close() _, err toml.DecodeReader(file, config) if err ! nil { fmt.Printf(Decode error: %v\n, err) return } fmt.Printf(DB Host: %s\n, config.Database.Host) fmt.Printf(DB Port: %d\n, config.Database.Port) }配置文件示例[database] host localhost port 3306 database example_db [server] port 8080十、总结Go语言提供了丰富的系统编程能力通过标准库可以轻松实现命令行参数解析使用flag或pflag包文件操作读写文件、目录操作进程管理启动子进程、管道通信信号处理捕获系统信号环境变量获取和设置环境变量系统信息获取OS、CPU、进程信息命令行框架使用Cobra构建复杂CLI工具日志记录使用logrus等日志库配置文件支持多种配置格式掌握这些系统编程能力可以帮助你构建强大的命令行工具和系统应用。