RanchiMall/electrumx
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'client-limiting' into develop

Browse Source
This commit is contained in:
Neil Booth 2016-11-17 06:30:46 +09:00
commit 64333c3dfd
1 changed files with 45 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
server/protocol.py
View File

@ -108,9 +108,11 @@ class ServerManager(LoggedClass):
def notify(self, height, touched): def notify(self, height, touched):
'''Notify sessions about height changes and touched addresses.''' '''Notify sessions about height changes and touched addresses.'''
sessions = [session for session in self.sessions cache = {}
if isinstance(session, ElectrumX)] for session in self.sessions:
ElectrumX.notify(sessions, height, touched) if isinstance(session, ElectrumX):
# Use a tuple to distinguish from JSON
session.jobs.put_nowait((height, touched, cache))
def stop(self): def stop(self):
'''Close listening servers.''' '''Close listening servers.'''
@ -196,7 +198,7 @@ class Session(JSONRPC):
self.coin = bp.coin self.coin = bp.coin
self.kind = kind self.kind = kind
self.hash168s = set() self.hash168s = set()
self.requests = asyncio.Queue() self.jobs = asyncio.Queue()
self.current_task = None self.current_task = None
self.client = 'unknown' self.client = 'unknown'
@ -222,26 +224,23 @@ class Session(JSONRPC):
def on_json_request(self, request): def on_json_request(self, request):
'''Queue the request for asynchronous handling.''' '''Queue the request for asynchronous handling.'''
self.requests.put_nowait(request) self.jobs.put_nowait(request)
async def serve_requests(self): async def serve_requests(self):
'''Asynchronously run through the task queue.''' '''Asynchronously run through the task queue.'''
while True: while True:
await asyncio.sleep(0) await asyncio.sleep(0)
request = await self.requests.get() job = await self.jobs.get()
try: try:
start = time.time() if isinstance(job, tuple): # Height / mempool notification
await self.handle_json_request(request) await self.notify(*job)
secs = time.time() - start else:
if secs > 1: await self.handle_json_request(job)
self.logger.warning('slow request for {} took {:.1f}s: {}'
.format(self.peername(), secs,
request))
except asyncio.CancelledError: except asyncio.CancelledError:
break break
except Exception: except Exception:
# Getting here should probably be considered a bug and fixed # Getting here should probably be considered a bug and fixed
self.logger.error('error handling request {}'.format(request)) self.logger.error('error handling request {}'.format(job))
traceback.print_exc() traceback.print_exc()
def peername(self, *, for_log=True): def peername(self, *, for_log=True):
@ -324,36 +323,41 @@ class ElectrumX(Session):
for prefix, suffixes in rpcs for prefix, suffixes in rpcs
for suffix in suffixes.split()} for suffix in suffixes.split()}
@classmethod async def notify(self, height, touched, cache):
def notify(cls, sessions, height, touched): '''Notify the client about changes in height and touched addresses.
headers_payload = height_payload = None
for session in sessions: Cache is a shared cache for this update.
if height != session.notified_height: '''
session.notified_height = height if height != self.notified_height:
if session.subscribe_headers: self.notified_height = height
if headers_payload is None: if self.subscribe_headers:
headers_payload = json_notification_payload( key = 'headers_payload'
'blockchain.headers.subscribe', if key not in cache:
(session.electrum_header(height), ), cache[key] = json_notification_payload(
) 'blockchain.headers.subscribe',
session.send_json(headers_payload) (self.electrum_header(height), ),
)
self.send_json(cache[key])
if session.subscribe_height: if self.subscribe_height:
if height_payload is None:
height_payload = json_notification_payload(
'blockchain.numblocks.subscribe',
(height, ),
)
session.send_json(height_payload)
hash168_to_address = session.coin.hash168_to_address
for hash168 in session.hash168s.intersection(touched):
address = hash168_to_address(hash168)
status = session.address_status(hash168)
payload = json_notification_payload( payload = json_notification_payload(
'blockchain.address.subscribe', (address, status)) 'blockchain.numblocks.subscribe',
session.send_json(payload) (height, ),
)
self.send_json(payload)
hash168_to_address = self.coin.hash168_to_address
matches = self.hash168s.intersection(touched)
for hash168 in matches:
address = hash168_to_address(hash168)
status = self.address_status(hash168)
payload = json_notification_payload(
'blockchain.address.subscribe', (address, status))
self.send_json(payload)
if matches:
self.logger.info('notified {} of {} addresses'
.format(self.peername(), len(matches)))
def height(self): def height(self):
'''Return the block processor's current height.''' '''Return the block processor's current height.'''