1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/worker.py	Sun Jun 13 04:34:03 2010 +0200
     1.3 @@ -0,0 +1,48 @@
     1.4 +#!/usr/bin/python
     1.5 +
     1.6 +import com
     1.7 +import model
     1.8 +import optparse
     1.9 +import pickle
    1.10 +import SocketServer
    1.11 +
    1.12 +option_list = [
    1.13 +	optparse.make_option("-q", "--quiet", action="store_true", dest="quiet", 
    1.14 +		help="quiet mode", default=False)]
    1.15 +
    1.16 +def spawn(host, port):
    1.17 +	return Worker(host, port, Handler, ())
    1.18 +		
    1.19 +class Handler(SocketServer.BaseRequestHandler):
    1.20 +	"""
    1.21 +	"""	
    1.22 +	def handle(self):
    1.23 +		data = self.request[0].strip()
    1.24 +		socket = self.request[1]
    1.25 +		print "%s wrote:" % self.client_address[0]
    1.26 +		model_size = int(data[:10])
    1.27 +		model = pickle.loads(data[10:10+model_size])
    1.28 +		max_iter = int(data[10+model_size:20+model_size])
    1.29 +		model.update(max_iter)
    1.30 +		socket.sendto("finished\n", self.client_address)
    1.31 +		
    1.32 +class Worker(SocketServer.UDPServer):
    1.33 +	def __init__(self, host, port, handler, options):
    1.34 +		SocketServer.UDPServer.__init__(self, (host, port), handler)
    1.35 +		self.host = host
    1.36 +		self.port = port
    1.37 +		self.handler = handler
    1.38 +		self.options = options
    1.39 +	def run(self):
    1.40 +		print "worker running at port %s" % self.port
    1.41 +		self.serve_forever()
    1.42 +
    1.43 +if __name__ == "__main__":
    1.44 +	parser = optparse.OptionParser(usage="usage: %prog [options] port",
    1.45 +		option_list=option_list)
    1.46 +	options, args = parser.parse_args()
    1.47 +	
    1.48 +	if len(args) > 0:
    1.49 +		port = int(args[0])
    1.50 +		worker = Worker("localhost", port, Handler, options)
    1.51 +		worker.run()