Fix worker break issue (#10)

SkyAPM · Aug 6, 2024 · 4aa9c59 · 4aa9c59
1 parent f330cdc
commit 4aa9c59
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/servers/simple/server.py b/servers/simple/server.py
@@ -82,7 +82,7 @@ async def feedRawData(self, request, context):
             # It should not be called
             return Empty()
 
-        uris = [str(uri.name) for uri in request.unrecognizedUris]
+        uris = [str(uri.name) for uri in request.unrecognizedUris if uri and uri.name]
         service = str(request.service)
 
         # This is an experimental mechanism to avoid identifying non-restful uris unnecessarily.

diff --git a/servers/simple/worker.py b/servers/simple/worker.py
@@ -52,5 +52,7 @@ def run_worker(uri_main_queue, shared_results_object):
             # increment here
             counter += 1
             print('-================-')
+        except Exception as e:
+            print(f"catch an unexpected error occurred: {e}")
         except queue.Empty:  # TODO Consider queue full
             pass