2. 程式人生 > >gocommand:一個跨平臺的golang命令列執行package

gocommand:一個跨平臺的golang命令列執行package

阿新 • • 發佈：2019-01-09

    最近在做一個專案的時候,需要使用golang來呼叫作業系統中的命令列,來執行shell命令或者直接呼叫第三方程式,這其中自然就用到了golang自帶的exec.Command.

    但是如果直接使用原生exec.Command會造成大量的重複程式碼,網上搜了一圈又沒有找到對exec.Command相應的封裝包,索性自己封裝了一個,取名為gocommand.目前支援Linux和Windows,歡迎各位大神在github上提交程式碼補充其他平臺的實現.

    下面介紹一下gocommand庫的實現思路:

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

package gocommand

// 命令列介面

type Commander interface {

// 執行命令列並返回結果

// args: 命令列引數

// return: 程序的pid, 命令列結果, 錯誤訊息

Exec(args ...string) (int, string, error)

// 非同步執行命令列並通過channel返回結果

// stdout: chan結果

// args: 命令列引數

// return: 程序的pid

// exception: 協程內的命令列發生錯誤時,會panic異常

ExecAsync(stdout chan string, args ...string) int

// 執行命令列(忽略返回值)

// args: 命令列引數

// return: 錯誤訊息

ExecNoWait(args ...string) error

}

gocommand目前的命令列執行函式都是源於Commander介面,目前該介面定義了3個函式,分別是:執行命令列病返回結果;非同步執行命令列並得到結果;執行命令列並忽略結果.

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

package gocommand

import (

"runtime"

)

// Command的初始化函式

func NewCommand() Commander {

var cmd Commander

switch runtime.GOOS {

case "linux":

cmd = NewLinuxCommand()

case "windows":

cmd = NewWindowsCommand()

default:

cmd = NewLinuxCommand()

}

return cmd

}

    建立一個Command的實現,並根據當前的作業系統,返回對應的實現函式,目前只實現了Linux和Windows,(Mac留給各位大神(土豪)了),其中LinuxCommand的程式碼實現如下:

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

package gocommand

import (

"io/ioutil"

"os"

"os/exec"

"syscall"

)

// LinuxCommand結構體

type LinuxCommand struct {

}

// LinuxCommand的初始化函式

func NewLinuxCommand() *LinuxCommand {

return &LinuxCommand{}

}

// 執行命令列並返回結果

// args: 命令列引數

// return: 程序的pid, 命令列結果, 錯誤訊息

func (lc *LinuxCommand) Exec(args ...string) (int, string, error) {

args = append([]string{"-c"}, args...)

cmd := exec.Command(os.Getenv("SHELL"), args...)

cmd.SysProcAttr = &syscall.SysProcAttr{}

outpip, err := cmd.StdoutPipe()

if err != nil {

return 0, "", err

}

err = cmd.Start()

if err != nil {

return 0, "", err

}

out, err := ioutil.ReadAll(outpip)

if err != nil {

return 0, "", err

}

return cmd.Process.Pid, string(out), nil

}

// 非同步執行命令列並通過channel返回結果

// stdout: chan結果

// args: 命令列引數

// return: 程序的pid

// exception: 協程內的命令列發生錯誤時,會panic異常

func (lc *LinuxCommand) ExecAsync(stdout chan string, args ...string) int {

var pidChan = make(chan int, 1)

go func() {

args = append([]string{"-c"}, args...)

cmd := exec.Command(os.Getenv("SHELL"), args...)

cmd.SysProcAttr = &syscall.SysProcAttr{}

outpip, err := cmd.StdoutPipe()

if err != nil {

panic(err)

}

err = cmd.Start()

if err != nil {

panic(err)

}

pidChan <- cmd.Process.Pid

out, err := ioutil.ReadAll(outpip)

if err != nil {

panic(err)

}

stdout <- string(out)

}()

return <-pidChan

}

// 執行命令列(忽略返回值)

// args: 命令列引數

// return: 錯誤訊息

func (lc *LinuxCommand) ExecNoWait(args ...string) error {

args = append([]string{"-c"}, args...)

cmd := exec.Command(os.Getenv("SHELL"), args...)

cmd.Stdout = os.Stdout

cmd.Stderr = os.Stderr

cmd.SysProcAttr = &syscall.SysProcAttr{}

err := cmd.Run()

return err

}

    Exec函式會在執行命令列後阻塞,直到得到命令的執行結果;ExecAsync函式在內部使用了協程來執行命令列,並通過引數中的chan變數把結果傳遞出去;ExecNoWait會無阻賽地執行命令列.Windows平臺上的實現類似,只是Shell命令換成了cmd.

    使用示例如下:

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

package main

import (

"log"

"github.com/lizongshen/gocommand"

)

func main() {

_, out, err := gocommand.NewCommand().Exec("ls /")

if err != nil {

log.Panic(err)

}

log.Println(out)

}

    程式碼的單元測試情況:

1

2

3

4

5

[[email protected] gocommand]$ go test

bin   dev  home  lib64  mnt  proc  run   srv  tmp  var

boot  etc  lib   media  opt  root  sbin  sys  usr

PASS

ok      gocommand   0.007s

相關推薦

gocommand:一個跨平臺的golang命令列執行package

    最近在做一個專案的時候,需要使用golang來呼叫作業系統中的命令列,來執行shell命令或者直接呼叫第三方程式,這其中自然就用到了golang自帶的exec.Command.     但是如果直接使用原生exec.Command會造成大量的重複程式碼,網上搜了一圈

命令列執行robotframework 遇到 AutoItLibrary . Run執行exe上傳指令碼失敗記錄

