feat: 重构任务处理逻辑以简化并发执行和结果处理

This commit is contained in:
晓丰 2025-05-30 22:48:40 +08:00
parent 2c4477aa51
commit 2f5febf7fb

34
main.py
View File

@ -358,21 +358,27 @@ def integrate_data_parallel():
if not tasks:
time.sleep(10)
continue
for payload, kitem in tasks:
proxname = proxiesdict.get(kitem['rn'])
print(proxname)
h = gettoken(proxname)
print(h)
get_searchInfo(kitem['keyword'], kitem['level'], h, proxiesdict)
futures = [
executor.submit(search_worker, payload, kitem, flag)
for payload, kitem in tasks
]
# 统计回滚
rollback = {0: [], 1: [], 2: []}
for fut in concurrent.futures.as_completed(futures):
ok, f_flag, payload, kitem, v_list = fut.result()
if not ok:
rollback[f_flag].append(payload)
continue
# futures = [
# executor.submit(search_worker, payload, kitem, flag)
# for payload, kitem in tasks
# ]
#
# # 统计回滚
# rollback = {0: [], 1: [], 2: []}
#
# for fut in concurrent.futures.as_completed(futures):
# ok, f_flag, payload, kitem, v_list = fut.result()
#
# if not ok:
# rollback[f_flag].append(payload)
# continue
for item in v_list:
if not v_list: