Go基础编程实践（八）—— 系统编程

捕捉信号

// 运行此程序，控制台将打印"Waiting for signal"
// 按Ctrl + C 发送信号以关闭程序，将发生中断
// 随后控制台依次打印"Signal .."、"Exiting..."
package main

import (
    "os"
    "os/signal"
    "syscall"
    "fmt"
)

func main() {
    // 缓冲通道捕捉信号
    signals := make(chan os.Signal, 1)
    done := make(chan bool)
    // Notify将输入信号转发到channel
    signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
    go func() {
        sig := <- signals
        fmt.Println(sig)
        fmt.Println("Signal captured and processed...")
        done <- true
    }()
    fmt.Println("Waiting for signal")
    <- done
    fmt.Println("Exiting the application...")
}

运行子进程

// 在Go程序中运行其他进程
package main

import (
    "os/exec"
    "fmt"
)

func main() {
    // Command接收两个参数：命令、命令参数
   // lsCommand := exec.Command("ls", "-a")
    lsCommand := exec.Command("ls")
    // Output执行命令并返回标准输出的切片。
    output, _ := lsCommand.Output()
    //Run函数阻塞进程直到lsCommand执行完毕，与之类似到Start函数不阻塞
    lsCommand.Run()
    // 获取进程ID
    fmt.Println(lsCommand.Process.Pid)
    // 获取ls命令结果
    fmt.Println(string(output))
}

处理命令行参数

package main

import (
    "os"
    "fmt"
)

func main() {
    // os.Args的第一个元素是可执行文件路径，所以获取参数从下标1开始
    realArgs := os.Args[1:]
    if len(realArgs) == 0 {
        fmt.Println("Please pass an argument.")
        return
    }
    if realArgs[0] == "a" {
        writeHelloWorld()
    } else if realArgs[0] == "b" {
        writeHelloMars()
    } else {
        fmt.Println("Please pass a valid argument.")
    }
}

func writeHelloWorld() {
    fmt.Println("Hello, World!")
}

func writeHelloMars() {
    fmt.Println("Hello, Mars!")
}
// 执行"go run main.go a"将输出"Hello, World!"
// 执行"go run main.go b"将输出"Hello, Mars!"