从昨晚写到今天啊，
第一版测试成功了。
PHP和JAVA的更新相对来说，PHP的自动化更新是简单多啦~~
当然，这只是运维工作当中的一环~~
#!/usr/bin/env python # -*- coding: utf-8 -*- import os,sys,commands,subprocess import re,time,logging from optparse import OptionParser #启动日志模块 logging.basicConfig(level=logging.INFO, format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s', datefmt='%a, %d %b %Y %H:%M:%S', filename='updateSVN.log', filemode='a') #将输入的参数与相应的配置文件匹配，并形成特殊字典，方便后续取值 def get_svn_list(name, ip, version): svn_update = {} for line in open("data_svn_src_java"): line_item = line.strip("\n").split("|") if ip == line_item[0] and name == line_item[1]: line_item = line.strip("\n").split("|") svn_update["salt_minion_ip"] = line_item[0] svn_update["svn_name"] = line_item[1] svn_update["salt_master_dir"] = "/srv/salt/%s/" % line_item[0] svn_update["server_dir"] = line_item[2] svn_update["svn_dir"] = line_item[3] svn_update["package_name"] = line_item[4] svn_update["svn_version"] = version if len(svn_update) != 7: error_output("database have none item at your xargs.") return svn_update #统一错误输出，红色文字提示 def error_output(error_str): print '\033[0;31;40m' print error_str + "<br>" print '\033[0m' sys.exit(1) #统一调用执行系统命令，并产生相应输出及日志记录 def exec_cmd(cmd): str_now = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) (status,output) = commands.getstatusoutput(cmd) try: if int(status) == 0: print "%s %s is ---OK!<br>" % (str_now, cmd) logging.info(cmd + " ---is OK!") return output else: logging.info(cmd + " cant't finish...") print "%s %s can't finish...<br>" % (str_now, cmd) sys.exit(1) except: logging.info(cmd + " is Wrong") print "%s %s --- is Wrong<br>" % (str_now, cmd) sys.exit(1) #在真正应用SALTSTACK之前，先测试是否正常连接 def salt_minion_test(salt_minion_ip): if "True" not in exec_cmd("salt '%s' test.ping" % salt_minion_ip): error_output("salt minion %s is not active,please restart this minion and execute is again. " % salt_minion_ip) else: print "salt minion ip %s is ok.<br>" % salt_minion_ip #从SVN相应目录里取出文件，存放在SALTSTACK的相应目录里（以IP作为目录存放位置） def get_svn_file(svn_dir, svn_version, salt_master_dir): exec_cmd("rm -rf %s" % salt_master_dir) exec_cmd("mkdir -p %s" % salt_master_dir) svnLog_cmd ="svn co %s %s --username cheng168 --password chengcheng -r %s" % (svn_dir, salt_master_dir, svn_version) output = exec_cmd(svnLog_cmd) if svn_version not in output: error_output("svn version is Wrong<br>") sys.exit(1) #用MAVEN生成相应的JAR或WAR包 def gen_package(salt_master_dir): mvn_cmd = "mvn clean install -Dmaven.test.skip=true -Pprod -f %spom.xml" % (salt_master_dir) exec_cmd(mvn_cmd) #由于数据频道特殊，脚本会在解压之后生成一个备份。 def update_back(): print "back_pass" #以下几个函数是为了区别consumer和provider的更新，主要是执行相关系统调用 def shutdown_consumer_tomcat(salt_minion_ip, server_dir): tomcat_dir = '/'.join(server_dir.split('/')[:-3]) shutdown_A_cmd = "salt '%s' cmd.run 'su - tomcat -c %s/bin/shutdown.sh'" % (salt_minion_ip, tomcat_dir) exec_cmd(shutdown_A_cmd) time.sleep(2) shutdown_B_cmd = """salt '%s' cmd.run "ps aufx|grep tomcat|grep %s/endorsed|grep -v grep|awk '{print $2}'|xargs kill -9" """% (salt_minion_ip, tomcat_dir) exec_cmd(shutdown_B_cmd) def shutdown_provider_java(salt_minion_ip, package_name): shutdown_jar_cmd = """salt '%s' cmd.run "ps aufx|grep %s|grep -v grep|awk '{print $2}'|xargs kill -9" """ % (salt_minion_ip, package_name) exec_cmd(shutdown_jar_cmd) def startup_consumer_tomcat(salt_minion_ip, server_dir): tomcat_dir = '/'.join(server_dir.split('/')[:-3]) startup_cmd = "salt '%s' cmd.run 'su - tomcat -c %s/bin/startup.sh'" % (salt_minion_ip, tomcat_dir) #exec_cmd(startup_cmd) #此处不能再调用exec_cmd，因为不能及时获取返回值，再开一个子进程，就流畅多啦。。 subprocess.Popen(startup_cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) def startup_provider_java(salt_minion_ip, server_dir, package_name): #此处太过技巧化，希望以后能写成更好理解的，三引号的使用，切换用户帐号，后台执行服务器命令都全了。 startup_cmd = """salt '%s' cmd.run 'su - tomcat -c "java -jar %s%s &"'""" % (salt_minion_ip, server_dir, package_name) print startup_cmd #exec_cmd(startup_cmd) subprocess.Popen(startup_cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) def update_consumer_package(salt_minion_ip, server_dir, package_name): #几个变量主要是为了精简配置文件，将备份放在合适的目录 package_dir = ''.join(package_name.split('.')[0]) tomcat_dir = '/'.join(server_dir.split('/')[:-3]) salt_master_file = "salt://%s/target/%s" % (salt_minion_ip, package_name) salt_minion_file = server_dir + package_name back_minion_dir = tomcat_dir + "/backup/" update_package_cmd = "salt '%s' cp.get_file %s %s makedirs=True" % (salt_minion_ip, salt_master_file, salt_minion_file) exec_cmd(update_package_cmd) unzip_cmd = "salt '%s' cmd.run 'unzip %s%s -d %s'" % (salt_minion_ip, server_dir, package_name, server_dir) exec_cmd(unzip_cmd) chown_cmd = "salt '%s' cmd.run 'chown tomcat:tomcat %s'" % (salt_minion_ip, server_dir) exec_cmd(chown_cmd) #小峤要求解压后删除WAR文件，便于后期直接更新静态文件，我在此处正好将它备份起来 mv_cmd = "salt '%s' cmd.run 'mv %s %s'" % (salt_minion_ip, salt_minion_file, back_minion_dir) exec_cmd(mv_cmd) #因为DUBBO的CONSUMER消费者为WAR包，服务提供者为JAR包，所以启动，关闭，更新上传都分成了两套函数 def update_provider_package(salt_minion_ip, server_dir, package_name): salt_master_file = "salt://%s/target/%s" % (salt_minion_ip, package_name) salt_minion_file = server_dir + package_name update_package_cmd = "salt '%s' cp.get_file %s %s makedirs=True" % (salt_minion_ip, salt_master_file, salt_minion_file) exec_cmd(update_package_cmd) cp_cmd = "salt '%s' cmd.run 'cp %s %s.bak'" % (salt_minion_ip, salt_minion_file, salt_minion_file) exec_cmd(cp_cmd) chown_cmd = "salt '%s' cmd.run 'chown tomcat:tomcat %s*'" % (salt_minion_ip, server_dir) exec_cmd(chown_cmd) def main(name, ip, version): svn_update = get_svn_list(name, ip, version) salt_minion_test(svn_update["salt_minion_ip"]) get_svn_file(svn_update["svn_dir"], svn_update["svn_version"], svn_update["salt_master_dir"]) gen_package(svn_update["salt_master_dir"]) update_back() #服务器和消费的前面操作都相同，但在此后，要分路径判断执行相应的操作 if svn_update["package_name"] == "consumer.war" : shutdown_consumer_tomcat(svn_update["salt_minion_ip"], svn_update["server_dir"]) update_consumer_package(svn_update["salt_minion_ip"], svn_update["server_dir"], svn_update["package_name"]) startup_consumer_tomcat(svn_update["salt_minion_ip"], svn_update["server_dir"]) if svn_update["package_name"] == "provider.jar" : shutdown_provider_java(svn_update["salt_minion_ip"], svn_update["package_name"]) update_provider_package(svn_update["salt_minion_ip"], svn_update["server_dir"], svn_update["package_name"]) startup_provider_java(svn_update["salt_minion_ip"], svn_update["server_dir"], svn_update["package_name"]) if __name__=="__main__": #输入参数判断，如有输入不完，及早提示并退出 usage = "usage: %prog [options] arg :./update.py -v 12292 -n bbs -i 192.168.11.75" parser = OptionParser(usage) parser.add_option("-v", "--version", dest="version", help="input the version(Sample: 15435...).") parser.add_option("-n", "--name", dest="name", help="input the svn name(consumer|provider)") parser.add_option("-i", "--ip", dest="ip", help="input the ip address(1.4.8.25|11.14.7.2)") (options, args) = parser.parse_args() version = options.version name = options.name ip = options.ip if version is None or name is None or ip is None : error_output("you must input all args: ./update_svn_shuju.py -i xx -n xx -v xx") main(name, ip, version)
