taskctl命令行类（sh、exe、python新增scp）插件升级扩展

ffw_880106

转载自 ： http://www.taskctl.com/forum/detail_129.html

    上次写了一个帖子 TASKCTL中不使用代理，通过ssh免密连接执行远程脚本配置（SSH插件扩展）http://www.taskctl.com/forum/detail_122.html 。写完后发现，其实在原本的sh作业类型上改造一下就可以了，同时延伸到以前是用command执行方式的作业类型，都可以适当扩展，命令行类插件升级20171110.rar就是 sh、exe、python的插件和新增拷贝一些小文件的scp作业类型。插件中主要使用的ssh免密及其相关的几个参数和bash的几种模式，这个只是一个基础版本，如果有特殊的需求基于上面的插件改造一下就可以了。下面是摘抄了原网页上写法写的一个sh脚本的扩展，其他几个类似。

使用TASKCTL调度服务时，如果需要跨用户或者跨服务器调度可以通过安装taskctl的代理来实现，但有时需要执行作业的服务器上仅仅只有几个脚本或者一两个可执行程序，此时再部署代理，就显得有点得不偿失了。我们知道TASKCTL支持任意作业类型的扩展，所以我们是不是可以考虑通过SSH远程连接来扩展这些插件，使其能执行这些远程少量的脚本程序。

    下面是通过扩展TASKCTL用ssh连接来支持调度远程作业的步骤，一方面解决一些朋友迫切的调度需求，另一方面抛砖引玉，让大家能够在TASKCTL中扩展更多的适应自己需求的作业插件。

    下面编写的过程参照 http://www.taskctl.com/forum/detail_33.html 。

    1、首先在服务端的$TASKCTLDIR/src/plugin/sh/shell/目录中新增cprunsh.sh作业插件，实际上就是一个shell程序，代码如下：

1. #!/bin/sh
   
2. #------------------------------------------------------------------------------
   
3. #        功能: [ssh方式]执行[远程]Shell脚本驱动插件程序
   
4. #
   
5. #        更新日志：20171030 由最开始的集成转换为插件，增强扩展原只能执行当前用户可读的 shell 脚本，
   
6. #        通过ssh协议，在免密的情况下可以执行本机不同用户和非本机的其他用户
   
7. #
   
8. #        参数:
   
9. #        1. progname 脚本的全路径名称
   
10. #        2. para 脚本的入口参数，格式为 para1 para2 ...
    
11. #        3. exppara 对应相应脚本的exppara属性，为脚本文件所在用户，当该参数为空时，默认为当前用户,如果不是当前用户时，为远程连接IP地址和用户名
    
12. #        sip=172.22.1.1 ssh远程IP地址
    
13. #        sport=9527 ssh远程连接端口
    
14. #        suser=taskctl ssh远程用户
    
15. #        返回值: 脚本执行后的实际返回值
    
16. #       
    
17. #        流程模式代码任务定义举例:
    
18. #        1、默认2个输入参数的安装用户可执行的服务器上的sh脚本(如果参数多个，以此类推)
    
19. #        <sh>
    
20. #        <name>job1</name>
    
21. #        <progname>/home/taskctl/test.sh</progname>
    
22. #        <para>para1 para2</para>
    
23. #        </sh>
    
24. #        2、在同一台服务器上的其他用户可执行，当前节点安装用户不可执行sh脚本，当前用户和目标用户做了ssh免密(如果参数多个，以此类推)
    
25. #        <sh>
    
26. #        <name>job1</name>
    
27. #        <progname>/home/username/test.sh</progname>
    
28. #        <para>para1 para2</para>
    
29. #        <exppara>[sport=22,]suser=taskctl</exppara>
    
30. #        </sh>
    
31. #        3、不同服务器上，但都是shell作业，当前节点安装用户已经做了ssh免密(如果参数多个，以此类推)
    
32. #        <sh>
    
33. #        <name>job1</name>
    
34. #        <progname>/home/username/test.sh</progname>
    
35. #        <para>para1 para2</para>
    
36. #        <exppara>sip=172.22.1.1,[sport=22,]suser=taskctl</exppara>
    
37. #        </sh>
    
38. #
    
39. #------------------------------------------------------------------------------
    
