用Go写一个轻量级的ssh批量操作工具的方法

2020-01-28 12:54:35丽君

可以看到,两个命令都得到执行了,并在执行完 exit 后退出连接。

比较一下和 session.Run() 的区别,可以发现在 session.Shell() 模式下,输出的内容包含了主机的名字,输入的命令等等。因为这是 tty 执行的结果嘛。如果我们只需要执行命令倒也无所谓,但是如果我们还需要从执行命令的结果中读取一些信息,这些内容就显得有些臃肿了。比如我们在一台 ubuntu 上跑一下看看


=== RUN  Test_SSH
--- PASS: Test_SSH (0.98s)
    ssh_test.go:50: Welcome to Ubuntu 16.04.3 LTS (GNU/Linux 4.4.0-98-generic x86_64)

         * Documentation: https://help.ubuntu.com
         * Management:   https://landscape.canonical.com
         * Support:    https://ubuntu.com/advantage

         System information as of Thu Jan 18 16:34:56 CST 2018

         System load: 0.0        Processes:       335
         Usage of /:  10.0% of 90.18GB  Users logged in:    0
         Memory usage: 2%         IP address for eth0:  192.168.80.131
         Swap usage:  0%         IP address for docker0: 172.17.0.1

         Graph this data and manage this system at:
          https://landscape.canonical.com/

        16 个可升级软件包。
        16 个安全更新。

        New release '17.10' available.
        Run 'do-release-upgrade' to upgrade to it.

        You have new mail.
        Last login: Thu Jan 18 16:31:41 2018 from 192.168.95.104
        root@ubuntu-docker-node3:~# root@ubuntu-docker-node3:/opt# /opt
        root@ubuntu-docker-node3:/opt# 注销

最起码,上面那一堆 System information 就用不着嘛。交换机是没有办法,Linux 上能不能通过一条命令,也就是想办法 session.Run() 来执行命令组合呢?

答案是可以的,把命令通过 && 连接起来就好了嘛。LInux 的 Shell 会帮我们拆开来分别运行的,比如上面的这个命令我们就可以合并成一条命令 cd /opt&&pwd&&exit


=== RUN  Test_SSH_run
--- PASS: Test_SSH_run (0.91s)
  ssh_test.go:76: /opt

立马就简洁了对不对?

轮子

ssh 执行命令这样就差不多了。要变成一个可以用 ssh 批量操作工具,我们还要给他加上并发执行,并发限制,超时控制,输入参数解析,输出格式等等

这里就不展开了,最终这个造出来的轮子长这样:https://github.com/shanghai-edu/multissh

可以直接命令行来执行,通过 ; 号或者 , 号作为命令和主机的分隔符。


# ./multissh -cmds "show clock" -hosts "192.168.31.21;192.168.15.102" -u admin -p password

也可以通过文本来存放主机组和命令组,通过换行符分隔。


# ./multissh -cmdfile cmd1.txt.example -hostfile host.txt.example -u admin -p password

特别的,如果输入的是 IP (-ips 或 -ipfile),那么允许 IP 地址段方式的输入,例如 192.168.15.101-192.168.15.110 。(还记得 swcollector 么,类似的实现方式)