Python自定义主从分布式架构实例分析

#encoding=utf-8#author: walker#date: 2014-12-03#function: 中心节点（主要功能是分配任务）import SocketServer, socket, QueueCenterIP = '127.0.0.1'  #中心节点IPCenterListenPort = 9999  #中心节点监听端口CenterClient = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) #中心节点用于发送网络消息的socketTaskQueue = Queue.Queue() #任务队列#获取任务队列def GetTaskQueue():  for i in range(1, 11):    TaskQueue.put(str(i))#CenterServer的回调函数，在接受到udp报文是触发class MyUDPHandler(SocketServer.BaseRequestHandler):  def handle(self):    data = self.request[0].strip()    socket = self.request[1]    print(data)    if data.startswith('wait'):      vec = data.split(':')      if len(vec) != 3:        print('Error: len(vec) != 3')      else:        nodeIP = vec[1]        nodeListenPort = vec[2]        nodeID = nodeIP + ':' + nodeListenPort        if not TaskQueue.empty():          task = TaskQueue.get()          print('send task ' + task + ' to ' + nodeID)          CenterClient.sendto('task:' + task, (nodeIP, int(nodeListenPort)))        else:          print('TaskQueue is empty!')GetTaskQueue() #获取任务队列CenterServer = SocketServer.UDPServer((CenterIP, CenterListenPort), MyUDPHandler)print('Listen port ' + str(CenterListenPort) + ' ...')CenterServer.serve_forever()

#encoding=utf-8#author: walker#date: 2014-12-03#function: 任务节点（请求/接收/执行任务）import time, socket, SocketServerfrom apscheduler.scheduler import SchedulerCenterIP = '127.0.0.1'  #中心节点IPCenterListenPort = 9999  #中心节点监听端口NodeIP = socket.gethostbyname(socket.gethostname())  #任务节点自身IPNodeClient = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  #任务节点用于发送网络消息的socket#任务:发送网络信息def jobSendNetMsg():  msg = ''  if NodeServer.TaskState == 'wait':    msg = 'wait:' + NodeIP + ':' + str(NodeListenPort)  elif NodeServer.TaskState == 'exec':    msg = 'exec:' + NodeIP + ':' + str(NodeListenPort)  print(msg)  NodeClient.sendto(msg, (CenterIP, CenterListenPort))#添加并启动定时任务def InitTimer():  sched = Scheduler()  sched.add_interval_job(jobSendNetMsg, seconds=1)  sched.start()#执行任务def ExecTask(task):  print('ExecTask ' + task + ' ...')  time.sleep(2)  print('ExecTask ' + task + ' over')#NodeServer的回调函数，在接受到udp报文是触发class MyUDPHandler(SocketServer.BaseRequestHandler):  def handle(self):    data = self.request[0].strip()    socket = self.request[1]    print('recv data: ' + data)    if data.startswith('task'):      vec = data.split(':')      if len(vec) != 2:        print('Error: len(vec) != 2')      else:        task = vec[1]        self.server.TaskState = 'exec'        ExecTask(task)        self.server.TaskState = 'wait'InitTimer()NodeServer = SocketServer.UDPServer(('', 0), MyUDPHandler)NodeServer.TaskState = 'wait' #(exec/wait)NodeListenPort = NodeServer.server_address[1]print('NodeListenPort:' + str(NodeListenPort))NodeServer.serve_forever()