GoLang执行需要输入密码的命令

佚名 6年前 (2019-04-24) 随笔 585人围观抢沙发百度已收录

需求

同步一组视频到其他节点
限制带宽，防止占用过多带宽对访问造成影响

分析

Linux命令scp，rsync满足需求
使用以上命令都需要输入服务器密码，这里给出2个方案
- 对各个节点做免密
- 使用gexpect库

免密（推荐）

实例演示服务器用户名为root，IP为192.168.1.208

本地生成公私钥

ssh-keygen （一路回车默认即可）

以上命令会在~/.ssh下创建两个密钥

id_rsa (私钥)
id_rsa.pub (公钥)

上传公钥

ssh-copy-id -i ~/.ssh/id_rsa.pub root@192.168.1.208

以上命令会在服务器的~/.ssh/authorized_keys文件中写入客户端的公钥

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

### 执行命令同步
前两步已经实现了服务端对客户端的免密，通过ssh root@192.168.1.208进行测试。然后就可以使用os/exec调用系统命令来进行同步，具体代码如下

func main() {
   args := []string{
       "-avz", "--delete",
       "--timeout=60", "--bwlimit=3000",
       "/home/wsl/video/",
       "root@192.168.1.208:/root/wsl/video/",
   }

   cmd := exec.Command("rsync", args...)
   var stdOut, stdErr bytes.Buffer
   cmd.Stderr = &stdErr
   cmd.Stdout = &stdOut
   err := cmd.Run()
   if err != nil {
       log.Fatalf("Got error:%s, msg:%s", err, stdErr.String())
   }
   fmt.Println("success:", stdOut.String())
}

使用`gexpect`

gexpect : https://github.com/ThomasRooney/gexpect

package main

import (
    "fmt"
    "log"
    "time"

    "github.com/ThomasRooney/gexpect"
)

func main() {
    cmd := "rsync -avz -e 'ssh -p 22' --delete --timeout=60 --bwlimit=3000 /home/video root@192.168.1.208:/root/"
    pwd := "yourpassword"
    
    child, err := gexpect.Spawn(cmd)
    if err != nil {
        log.Fatal("Spawn cmd error ", err)
    }

    if err := child.ExpectTimeout("password: ", 10*time.Second); err != nil {
        log.Fatal("Expect timieout error ", err)
    }

    if err := child.SendLine(pwd); err != nil {
        log.Fatal("SendLine password error ", err)
    }

    if err := child.Wait(); err != nil {
        log.Fatal("Wait error: ", err)
    }

    fmt.Println("Success")
}