我在公司做自動化指令碼選擇的是robotframework 最近公司領導要求把做的頁面自動化成果展示在jenkins上，那就需要把寫好的指令碼放到Windows環境上，從jenkins上啟動指令碼使用的是最簡單粗暴的方式：命令列執行pybot D:\robotframe\web\ 當時使用在

java 命令列執行時報錯:找不到或無法載入主類

原始碼如下: 路徑目錄(...\src\object) 1 //: ch1.10/Command.java 2 /**@author feilong 3 * @version 1.0 4 */ 5 package object; 6 import java.util.*; 7 8

Java帶包的cmd命令列執行方式

喜歡用命令列cmd+Notepad++編寫Java程式的朋友經常發現出現找不到同一個包下的其他類的情況。 例如：在包test下有兩個程式A.java和B.java，但是在A.java中例項化一個B物件會報錯。程式碼如下： A.java package test; public class

maven命令列執行Main函式

不帶引數 package com.test; public class Test { public static void main(String[] args) throws Exception { System.out.println("test"); } } 編譯

Ubuntu命令列執行C程式和C++程式

首先Ctrl + T 開啟一個終端，cd到你建立C/C++檔案的目錄下。 下面以建立 helloc.c 和 hellocpp.cpp 進行演示 vim helloc.c 按 i 進入插入操作，然後寫C程式碼： #include<stdio.h> int

Python 命令列執行指令碼時，提示匯入的包找不到

在Pydev能正常執行的指令碼，在匯出後在命令列執行，通常會報自己寫的包匯入時找不到。 一：報錯原因 在PyDev中，test.py 中匯入TestUserCase裡面的py檔案時，會寫： from TestUserCase.Test import Test 這在命Pydev中沒錯，

演算法第四版程式命令列執行注意要點：

將algs4.jar解壓得到的源程式中package edu…改為import edu.princeton.cs.algs4.*; 否則會提示無法載入主類 import numpy as np from DEIndividual import DEIndividual im

Jmeter命令列執行並生成HTML報告

前提：準備好jmeter指令碼，找到jmeter配置檔案檢視生成的日誌格式是否為csv，如果不是請改為csv 注意：使用命令執行jmeter指令碼必須使用jmeter 3.0及以上版本1、使用命令列執行指令碼並生成日誌檔案進入到jmeter的bin目錄下執行命令：jmeter -n -t D:\test

kettle中命令列執行轉換或者作業

很多時候，我們都會講做好的kettle轉換或者作業放到linux伺服器執行，基於linux沒有圖形介面，所以此時用命令列是必要的了。 一、windows下cmd執行方式： 首先說一下windows下如何執行，算是一個回顧吧： 找到kettle安裝目錄，在目錄下直接cmd可以進入該目錄

golang命令列引數解析

package main import ( "fmt" "os" ) func main(){ s:= os.Args fmt.Println(s) } 　　直接執行 輸出結果：[C:\Users\Administrator\AppData\Local\Temp\___go_bui

在命令列執行C++程式帶來的思考（C++學習筆記 26）

我的電腦是Win10 x64位，裝有編譯程式的軟體有CodeBlocks和Dev-C++。 我想用windows的cmd命令列來帶引數地進行編譯C++程式碼。但是網上說下載MinGw，但這個MinGw是幹什麼用的？ 首先了解以下概念： IDE: 整合開發環境（IDE，Integr

python django專案命令列執行出現ModuleNotFoundError: No module named

python專案結構描述 開發工具： pycharm 開發環境： python=3.6 django=2.1 專案結構： 問題描述 直接在pycharm中啟動django專案，可以正常直接執行transfer_util.py檔案，但在伺服器上單獨用pytho

命令列執行Python指令碼時傳入引數的三種方式

三種常用的方式 如果在執行python指令碼時需要傳入一些引數，例如gpus與batch_size，可以使用如下三種方式。 python script.py 0,1,2 10 python script.py -gpus=0,1,2 --batch-size=10 p

怎麼用命令列執行.java檔案

第一步 假如在e盤裡有個資料夾名為Java，在Java資料夾裡有一個事先寫好的.java程式，如下： 第二步 開啟命令列視窗（Window+R），輸入cmd，點確定，如下： 第三步 轉到HelloWorld.java檔案所在的資料夾。 （1）檔案在E盤，首先輸入

Windows命令列執行C/C++

時間：2018-11-04 IDE：vs2013 系統：windows8.1 簡介：在命令提示符通過vs2013的c/c++編譯器等執行相應程式 一、配置環境變數 步驟1：C:\Program Files (x86)\Microsoft Visual St

jmeter命令列執行與生成報告 Jmeter Thread Group中如果存在HTTP request執行失敗，就對整個Thread Group重新執行，限定最大執行次數N次

轉自：http://www.cnblogs.com/kongzhongqijing/p/7216693.html 一、     使用命令列方式執行Jmeter 1.1　為什麼 使用GUI方式啟動jmeter，執行執行緒較多的測試時，會造成記憶

C#呼叫命令列執行python指令碼，這個辦法可以呼叫python第三方模組和對本地檔案進行操作

string pythonScriptPath = Server.MapPath(@"~\pythonScript");//python指令碼所在的目錄 ProcessStartInfo start = new Pro

C# 呼叫命令列執行Cmd命令

1.不知道為啥   process.StartInfo.Arguments = "/c" + "start D:/Tim/Bin/QQScLauncher.exe";  這個執行命令一定要加/c ，/c ，/c，重要的事說3遍 才能正常編譯並執行 cmd /c dir：是執

自己寫一個類,模擬命令列引數

1,命令列引數 #parser = argparse.ArgumentParser() #parser.add_argument('dataroot', help='path to dataset of kaggle ultrasound nerve segmentation') ## pars