1 from __future__ import division
8 from twisted.internet import defer
11 from p2pool.bitcoin import getwork
12 from p2pool.util import jsonrpc, variable
14 class _Page(jsonrpc.Server):
15 def __init__(self, parent, long_poll):
16 jsonrpc.Server.__init__(self)
18 self.long_poll = long_poll
20 def rpc_getwork(self, request, data=None):
21 return self.parent._getwork(request, data, long_poll=self.long_poll)
23 def render_GET(self, request):
24 request.content = StringIO.StringIO(json.dumps(dict(id=0, method='getwork')))
25 return self.render_POST(request)
27 class WorkerInterface(object):
28 def __init__(self, compute, response_callback, new_work_event=variable.Event()):
29 self.compute = compute
30 self.response_callback = response_callback
31 self.new_work_event = new_work_event
33 self.worker_views = {}
35 self.work_cache = {} # username -> blockattempt
36 watch_id = new_work_event.watch(lambda *args: self_ref().work_cache.clear())
37 self_ref = weakref.ref(self, lambda _: new_work_event.unwatch(watch_id))
39 def attach_to(self, res):
40 res.putChild('', _Page(self, long_poll=False))
41 res.putChild('long-polling', _Page(self, long_poll=True))
43 @defer.inlineCallbacks
44 def _getwork(self, request, data, long_poll):
45 request.setHeader('X-Long-Polling', '/long-polling')
46 request.setHeader('X-Roll-NTime', 'expire=10')
49 defer.returnValue(self.response_callback(getwork.decode_data(data), request))
52 id = random.randrange(1000, 10000)
53 print 'POLL %i START long_poll=%r user_agent=%r x-work-identifier=%r user=%r' % (id, long_poll, request.getHeader('User-Agent'), request.getHeader('X-Work-Identifier'), request.getUser())
56 request_id = request.getClientIP(), request.getHeader('Authorization')
57 if self.worker_views.get(request_id, self.new_work_event.times) != self.new_work_event.times:
59 print 'POLL %i PUSH user=%r' % (id, request.getUser())
62 print 'POLL %i WAITING user=%r' % (id, request.getUser())
63 yield self.new_work_event.get_deferred()
64 self.worker_views[request_id] = self.new_work_event.times
66 if request.getUser() in self.work_cache:
67 res = self.work_cache[request.getUser()]
69 res = self.compute(request.getUser())
71 self.work_cache[request.getUser()] = res.update(timestamp=res.timestamp + 12) # XXX doesn't bound timestamp
74 print 'POLL %i END identifier=%i user=%r' % (id, self.new_work_event.times, request.getUser())
76 defer.returnValue(res.getwork(identifier=str(self.new_work_event.times)))