模块 liteyuki.core.manager

Copyright (C) 2020-2024 LiteyukiStudio. All Rights Reserved

@Time : 2024/7/27 上午11:12 @Author : snowykami @Email : snowykami@outlook.com @File : manager.py @Software: PyCharm

class ChannelDeliver

---

func __init__(self, active: Channel[Any], passive: Channel[Any], channel_deliver_active: Channel[Channel[Any]], channel_deliver_passive: Channel[tuple[str, dict]], publish: Channel[tuple[str, Any]])

源代码 或 在GitHub上查看

def __init__(self, active: Channel[Any], passive: Channel[Any], channel_deliver_active: Channel[Channel[Any]], channel_deliver_passive: Channel[tuple[str, dict]], publish: Channel[tuple[str, Any]]):
    self.active = active
    self.passive = passive
    self.channel_deliver_active = channel_deliver_active
    self.channel_deliver_passive = channel_deliver_passive
    self.publish = publish

class ProcessManager

---

func __init__(self, lifespan: Lifespan)

源代码 或 在GitHub上查看

def __init__(self, lifespan: 'Lifespan'):
    self.lifespan = lifespan
    self.targets: dict[str, tuple[Callable, tuple, dict]] = {}
    self.processes: dict[str, Process] = {}

---

func start_all(self)

说明: 对外启动方法，启动所有进程，创建asyncio task

源代码 或 在GitHub上查看

def start_all(self):
    for name in self.targets:
        logger.debug(f'Starting process {name}')
        threading.Thread(target=self._run_process, args=(name,), daemon=True).start()

---

func add_target(self, name: str, target: TARGET_FUNC, args: tuple = (), kwargs = None)

说明: 添加进程

参数:

• name: 进程名，用于获取和唯一标识
• target: 进程函数
• args: 进程函数参数
• kwargs: 进程函数关键字参数，通常会默认传入chan_active和chan_passive

源代码 或 在GitHub上查看

def add_target(self, name: str, target: TARGET_FUNC, args: tuple=(), kwargs=None):
    if kwargs is None:
        kwargs = {}
    chan_active: Channel = Channel(name=f'{name}-active')
    chan_passive: Channel = Channel(name=f'{name}-passive')
    channel_deliver = ChannelDeliver(active=chan_active, passive=chan_passive, channel_deliver_active=channel_deliver_active_channel, channel_deliver_passive=channel_deliver_passive_channel, publish=publish_channel)
    self.targets[name] = (_delivery_channel_wrapper, (target, channel_deliver, shared_memory, *args), kwargs)

---

func join_all(self)

源代码 或 在GitHub上查看

def join_all(self):
    for name, process in self.targets:
        process.join()

---

func terminate(self, name: str)

说明: 终止进程并从进程字典中删除

参数:

• name:

源代码 或 在GitHub上查看

def terminate(self, name: str):
    if name not in self.processes:
        logger.warning(f'Process {name} not found.')
        return
    process = self.processes[name]
    process.terminate()
    process.join(TIMEOUT)
    if process.is_alive():
        process.kill()
    logger.success(f'Process {name} terminated.')

---

func terminate_all(self)

源代码 或 在GitHub上查看

def terminate_all(self):
    for name in self.targets:
        self.terminate(name)

---

func is_process_alive(self, name: str) -> bool

说明: 检查进程是否存活

参数:

• name:

源代码 或 在GitHub上查看

def is_process_alive(self, name: str) -> bool:
    if name not in self.targets:
        logger.warning(f'Process {name} not found.')
    return self.processes[name].is_alive()

class _SubProcessManager

---

func __init__(self)

源代码 或 在GitHub上查看

def __init__(self):
    self.processes: dict[str, SubProcess] = {}

---

func add(self, name: str, func: ProcessFuncType, *args, **kwargs)

说明: 添加子进程

参数:

• func: 子进程函数
• name: 子进程名称
• args: 子进程函数参数
• kwargs: 子进程函数关键字参数

源代码 或 在GitHub上查看

def add(self, name: str, func: ProcessFuncType, *args, **kwargs):
    self.processes[name] = SubProcess(name, func, *args, **kwargs)

---

func start(self, name: str)

说明: 启动指定子进程

参数:

• name: 子进程名称

源代码 或 在GitHub上查看

def start(self, name: str):
    if name not in self.processes:
        raise KeyError(f'Process {name} not found.')
    self.processes[name].start()

---

func start_all(self)

说明: 启动所有子进程

源代码 或 在GitHub上查看

def start_all(self):
    for name, process in self.processes.items():
        process.start()
        logger.debug(f'Starting process {name}')

---

func terminate(self, name: str)

说明: 终止指定子进程

参数:

• name: 子进程名称

源代码 或 在GitHub上查看

def terminate(self, name: str):
    if name not in self.processes:
        raise KeyError(f'Process {name} not found.')
    self.processes[name].terminate()

---

func terminate_all(self)

说明: 终止所有子进程

源代码 或 在GitHub上查看

def terminate_all(self):
    for name, process in self.processes.items():
        process.terminate()
        logger.debug(f'Terminating process {name}')

---

func get_process(self, name: str) -> SubProcess | None

说明: 获取指定子进程

参数:

• name: 子进程名称

源代码 或 在GitHub上查看

def get_process(self, name: str) -> SubProcess | None:
    return self.processes.get(name, None)