sem/5g_drone_ROS2
1
0
Fork 0
Code Issues 1 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

try with consumer

Browse Source
This commit is contained in:
Sem van der Hoeven
2023-05-22 16:36:15 +02:00
parent 9f2f47aac4
commit 5740e14fac
1 changed files with 22 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/api_communication/api_communication/api_listener.py
View File

@@ -35,35 +35,36 @@ class ApiListener(Node):
async def handle_api(self):
self.get_logger().info('Starting API')
self.server = await websockets.serve(self.api_handler, '0.0.0.0', 9001)
self.websocket = websockets.connect('ws://100.100.100.106:9001')
self.get_logger().info('API started')
await self.server.wait_closed()
async for message in self.websocket:
self.get_logger().info(f"Received message: {message}")
await self.websocket.send(message)
async def handle_message_receive(self,websocket):
self.get_logger().info(f"Received message: {self.last_message}")
self.last_message = await websocket.recv()
# async def handle_message_receive(self,websocket):
# self.last_message = await websocket.recv()
def message_received_callback(self):
self.get_logger().info(f"Received message callback: {self.last_message}")
self.message_queue.append(self.last_message)
self.checking_for_message = False
# def message_received_callback(self):
# self.get_logger().info(f"Received message callback: {self.last_message}")
# self.message_queue.append(self.last_message)
# self.checking_for_message = False
# def handle_message(self, message):
# deserialized_msg = json.loads(message)
async def api_handler(self, websocket):
try:
while True:
if not self.checking_for_message:
self.get_logger().info('Waiting for message')
self.checking_for_message = True
task = asyncio.create_task(self.handle_message_receive(websocket))
task.add_done_callback(self.message_received_callback)
if len(self.message_queue) > 0:
websocket.send(self.message_queue.pop(0))
# async def api_handler(self, websocket):
# try:
# while True:
# if not self.checking_for_message:
# self.get_logger().info('Waiting for message')
# self.checking_for_message = True
# task = asyncio.create_task(self.handle_message_receive(websocket))
# task.add_done_callback(self.message_received_callback)
# if len(self.message_queue) > 0:
# websocket.send(self.message_queue.pop(0))
except websockets.exceptions.ConnectionClosed:
self.get_logger().info('Connection closed')
# except websockets.exceptions.ConnectionClosed:
# self.get_logger().info('Connection closed')
def main(args=None):