40. 
    
41. if [ $# -ne 3 ]
    
42. then
    
43. echo "Param error !"
    
44. echo "Usage: $0 progname para expara"
    
45. exit 126
    
46. fi
    
47. 
    
48. #------------------------------------------------------------------------------
    
49. # 第一步： 接收参数
    
50. #------------------------------------------------------------------------------
    
51. ProgName=$1
    
52. Para=$2
    
53. ExpPara=$3
    
54. 
    
55. #------------------------------------------------------------------------------
    
56. # 第二步： 解析 exppara
    
57. #------------------------------------------------------------------------------
    
58. function getValue(){
    
59. if [ $# -ne 2 ]
    
60. then
    
61. echo "......."
    
62. exit 1
    
63. fi
    
64. 
    
65. str=`echo $2|awk -v record=$1 'BEGIN {
    
66. split(record,myarray,",");
    
67. 
    
68. }
    
69. END{
    
70. for (i in myarray){
    
71. if (myarray[i]~$1){
    
72. if ( index(myarray[i],$1) == 1){
    
73. print myarray[i];
    
74. break;
    
75. }
    
76. }
    
77. }
    
78. }
    
79. '`
    
80. echo $str|awk -F"=" '{print $2}'
    
81. }
    
82. 
    
83. ExpPara=`echo $ExpPara` #此处为了去前后空格,以便判断是否真正有入口参数
    
84. #通过判断 exppara 是否为空，决定是直接执行sh还是走ssh通道
    
85. #直接执行
    
86. if [ ! -n "$ExpPara" ]
    
87. then
    
88. sh $ProgName $Para
    
89. ret=$?
    
90. exit $ret
    
91. fi
    
92. #走ssh通道
    
93. sip=`getValue $ExpPara sip`
    
94. #通过判断 sip 是否为空，决定是直接执行本机其他用户还是远程服务器上的sh脚本
    
95. if [ ! -n "$sip" ]
    
96. then
    
97. sip=localhost
    
98. fi
    
99. #sport=`getValue $ExpPara sport` #默认没有修改端口 传参数的时候也没有传 先屏蔽
    
100. sport=22
     
101. suser=`getValue $ExpPara suser`
     
102. 
     
103. #------------------------------------------------------------------------------
     
104. # 第三步： 检测是否配置了ssh免密连接
     
105. #------------------------------------------------------------------------------
     
106. ssh -o ConnectTimeout=5 -o NumberOfPasswordPrompts=0 -o StrictHostKeyChecking=no -p $sport $suser@$sip 'pwd' &>/dev/null
     
107. if [ $? != 0 ];then
     
108. echo -e "\nSSH Connection failed $sip"
     
109. echo -e "\nssh-keygen -t rsa -P ''"
     
110. echo -e "\nssh-copy-id -p $sport $suser@$sip"
     
111. exit 1
     
112. else
     
113. echo -e "\nSSH connection succeeded $sip"
     
114. fi
     
115. 
     
116. #------------------------------------------------------------------------------
     
117. # 第四步： 运行JOB，并返回结果
     
118. #------------------------------------------------------------------------------
     
119. #echo "ssh -o NumberOfPasswordPrompts=0 -o StrictHostKeyChecking=no -p $sport $suser@$sip bash --login $ProgName $Para" #此处为打印执行命令行,以便调试判断是否真正有入口参数
     
120. ssh -o NumberOfPasswordPrompts=0 -o StrictHostKeyChecking=no -p $sport $suser@$sip "bash --login $ProgName $Para"
     
121. ret=$?
     
122. exit $ret

复制代码

    2、在桌面软件admin中进入"任务类型"功能中，设置ssh作业类型，如下图所示：

    3、在桌面软件designer中设计 sh 作业如下代码片段：

1. <!-- 按插件的规则编写的流程设置 -->
   
2. <sh>
   
3. <name>MainModul_JobNode0</name>
   
4. <progname>/home/cdchen/successjob.sh</progname>
   
5. <para>para1 para2</para>
   
6. <exppara>sip=192.168.0.192,suser=cdchen</exppara>
   
7. <jobdesc>测试远程sh作业</jobdesc>
   
8. </sh>

复制代码

    4、调试测试作业是否符合预期