Created
February 22, 2022 08:14
-
-
Save rrmerugu/17ec6553760513e593b90a1c4bdb299b to your computer and use it in GitHub Desktop.
gremlin query with web sockets
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import asyncio | |
import websockets | |
class WebsocketHandler(object): | |
def __init__(self): | |
self.conn = None | |
async def connect(self, url): | |
print("Connecting") | |
self.conn = await websockets.connect(url) | |
print("Connected") | |
async def send(self, msg): | |
await self.conn.send(msg) | |
async def recv(self): | |
return await self.conn.recv() | |
async def close(self): | |
await self.conn.close() | |
async def main(): | |
handler = WebsocketHandler() | |
await handler.connect('ws://192.168.0.10:8182/gremlin') | |
message = b'!application/vnd.gremlin-v3.0+json{"requestId": {"@type": "g:UUID", "@value": "16f19ea9-3d61-45b5-b823-7f67199ff1b2"}, "processor": "traversal", "op": "bytecode", "args": {"gremlin": {"@type": "g:Bytecode", "@value": {"step": [["V", "98528"], ["valueMap", true]]}}, "aliases": {"g": "g"}}}' | |
await handler.send(message) | |
# await handler.send('world') | |
a = await handler.recv() | |
print('response is ',a) | |
await handler.close() | |
loop = asyncio.get_event_loop() | |
loop.run_until_complete(main()) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import asyncio | |
import websockets | |
class WebsocketHandler(object): | |
def __init__(self): | |
self.conn = None | |
self._loop = asyncio.get_event_loop() | |
def connect(self, url): | |
print("Connecting") | |
self.conn = self._loop.run_until_complete( | |
websockets.connect(url) | |
) | |
print("Connected") | |
def send(self, msg): | |
return self._loop.run_until_complete( | |
self.conn.send(msg) | |
) | |
def recv(self): | |
return self._loop.run_until_complete( | |
# asyncio.wait[ | |
self.conn.recv() | |
# ] | |
) | |
def close(self): | |
self._loop.run_until_complete( | |
self.conn.close() | |
) | |
def main(): | |
handler = WebsocketHandler() | |
handler.connect('ws://192.168.0.10:8182/gremlin') | |
message = b'!application/vnd.gremlin-v3.0+json{"requestId": {"@type": "g:UUID", "@value": "16f19ea9-3d61-45b5-b823-7f67199ff1b2"}, "processor": "traversal", "op": "bytecode", "args": {"gremlin": {"@type": "g:Bytecode", "@value": {"step": [["V", "98528"], ["valueMap", true]]}}, "aliases": {"g": "g"}}}' | |
handler.send(message) | |
# await handler.send('world') | |
a = handler.recv() | |
print('response is ', a) | |
handler.close() | |
if __name__ == '__main__': | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment