p2pool/bitcoin/worker_interface.py

   1 from __future__ import division
   2
   3 import random
   4 import weakref
   5
   6 from twisted.internet import defer
   7
   8 import p2pool
   9 from p2pool import data as p2pool_data
  10 from p2pool.bitcoin import getwork
  11 from p2pool.util import jsonrpc, variable
  12
  13 class _Page(jsonrpc.Server):
  14     def __init__(self, parent, long_poll):
  15         jsonrpc.Server.__init__(self)
  16         self.parent = parent
  17         self.long_poll = long_poll
  18
  19     def rpc_getwork(self, request, data=None):
  20         return self.parent._getwork(request, data, long_poll=self.long_poll)
  21
  22 class WorkerInterface(object):
  23     def __init__(self, compute, response_callback, new_work_event=variable.Event()):
  24         self.compute = compute
  25         self.response_callback = response_callback
  26         self.new_work_event = new_work_event
  27
  28         self.worker_views = {}
  29
  30         self.work_cache = {} # username -> blockattempt
  31         watch_id = new_work_event.watch(lambda *args: self_ref().work_cache.clear())
  32         self_ref = weakref.ref(self, lambda _: new_work_event.unwatch(watch_id))
  33
  34     def attach_to(self, res):
  35         res.putChild('', _Page(self, long_poll=False))
  36         res.putChild('long-polling', _Page(self, long_poll=True))
  37
  38     @defer.inlineCallbacks
  39     def _getwork(self, request, data, long_poll):
  40         request.setHeader('X-Long-Polling', '/long-polling')
  41         request.setHeader('X-Roll-NTime', 'expire=10')
  42
  43         if data is not None:
  44             defer.returnValue(self.response_callback(getwork.decode_data(data), request))
  45
  46         if p2pool.DEBUG:
  47             id = random.randrange(1000, 10000)
  48             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())
  49
  50         if long_poll:
  51             request_id = request.getClientIP(), request.getHeader('Authorization')
  52             if self.worker_views.get(request_id, self.new_work_event.times) != self.new_work_event.times:
  53                 if p2pool.DEBUG:
  54                     print 'POLL %i PUSH user=%r' % (id, request.getUser())
  55             else:
  56                 if p2pool.DEBUG:
  57                     print 'POLL %i WAITING user=%r' % (id, request.getUser())
  58                 yield self.new_work_event.get_deferred()
  59             self.worker_views[request_id] = self.new_work_event.times
  60
  61         if request.getUser() in self.work_cache:
  62             res = self.work_cache[request.getUser()]
  63         else:
  64             res = self.compute(request.getUser())
  65
  66         self.work_cache[request.getUser()] = res.update(timestamp=res.timestamp + 12) # XXX doesn't bound timestamp
  67
  68         if p2pool.DEBUG:
  69             print 'POLL %i END identifier=%i user=%r' % (id, self.new_work_event.times, request.getUser())
  70
  71         defer.returnValue(res.getwork(identifier=str(self.new_work_event.times)))