async def genrloop():
try:
while True:
mesg = await link.rx()
if mesg is None:
raise s_exc.LinkShutDown(mesg=mesg)
if mesg[0] != 't2:yield': # pragma: no cover
info = 'Telepath protocol violation: unexpected message received'
raise s_exc.BadMesgFormat(mesg=info)
retn = mesg[1].get('retn')
if retn is None:
await self._putPoolLink(link)
return
# if this is an exception, it's the end...
if not retn[0]:
await self._putPoolLink(link)
yield s_common.result(retn)
except GeneratorExit:
# if they bail early on the genr, fini the link
await link.fini()
async def __aiter__(self):
taskdone = False
while not self.isfini:
if not taskdone and self.task.done():
taskdone = True
self.task.result()
if taskdone and self.count == 0:
if not self.link.isfini:
await self.proxy._putPoolLink(self.link)
await self.fini()
return
mesg = await self.link.rx()
if self.taskexc:
raise self.taskexc
if mesg is None:
raise s_exc.LinkShutDown(mesg='Remote peer disconnected')
if mesg[0] == 't2:fini':
self.count -= 1
yield mesg[1].get('retn')
continue
logger.warning(f'Pipeline got unhandled message: {mesg!r}.') # pragma: no cover
async def taskv2(self, todo, name=None):
mesg = ('t2:init', {
'todo': todo,
'name': name,
'sess': self.sess})
link = await self.getPoolLink()
await link.tx(mesg)
mesg = await link.rx()
if mesg is None:
raise s_exc.LinkShutDown(mesg='Remote peer disconnected')
if mesg[0] == 't2:fini':
await self._putPoolLink(link)
retn = mesg[1].get('retn')
return s_common.result(retn)
if mesg[0] == 't2:genr':
async def genrloop():
try:
while True:
mesg = await link.rx()
if mesg is None:
raise s_exc.LinkShutDown(mesg=mesg)
if mesg[0] != 't2:yield': # pragma: no cover
info = 'Telepath protocol violation: unexpected message received'
raise s_exc.BadMesgFormat(mesg=info)
retn = mesg[1].get('retn')
if retn is None:
await self._putPoolLink(link)
return
# if this is an exception, it's the end...
if not retn[0]:
await self._putPoolLink(link)
yield s_common.result(retn)
except GeneratorExit:
# if they bail early on the genr, fini the link
await link.fini()
return s_coro.GenrHelp(genrloop())
if mesg[0] == 't2:share':
iden = mesg[1].get('iden')
sharinfo = mesg[1].get('sharinfo')
await self._putPoolLink(link)
return await Share.anit(self, iden, sharinfo)
async def handshake(self, auth=None):
mesg = ('tele:syn', {
'auth': auth,
'vers': televers,
'name': self.name,
})
await self.link.tx(mesg)
self.synack = await self.link.rx()
if self.synack is None:
mesg = 'socket closed by server before handshake'
raise s_exc.LinkShutDown(mesg=mesg)
self.sess = self.synack[1].get('sess')
self.sharinfo = self.synack[1].get('sharinfo', {})
self.methinfo = self.sharinfo.get('meths', {})
vers = self.synack[1].get('vers')
if vers[0] != televers[0]:
raise s_exc.BadMesgVers(myver=televers, hisver=vers)
async def rxloop():
while not self.link.isfini:
mesg = await self.link.rx()
if mesg is None:
return
try:
func = self.handlers.get(mesg[0])
if func is None:
logger.warning('Proxy.rxloop: Invalid Message: %r' %
(mesg, ))
return
await func(mesg)
except asyncio.CancelledError: # pragma: no cover
raise
except Exception:
logger.exception('Proxy.rxloop for %r' % (mesg, ))
retn = self.synack[1].get('retn')
valu = s_common.result(retn)
self.schedCoro(rxloop())
return valu
async def __aiter__(self):
try:
while not self.isfini:
for retn in await self.queue.slice():
if retn is None:
return
yield s_common.result(retn)
raise s_exc.LinkShutDown(mesg='Remote peer disconnected')
finally:
await self.fini()
