36 lines
923 B
Python
36 lines
923 B
Python
|
import os
|
||
|
|
from itertools import cycle
|
||
|
|
import threading
|
||
|
|
from concurrent.futures import ThreadPoolExecutor
|
||
|
|
|
||
|
|
# 定义三个 API Key 的循环器
|
||
|
|
api_keys = cycle([1, 2, 3])
|
||
|
|
# 使用锁保证线程安全
|
||
|
|
lock = threading.Lock()
|
||
|
|
|
||
|
|
def test():
|
||
|
|
# 使用锁保护对 api_keys 的访问
|
||
|
|
with lock:
|
||
|
|
key = next(api_keys)
|
||
|
|
print(key)
|
||
|
|
return key
|
||
|
|
|
||
|
|
def main():
|
||
|
|
results = []
|
||
|
|
num_calls = 100 # 模拟 100 次并发调用
|
||
|
|
# 使用 ThreadPoolExecutor 模拟高并发
|
||
|
|
with ThreadPoolExecutor(max_workers=20) as executor:
|
||
|
|
futures = [executor.submit(test) for _ in range(num_calls)]
|
||
|
|
# 收集所有返回的结果
|
||
|
|
for future in futures:
|
||
|
|
results.append(future.result())
|
||
|
|
|
||
|
|
# 统计各个 API Key 被使用的次数
|
||
|
|
counts = {1: 0, 2: 0, 3: 0}
|
||
|
|
for key in results:
|
||
|
|
counts[key] += 1
|
||
|
|
print("调用分布:", counts)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|