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 WorkerBridge(object):
29 self.new_work_event = variable.Event()
31 def preprocess_request(self, request):
32 return request, # *args to self.compute
34 def get_work(self, request):
35 raise NotImplementedError()
37 def got_response(self, block_header):
38 print self.got_response, "called with", block_header
40 class WorkerInterface(object):
41 def __init__(self, worker_bridge):
42 self.worker_bridge = worker_bridge
44 self.worker_views = {}
46 self.work_cache = {} # request_process_func(request) -> blockattempt
48 new_work_event = self.worker_bridge.new_work_event
49 watch_id = new_work_event.watch(lambda *args: self_ref().work_cache.clear())
50 self_ref = weakref.ref(self, lambda _: new_work_event.unwatch(watch_id))
52 def attach_to(self, res):
53 res.putChild('', _Page(self, long_poll=False))
54 res.putChild('long-polling', _Page(self, long_poll=True))
56 @defer.inlineCallbacks
57 def _getwork(self, request, data, long_poll):
58 request.setHeader('X-Long-Polling', '/long-polling')
59 request.setHeader('X-Roll-NTime', 'expire=10')
62 defer.returnValue(self.worker_bridge.got_response(getwork.decode_data(data), request))
65 id = random.randrange(1000, 10000)
66 print 'POLL %i START is_long_poll=%r user_agent=%r user=%r' % (id, long_poll, request.getHeader('User-Agent'), request.getUser())
69 request_id = request.getClientIP(), request.getHeader('Authorization')
70 if self.worker_views.get(request_id, self.worker_bridge.new_work_event.times) != self.worker_bridge.new_work_event.times:
72 print 'POLL %i PUSH' % (id,)
75 print 'POLL %i WAITING' % (id,)
76 yield self.worker_bridge.new_work_event.get_deferred()
77 self.worker_views[request_id] = self.worker_bridge.new_work_event.times
79 key = self.worker_bridge.preprocess_request(request)
81 if key in self.work_cache:
82 res, orig_timestamp = self.work_cache.pop(key)
84 res = self.worker_bridge.get_work(*key)
85 orig_timestamp = res.timestamp
87 if res.timestamp + 12 < orig_timestamp + 600:
88 self.work_cache[key] = res.update(timestamp=res.timestamp + 12), orig_timestamp
91 print 'POLL %i END identifier=%i' % (id, self.worker_bridge.new_work_event.times)
93 defer.returnValue(res.getwork(identifier=str(self.worker_bridge.new_work_event.times)))