gorun, golang, shell

用 golang 语言编写脚本

install gorun

1
    go get github.com/erning/gorun

能用 “./” 执行的 golang代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
/// 2>/dev/null ; gorun "$0" "$@" ; exit $?

package main
import (
    "os/exec"
    "fmt"
    "os"
)
func main() {
    println("Hello world!")
    var whoami []byte
    var err error
    var cmd *exec.Cmd
    cmd = exec.Command("whoami")
    if whoami, err = cmd.Output(); err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
    // 默认输出有一个换行
    fmt.Println(string(whoami))
}

1
./hello.go

golang shell, 在 golang 中调用 shell

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
func shellExec(command string) string {
    log.Printf("exec: %s\n", command)
    cmd := exec.Command("/bin/sh", "-c", command)
    var out bytes.Buffer

    cmd.Stdout = &out
    err := cmd.Run()
    if err != nil {
        log.Printf("failed to exec: %s, err: %v", command, err)
        return ""
    }
    ourStr := out.String()
    log.Printf("exec response: \n%s", ourStr)
    return ourStr
}
func shellExecFmt(format string, params ...interface{}) string {
    return shellExec(fmt.Sprintf(format, params...))
}

go 调用 shell

https://blog.csdn.net/qq_36874881/article/details/78234005

阻塞方式(需要执行结果)

主要用于执行shell命令,并且返回shell的标准输出

适用于执行普通非阻塞shell命令,且需要shell标准输出的

//阻塞式的执行外部shell命令的函数,等待执行完毕并返回标准输出

func exec_shell(s string) (string, error){

//函数返回一个*Cmd,用于使用给出的参数执行name指定的程序

cmd := exec.Command("/bin/bash”, “-c”, s)

//读取io.Writer类型的cmd.Stdout,再通过bytes.Buffer(缓冲byte类型的缓冲器)将byte类型转化为string类型(out.String():这是bytes类型提供的接口)
var out bytes.Buffer
cmd.Stdout = &out

//Run执行c包含的命令,并阻塞直到完成。  这里stdout被取出,cmd.Wait()无法正确获取stdin,stdout,stderr,则阻塞在那了
err := cmd.Run()
checkErr(err)


return out.String(), err

}

需要对shell标准输出的逐行实时进行处理的

func execCommand(commandName string, params []string) bool {

//函数返回一个*Cmd,用于使用给出的参数执行name指定的程序

cmd := exec.Command(commandName, params…)

//显示运行的命令
fmt.Println(cmd.Args)
//StdoutPipe方法返回一个在命令Start后与命令标准输出关联的管道。Wait方法获知命令结束后会关闭这个管道,一般不需要显式的关闭该管道。
stdout, err := cmd.StdoutPipe()

if err != nil {
    fmt.Println(err)
    return false
}

cmd.Start()
//创建一个流来读取管道内内容,这里逻辑是通过一行一行的读取的
reader := bufio.NewReader(stdout)

//实时循环读取输出流中的一行内容
for {
    line, err2 := reader.ReadString('\n')
    if err2 != nil || io.EOF == err2 {
        break
    }
    fmt.Println(line)
}

//阻塞直到该命令执行完成,该命令必须是被Start方法开始执行的
cmd.Wait()
return true

}

非阻塞方式(不需要执行结果)

通过shell调用自己的程序,并且程序是死循环,此时无法获取返回结果(否则程序会一直阻塞直至调用的程序结束)

适用于调用自己写的程序(服务器死循环,且不需要返回结果的)

//不需要执行命令的结果与成功与否,执行命令马上就返回

func exec_shell_no_result(command string) {

//处理启动参数,通过空格分离如: setsid /home/luojing/gotest/src/test_main/iwatch/test/while_little &

command_name_and_args := strings.FieldsFunc(command, splite_command)

//开始执行c包含的命令,但并不会等待该命令完成即返回

cmd.Start()

if err != nil {

fmt.Printf("%v: exec command:%v error:%v\n”, get_time(), command, err)

}

fmt.Printf(“Waiting for command:%v to finish…\n”, command)

//阻塞等待fork出的子进程执行的结果,和cmd.Start()配合使用[不等待回收资源,会导致fork出执行shell命令的子进程变为僵尸进程]

err = cmd.Wait()

if err != nil {

fmt.Printf("%v: Command finished with error: %v\n”, get_time(), err)

}

return

}

/错误处理函数

func checkErr(err error) {

if err != nil {

fmt.Println(err)

panic(err)

}

官网的标准中文库

引用部分文档

func (*Cmd) Run

func (c *Cmd) Run() error

Run执行c包含的命令,并阻塞直到完成。

如果命令成功执行,stdin、stdout、stderr的转交没有问题,并且返回状态码为0,方法的返回值为nil；如果命令没有执行或者执行失败,会返回_ExitError类型的错误；否则返回的error可能是表示I/O问题。

func (_Cmd) Start

func (c *Cmd) Start() error

Start开始执行c包含的命令,但并不会等待该命令完成即返回。Wait方法会返回命令的返回状态码并在命令返回后释放相关的资源。

func (*Cmd) Wait

func (c *Cmd) Wait() error

Wait会阻塞直到该命令执行完成,该命令必须是被Start方法开始执行的。

如果命令成功执行,stdin、stdout、stderr的转交没有问题,并且返回状态码为0,方法的返回值为nil；如果命令没有执行或者执行失败,会返回_ExitError类型的错误；否则返回的error可能是表示I/O问题。Wait方法会在命令返回后释放相关的资源。

func (_Cmd) Output

func (c _Cmd) Output() ([]byte, error)

执行命令并返回标准输出的切片。

func (_Cmd) StderrPipe

func (c *Cmd) StderrPipe() (io.ReadCloser, error)

StderrPipe方法返回一个在命令Start后与命令标准错误输出关联的管道。Wait方法获知命令结束后会关闭这个管道,一般不需要显式的关闭该管道。但是在从管道读取完全部数据之前调用Wait是错误的；同样使用StderrPipe方法时调用Run函数也是错误的。

https://www.infoq.cn/article/mbzyz8sedtbz5*4mhizo https://github.com/erning/gorun

Contents