oocana-python-executor 0.16.9__tar.gz → 0.16.11__tar.gz

{oocana_python_executor-0.16.9 → oocana_python_executor-0.16.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oocana-python-executor
-Version: 0.16.9
+Version: 0.16.11
 Summary: a client subscribe mqtt topic to execute oocana's block
 Author-Email: l1shen <lishen1635@gmail.com>, yleaf <11785335+leavesster@users.noreply.github.com>
 License: MIT

{oocana_python_executor-0.16.9 → oocana_python_executor-0.16.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "oocana-python-executor"
-version = "0.16.9"
+version = "0.16.11"
 authors = [
     { name = "l1shen", email = "lishen1635@gmail.com" },
     { name = "yleaf", email = "11785335+leavesster@users.noreply.github.com" },

{oocana_python_executor-0.16.9 → oocana_python_executor-0.16.11}/python_executor/context.py

@@ -59,7 +59,18 @@ def createContext(
     
     blockInfo = BlockInfo(**node_props)
 
-    ctx = Context(inputs, blockInfo, mainframe, store, output, session_dir, tmp_dir, package_name=package_name, pkg_dir=pkg_dir)
+    ctx = Context(
+        inputs=inputs,
+        inputs_def=inputs_def,
+        blockInfo=blockInfo,
+        mainframe=mainframe,
+        store=store,
+        outputs_def=output,
+        session_dir=session_dir,
+        tmp_dir=tmp_dir,
+        package_name=package_name,
+        pkg_dir=pkg_dir
+    )
     # 跟 executor 日志分开，避免有的库在 logger 里面使用 print，导致 hook 出现递归调用。
     block_logger = logging.getLogger(f"block {job_id}")
     ctx_handler = ContextHandler(ctx)

{oocana_python_executor-0.16.9 → oocana_python_executor-0.16.11}/python_executor/executor.py

@@ -91,7 +91,6 @@ async def run_executor(address: str, session_id: str, tmp_dir: str, package: str
     # 目前的 mqtt 库，在 subscribe 回调里 publish 消息会导致死锁无法工作，参考 https://github.com/eclipse/paho.mqtt.python/issues/527 或者 https://stackoverflow.com/a/36964192/4770006
     # 通过这种方式来绕过，所有需要 callback 后 publish message 的情况，都需要使用 future 类似方式来绕过。
     fs = queue.Queue()
-    loop = asyncio.get_event_loop()
 
     def execute_block(message):
         if not_current_session(message):
@@ -108,9 +107,7 @@ async def run_executor(address: str, session_id: str, tmp_dir: str, package: str
         job_set.add(job_id)
 
         nonlocal fs
-        f = loop.create_future()
-        fs.put(f)
-        f.set_result(message)
+        fs.put(message)
     
     def execute_service_block(message):
         if not_current_session(message):
@@ -119,10 +116,7 @@ async def run_executor(address: str, session_id: str, tmp_dir: str, package: str
         if not_current_job(message):
             return
 
-        nonlocal fs
-        f = loop.create_future()
-        fs.put(f)
-        f.set_result(message)
+        fs.put(message)
 
     def service_exit(message: ReportStatusPayload):
         service_hash = message.get("service_hash")
@@ -203,24 +197,21 @@ async def run_executor(address: str, session_id: str, tmp_dir: str, package: str
         mainframe.publish(run_action_topic(params), message)
 
     while True:
-        await asyncio.sleep(0.1)  # Avoid busy waiting
-        if not fs.empty():
-            f = fs.get()
-            message = await f
-            if message.get("service_executor") is not None:
-                service_hash = message.get("service_hash")
-                status = service_store.get(service_hash)
-                if status is None:
-                    asyncio.create_task(spawn_service(message, service_hash))
-                elif status == "running":
-                    run_service_block(message)
-                elif status == "launching":
-                    logger.info(f"service {service_hash} is launching, set message back to fs to wait next time")
-                    fs.put(f)
-            else:
-                if not_current_session(message):
-                    continue
-                run_block_in_new_thread(message, mainframe, session_dir=session_dir, tmp_dir=tmp_dir, package_name=package_name, pkg_dir=pkg_dir)
+        message = fs.get()
+        if message.get("service_executor") is not None:
+            service_hash = message.get("service_hash")
+            status = service_store.get(service_hash)
+            if status is None:
+                await spawn_service(message, service_hash)
+            elif status == "running":
+                run_service_block(message)
+            elif status == "launching":
+                logger.info(f"service {service_hash} is launching, set message back to fs to wait next time")
+                fs.put(message)  
+        else:
+            if not_current_session(message):
+                continue
+            run_block_in_new_thread(message, mainframe, session_dir=session_dir, tmp_dir=tmp_dir, package_name=package_name, pkg_dir=pkg_dir)
 
 def run_block_in_new_thread(message, mainframe: Mainframe, session_dir: str, tmp_dir: str, package_name: str, pkg_dir: str):