批量更新的代码java_java应用多线程批量更新脚本

#!/usr/bin/env python

# -*- coding:utf-8 -*-

# Author: 陈二狗

# 灰度环境更新脚本 执行方式：python deploy.py (all|appname)

import os, sys

import paramiko

import select

import time

import re

import threading

# 远程操作

def SSHCMD(host, command):

pkey = paramiko.RSAKey.from_private_key_file('/home/tzg/.ssh/id_rsa')

trans = paramiko.Transport((host, 22))

trans.connect(username='tzg', pkey=pkey)

ssh = paramiko.SSHClient()

ssh._transport = trans

stdin, stdout, stderr = ssh.exec_command(command)

print(stdout.read().decode())

trans.close()

# 把应用包上传到相应服务器

def SSHSCP(host, localpath, remotepath):

pkey = paramiko.RSAKey.from_private_key_file('/home/tzg/.ssh/id_rsa')

trans = paramiko.Transport((host, 22))

trans.connect(username='tzg', pkey=pkey)

sftp = paramiko.SFTPClient.from_transport(trans)

sftp.put(localpath, remotepath)

trans.close()

# 重启tomcat 打印ERROR日志和启动完成日志

def RUNTOMCAT(host, command):

pkey = paramiko.RSAKey.from_private_key_file('/home/tzg/.ssh/id_rsa')

s = paramiko.SSHClient()

s.set_missing_host_key_policy(paramiko.AutoAddPolicy())

s.connect(host, username='tzg', pkey=pkey)

transport = s.get_transport()

channel = transport.open_session()

channel.exec_command(command)

while 1:

if channel.exit_status_ready():

break

else:

rl, wl, xl = select.select([channel], [], [], 1)

if len(rl) > 0:

recv = channel.recv(1024)

if re.search('] ERROR', recv) <> None:

print recv

if re.search('Server startup in', recv) <> None:

print recv

break

# 部署单个java应用在单台服务器上

def DEPLOYWAR(host, warname, appname):

print "%s 上 %s 开始更新" % (host, appname)

SSHSCP(host, '/opt/tomcat-war/%s' % warname, '/tmp/%s' % warname)

SSHCMD(host, 'mkdir /opt/web-data-bak/tzg-%s_%s' % (appname, deploytime))

SSHCMD(host, 'unzip -oq /tmp/%s -d /opt/web-data-bak/tzg-%s_%s' % (warname, appname, deploytime))

SSHCMD(host, 'rm -rf /opt/web-data/tzg-%s' % appname)

SSHCMD(host, 'ln -s /opt/web-data-bak/tzg-%s_%s /opt/web-data/tzg-%s' % (appname, deploytime, appname))

RUNTOMCAT(host, '/opt/scripts/tomcat.sh %s restart' % appname)

print "\033[1;33m %s 上 %s 已更新完成 \033[0m" % (host, appname)

# 部署单个java应用在多台服务器上

def RUNTOMCATALL(name):

for host in warmap.get(name)[2]:

DEPLOYWAR(host, warmap.get(name)[0], name)

# 部署一批应用，多线程方式同时部署节省时间

def DEPLOY(list):

if len(list) > 0:

for name in list:

t = threading.Thread(target=RUNTOMCATALL, args=(name,))

t.setDaemon(False)

t.start()

time.sleep(15)

# 部署时间，给应用加上时间戳

deploytime = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime())

# 所有应用和服务器信息，如应用部署和服务器信息有变动应及时修改， key[1] 有 0 2 4 三种值 代表分批启动的三种类型应用

warmap={'b': ['b.war', 4, ['cunguanweb01',]],

'actprovider': ['tzg-actprovider.war', 2, ['cunguanrmi01', ]],

'act': ['tzg-act.war', 4, ['cunguanweb01',]],

'asset': ['tzg-asset.war',4, ['cunguanapp01', ]],

'autoinvest': ['tzg-autoinvest.war', 2, ['cunguanrmi01', ]],

'awardsgrant': ['tzg-awardsgrant.war', 2, ['cunguanrmi02',]],

'bi': ['tzg-bi-provider.war', 2, ['cunguanrmi01',]],

'bondstransfer': ['tzg-bondstransfer-provider.war', 2, ['cunguanrmi02',]],

'callback': ['tzg-callback-provider.war', 2, ['cunguanweb01', ]],

'copperbao': ['tzg-copperBao-provider.war', 2, ['cunguanrmi03', ]],

'growthpoint': ['tzg-growthpoint-provider.war', 2, ['cunguanrmi02', ]],

'merchant': ['tzg-merchant.war', 4, ['cunguanapp01', ]],

'message': ['tzg-message-provider.war', 2, ['cunguanrmi03', ]],

'mq': ['tzg-mq.war', 4, ['cunguanrmi03',]],

'pay': ['tzg-pay-provider.war', 2, ['cunguanrmi02',]],

'pyramid': ['tzg-pyramid-provider.war', 2, ['cunguanrmi01', ]],

'rest': ['tzg-rest.war', 4, ['cunguanweb01', ]],

'rmi': ['tzg-rmi-server.war', 0, ['cunguanrmi01',]],

'salary': ['tzg-salary.war', 2, ['cunguanrmi03',]],

'wap': ['tzg-wap.war', 4, ['cunguanweb01',]],

'web': ['tzg-web.war', 4, ['cunguanweb01',]],

'cas': ['cas.war', 4, ['cunguanapp01',]],

'partner': ['tzg-partner.war', 4, ['cunguanapp01',]],

'portal': ['tzg-portal.war', 4, ['cunguanapp01',]],

'risk': ['tzg-risk.war', 4, ['cunguanapp01',]],

'store': ['tzg-store.war', 4, ['cunguanapp01',]],

}

# 定义分三批启动的应用列表

firstlist = []

secondlist = []

thirdlist = []

# 外部传参

applist = sys.argv

applist.remove(sys.argv[0])

# 定义参数为 all 时的应用列表

if applist[0] == 'all':

firstlist = ['rmi']

secondlist = ['actprovider', 'autoinvest', 'awardsgrant', 'bi', 'bondstransfer', 'callback', 'copperbao', 'growthpoint', 'message', 'pay', 'pyramid', 'salary']

thirdlist = ['b', 'act', 'rest', 'web', 'wap','mq']

else:

for name in applist:

if warmap.get(name)[1] == 0:

firstlist.append(name)

elif warmap.get(name)[1] == 2:

secondlist.append(name)

elif warmap.get(name)[1] == 4:

thirdlist.append(name)

else:

print "%s 不存在" % name

if firstlist == []:

pass

else:

print firstlist

if secondlist == []:

pass

else:

print secondlist

if thirdlist == []:

pass

else:

print thirdlist

alllist = [firstlist,secondlist,thirdlist]

if __name__ == "__main__":

for i in alllist:

DEPLOY(i)