104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
|
import asyncio
|
||
|
|
import time
|
||
|
|
|
||
|
|
from zhenxun.services.log import logger
|
||
|
|
|
||
|
|
|
||
|
|
class Signal:
|
||
|
|
def __init__(self):
|
||
|
|
self._slots = [] #绑定的槽函数列表
|
||
|
|
self._onceSlots = [] #只触发一次的槽函数列表
|
||
|
|
|
||
|
|
def connect(self, slot, priority=0):
|
||
|
|
if callable(slot) and not any(s[0] == slot for s in self._slots):
|
||
|
|
self._slots.append((slot, priority))
|
||
|
|
self._slots.sort(key=lambda x: -x[1])
|
||
|
|
|
||
|
|
def connectOnce(self, slot, priority=0):
|
||
|
|
if callable(slot) and not any(s[0] == slot for s in self._onceSlots):
|
||
|
|
self._onceSlots.append((slot, priority))
|
||
|
|
self._onceSlots.sort(key=lambda x: -x[1])
|
||
|
|
|
||
|
|
def disconnect(self, slot):
|
||
|
|
self._slots = [s for s in self._slots if s[0] != slot]
|
||
|
|
self._onceSlots = [s for s in self._onceSlots if s[0] != slot]
|
||
|
|
|
||
|
|
async def emit(self, *args, **kwargs):
|
||
|
|
slots = list(self._slots)
|
||
|
|
onceSlots = list(self._onceSlots)
|
||
|
|
self._onceSlots.clear()
|
||
|
|
|
||
|
|
for slot, _ in slots + onceSlots:
|
||
|
|
startTime = time.time()
|
||
|
|
try:
|
||
|
|
if asyncio.iscoroutinefunction(slot):
|
||
|
|
await slot(*args, **kwargs)
|
||
|
|
else:
|
||
|
|
slot(*args, **kwargs)
|
||
|
|
duration = (time.time() - startTime) * 1000
|
||
|
|
logger.debug(f"事件槽 {slot.__name__} 执行完成，耗时 {duration:.2f} ms")
|
||
|
|
except Exception as e:
|
||
|
|
logger.warning(f"事件槽 {slot.__name__} 触发异常: {e}")
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class FarmEventManager:
|
||
|
|
def __init__(self):
|
||
|
|
|
||
|
|
self.m_beforePlant = Signal()
|
||
|
|
"""播种前信号
|
||
|
|
|
||
|
|
Args:
|
||
|
|
uid (str): 用户Uid
|
||
|
|
name (str): 播种种子名称
|
||
|
|
num (int): 播种数量
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.m_afterPlant = Signal()
|
||
|
|
"""播种后信号 每块地播种都会触发该信号
|
||
|
|
|
||
|
|
Args:
|
||
|
|
uid (str): 用户Uid
|
||
|
|
name (str): 播种种子名称
|
||
|
|
soilIndex (int): 播种地块索引 从1开始
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.m_beforeHarvest = Signal()
|
||
|
|
"""收获前信号
|
||
|
|
|
||
|
|
Args:
|
||
|
|
uid (str): 用户Uid
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.m_afterHarvest = Signal()
|
||
|
|
"""收获后信号 每块地收获都会触发该信号
|
||
|
|
|
||
|
|
Args:
|
||
|
|
uid (str): 用户Uid
|
||
|
|
name (str): 收获作物名称
|
||
|
|
num (int): 收获数量
|
||
|
|
soilIndex (int): 收获地块索引 从1开始
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.m_beforeEradicate = Signal()
|
||
|
|
"""铲除前信号
|
||
|
|
|
||
|
|
Args:
|
||
|
|
uid (str): 用户Uid
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.m_afterEradicate = Signal()
|
||
|
|
"""铲除后信号 每块地铲除都会触发该信号
|
||
|
|
|
||
|
|
Args:
|
||
|
|
uid (str): 用户Uid
|
||
|
|
soilIndex (index): 铲除地块索引 从1开始
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.m_beforeExpand = Signal()
|
||
|
|
self.m_afterExpand = Signal()
|
||
|
|
self.m_beforeSteal = Signal()
|
||
|
|
self.m_afterSteal = Signal()
|
||
|
|
|
||
|
|
g_pEventManager = FarmEventManager()
|