首先来描述下环境,在机器上有很多个JAVA程序,我们在每个JAVA程序里都配置了一个启动|停止|重启的脚本
举个例子:
我们现在要同时运行这些脚本,来达到快速启动所有的JAVA程序,如果我们只用多线程的话,线程是不会返回消息给父进程,我们如何才能知道这些程序是启动成功了呢?
所以我们用到了队列来管理。
"""我试过gevent,但是会在command这里造成阻塞"""
gevent代码如下 如果有朋友知道如何优化,请您告诉我
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
#!/usr/bin/python2.7
# -*- coding:utf-8 -*-
import os,sys
from datetime import datetime
import commands
import gevent.monkey
gevent.monkey.patch_os()
import gevent
def Servers():
servers = commands.getoutput( '''ls /data/program/payment/ | grep 'payment' ''' )
servers = servers.split( '\n' )
return servers
def handle(servername):
if sys.argv[ 1 ] = = 'start' or sys.argv[ 1 ] = = 'stop' or sys.argv[ 1 ] = = 'restart' :
print '\033[1;31;40m'
print '========================>>>go to handle %s<<<=========================' % servername
print '\033[0m'
r = commands.getoutput( '''su - tomcat -c "/data/program/payment/%s/bin/server.sh %s &" ''' % (servername,sys.argv[ 1 ])) #在这里会阻塞,我们无法找到合适的地方进行协程的切换
gevent.sleep( 0 ) #无论放到何处,不是之前就是切换之后都会阻塞。
print r
else :
print 'Please Use start | stop | restart To Handle The Command'
sys.exit( 1 )
if __name__ = = '__main__' :
s = Servers()
threads = []
for i in s:
threads.append(gevent.spawn(handle,i))
# print threads
gevent.joinall(threads)
|
多线程代码如下
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
#!/usr/bin/python2.7
# -*- coding:utf-8 -*-
from datetime import datetime
import commands
from Queue import Queue
from threading import Thread
_sentinel = object ()
def Servers():
servers = commands.getoutput( '''ls /data/program/payment/ | grep 'payment' ''' )
servers = servers.split( '\n' )
return servers
def producer(servername,out_q):
if sys.argv[ 1 ] = = 'start' or sys.argv[ 1 ] = = 'stop' or sys.argv[ 1 ] = = 'restart' :
print '\033[1;31;40m'
print '========================>>>put %s in Queue<<<=========================' % servername
print '\033[0m'
out_q.put_nowait(commands.getoutput( '''su - tomcat -c "/data/program/payment/%s/bin/server.sh %s &" ''' % (servername,sys.argv[ 1 ]))) #放入队列的对象
else :
print 'Please Use start | stop | restart To Handle The Command'
sys.exit( 1 )
def consumer(servername,in_q):
n = len (servername)
while n > 0 : #循环在队列中取结果,直到循环结束
data = in_q.get()
n - = 1
print '\033[1;31;40m'
print data
print '\033[0m'
print '\033[1;31;40m'
print 'consumer was done!!!!!!!'
print '\033[0m'
if __name__ = = '__main__' :
s = Servers()
q = Queue()
t1 = Thread(target = consumer, args = (s,q,)) #消费者在队列中获取结果,前面的函数内部已经循环获取
for i in s:
t2 = Thread(target = producer, args = (i,q,)) #讲线程进行管理,放入队列
t2.start() #启动生产者线程
# t2.join() #启动生产者以后放弃校验线程是否结束,进行并发,因为我们是把线程放入队列进行管理的,所以不用在这里等待线程结束,如果使用了join这里会阻塞我们的程序。线程结束后,消费者会通知父进程线程已经结束。
t1.start() #启动消费者线程
t1.join() #在获取完成之前进行线程的阻塞
|
简单的说下join这个方法:
调用Thread.join将会使主调线程堵塞,直到被调用线程运行结束或超时。参数timeout是一个数值类型,表示超时时间,如果未提供该参数,那么主调线程将一直堵塞到被调线程结束。
以上所述就是本文的全部内容了,希望大家能够喜欢。