import{_ as s,c as i,o as a,a9 as n}from"./chunks/framework.C4_mTacX.js";const o=JSON.parse('{"title":"liteyuki.comm.channel","description":"","frontmatter":{"title":"liteyuki.comm.channel"},"headers":[],"relativePath":"dev/api/comm/channel.md","filePath":"zh/dev/api/comm/channel.md","lastUpdated":null}'),e={name:"dev/api/comm/channel.md"},l=n(`
liteyuki.comm.channel
本模块定义了一个通用的通道类,用于进程间通信
SYNC_ON_RECEIVE_FUNC
说明: 同步接收函数
类型: TypeAlias
默认值: Callable[[T], Any]
ASYNC_ON_RECEIVE_FUNC
说明: 异步接收函数
类型: TypeAlias
默认值: Callable[[T], Coroutine[Any, Any, Any]]
ON_RECEIVE_FUNC
说明: 接收函数
类型: TypeAlias
默认值: SYNC_ON_RECEIVE_FUNC | ASYNC_ON_RECEIVE_FUNC
SYNC_FILTER_FUNC
说明: 同步过滤函数
类型: TypeAlias
默认值: Callable[[T], bool]
ASYNC_FILTER_FUNC
说明: 异步过滤函数
类型: TypeAlias
默认值: Callable[[T], Coroutine[Any, Any, bool]]
FILTER_FUNC
说明: 过滤函数
类型: TypeAlias
默认值: SYNC_FILTER_FUNC | ASYNC_FILTER_FUNC
Channel(Generic[T])
__init__(self, name: str, type_check: Optional[bool] = None)
说明: 初始化通道
参数:
- name: 通道ID
- type_check: 是否开启类型检查, 若为空,则传入泛型默认开启,否则默认关闭
def __init__(self, name: str, type_check: Optional[bool]=None):
self.conn_send, self.conn_recv = Pipe()
self._conn_send_inner, self._conn_recv_inner = Pipe()
self._closed = False
self._on_main_receive_func_ids: list[int] = []
self._on_sub_receive_func_ids: list[int] = []
self.name: str = name
self.is_receive_loop_running = False
if type_check is None:
type_check = self._get_generic_type() is not None
elif type_check:
if self._get_generic_type() is None:
raise TypeError('Type hint is required for enforcing type check.')
self.type_check = type_check
if name in _channel:
raise ValueError(f'Channel {name} already exists')
if IS_MAIN_PROCESS:
if name in _channel:
raise ValueError(f'Channel {name} already exists')
_channel[name] = self
logger.debug(f'Channel {name} initialized in main process')
else:
logger.debug(f'Channel {name} initialized in sub process, should manually set in main process')
send(self, data: T)
说明: 发送数据,发送函数为同步函数,没有异步的必要
参数:
- data (T): 数据
def send(self, data: T):
if self.type_check:
_type = self._get_generic_type()
if _type is not None and (not self._validate_structure(data, _type)):
raise TypeError(f'Data must be an instance of {_type}, {type(data)} found')
if self._closed:
raise RuntimeError('Cannot send to a closed channel_')
self.conn_send.send(data)
receive(self) -> T
说明: 同步接收数据,会阻塞线程
返回: T: 数据
def receive(self) -> T:
if self._closed:
raise RuntimeError('Cannot receive from a closed channel_')
while True:
data = self.conn_recv.recv()
return data
async_receive(self) -> T
说明: 异步接收数据,会挂起等待
返回: T: 数据
async def async_receive(self) -> T:
loop = asyncio.get_running_loop()
data = await loop.run_in_executor(None, self.receive)
return data
on_receive(self, filter_func: Optional[FILTER_FUNC] = None) -> Callable[[Callable[[T], Any]], Callable[[T], Any]]
说明: 接收数据并执行函数
参数:
- filter_func (
Optional
[FILTER_FUNC
], optional): 过滤函数. Defaults to None.
返回: Callable[[Callable[[T], Any]], Callable[[T], Any]]: 装饰器
def on_receive(self, filter_func: Optional[FILTER_FUNC]=None) -> Callable[[Callable[[T], Any]], Callable[[T], Any]]:
if not IS_MAIN_PROCESS:
raise RuntimeError('on_receive can only be used in main process')
def decorator(func: Callable[[T], Any]) -> Callable[[T], Any]:
global _func_id
async def wrapper(data: T) -> Any:
if filter_func is not None:
if is_coroutine_callable(filter_func):
if not await filter_func(data):
return
elif not filter_func(data):
return
if is_coroutine_callable(func):
return await func(data)
else:
return func(data)
_callback_funcs[_func_id] = wrapper
if IS_MAIN_PROCESS:
self._on_main_receive_func_ids.append(_func_id)
else:
self._on_sub_receive_func_ids.append(_func_id)
_func_id += 1
return func
return decorator
active_channel
说明: 子进程可用的主动和被动通道
类型: Channel
默认值: Channel(name='active_channel')
passive_channel
说明: 被动通道
类型: Channel
默认值: Channel(name='passive_channel')
publish_channel
说明: 通道传递通道,主进程创建单例,子进程初始化时实例化
类型: Channel[tuple[str, dict[str, Any]]]
默认值: Channel(name='publish_channel')
channel_deliver_active_channel
说明: 通道传递通道,主进程创建单例,子进程初始化时实例化
类型: Channel[Channel[Any]]
默认值: NO_DEFAULT
channel_deliver_passive_channel
说明: 被动通道传递通道
类型: Channel[tuple[str, dict[str, Any]]]
默认值: NO_DEFAULT
set_channel(name: str, channel: Channel)
说明: 设置通道实例
参数:
def set_channel(name: str, channel: 'Channel'):
if not isinstance(channel, Channel):
raise TypeError(f'channel_ must be an instance of Channel, {type(channel)} found')
if IS_MAIN_PROCESS:
if name in _channel:
raise ValueError(f'Channel {name} already exists')
_channel[name] = channel
else:
channel_deliver_passive_channel.send(('set_channel', {'name': name, 'channel_': channel}))
set_channels(channels: dict[str, Channel])
说明: 设置通道实例
参数:
def set_channels(channels: dict[str, 'Channel']):
for name, channel in channels.items():
set_channel(name, channel)
get_channel(name: str) -> Channel
说明: 获取通道实例
参数:
- name (
str
): 通道名称
返回: Channel
: 通道实例
def get_channel(name: str) -> 'Channel':
if IS_MAIN_PROCESS:
return _channel[name]
else:
recv_chan = Channel[Channel[Any]]('recv_chan')
channel_deliver_passive_channel.send(('get_channel', {'name': name, 'recv_chan': recv_chan}))
return recv_chan.receive()
get_channels() -> dict[str, Channel]
说明: 获取通道实例们
def get_channels() -> dict[str, 'Channel']:
if IS_MAIN_PROCESS:
return _channel
else:
recv_chan = Channel[dict[str, Channel[Any]]]('recv_chan')
channel_deliver_passive_channel.send(('get_channels', {'recv_chan': recv_chan}))
return recv_chan.receive()
channel_deliver_active_channel
说明: 主动通道传递通道
默认值: Channel(name='channel_deliver_active_channel')
channel_deliver_passive_channel
说明: 被动通道传递通道
默认值: Channel(name='channel_deliver_passive_channel')
@channel_deliver_passive_channel.on_receive(filter_func=lambda data: data[0] == 'set_channel')
on_set_channel(data: tuple[str, dict[str, Any]])
@channel_deliver_passive_channel.on_receive(filter_func=lambda data: data[0] == 'set_channel')
def on_set_channel(data: tuple[str, dict[str, Any]]):
name, channel = (data[1]['name'], data[1]['channel_'])
set_channel(name, channel)
@channel_deliver_passive_channel.on_receive(filter_func=lambda data: data[0] == 'get_channel')
on_get_channel(data: tuple[str, dict[str, Any]])
@channel_deliver_passive_channel.on_receive(filter_func=lambda data: data[0] == 'get_channel')
def on_get_channel(data: tuple[str, dict[str, Any]]):
name, recv_chan = (data[1]['name'], data[1]['recv_chan'])
recv_chan.send(get_channel(name))
@channel_deliver_passive_channel.on_receive(filter_func=lambda data: data[0] == 'get_channels')
on_get_channels(data: tuple[str, dict[str, Any]])
@channel_deliver_passive_channel.on_receive(filter_func=lambda data: data[0] == 'get_channels')
def on_get_channels(data: tuple[str, dict[str, Any]]):
recv_chan = data[1]['recv_chan']
recv_chan.send(get_channels())