suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2025-06-04 02:26:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 插入新的Schedulable时应传入use_batch参数 (#594)

Browse Source
This commit is contained in:
Azide 2024-07-13 00:38:57 +08:00 committed by GitHub
parent e089e87b84
commit c50243cc57
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nonebot_bison/scheduler/scheduler.py
View File

@ -43,7 +43,7 @@ class Scheduler:
self.scheduler_config_obj = self.scheduler_config() self.scheduler_config_obj = self.scheduler_config()
self.schedulable_list = [] self.schedulable_list = []
self.batch_platform_name_targets_cache: dict[str, list[Target]] = defaultdict(list) self.batch_platform_name_targets_cache = defaultdict(list)
for platform_name, target, use_batch in schedulables: for platform_name, target, use_batch in schedulables:
if use_batch: if use_batch:
self.batch_platform_name_targets_cache[platform_name].append(target) self.batch_platform_name_targets_cache[platform_name].append(target)
@ -130,12 +130,13 @@ class Scheduler:
def insert_new_schedulable(self, platform_name: str, target: Target): def insert_new_schedulable(self, platform_name: str, target: Target):
self.pre_weight_val += 1000 self.pre_weight_val += 1000
self.schedulable_list.append(Schedulable(platform_name, target, 1000)) new_schedulable = Schedulable(platform_name, target, 1000, platform_manager[platform_name].use_batch)
if platform_manager[platform_name].use_batch: if new_schedulable.use_batch:
self.batch_platform_name_targets_cache[platform_name].append(target) self.batch_platform_name_targets_cache[platform_name].append(target)
self._refresh_batch_api_target_cache() self._refresh_batch_api_target_cache()
self.schedulable_list.append(new_schedulable)
logger.info(f"insert [{platform_name}]{target} to Schduler({self.scheduler_config.name})") logger.info(f"insert [{platform_name}]{target} to Schduler({self.scheduler_config.name})")
def delete_schedulable(self, platform_name, target: Target): def delete_schedulable(self, platform_name, target: Target):