import multiprocessing
import time

from huaxin_client import l2_client_test, l1_subscript_codes_manager


def run():
    codes_sh, codes_sz = l1_subscript_codes_manager.get_codes()
    codes = [x.decode() for x in codes_sh]
    codes.extend([x.decode() for x in codes_sz])
    codes.sort()
    cpu_count = 16
    page_size = int(len(codes) / cpu_count) + 1

    for i in range(cpu_count):
        process = multiprocessing.Process(target=l2_client_test.run,
                                          args=(codes[i * page_size:(i + 1) * page_size],))
        process.start()


if __name__ == "__main__":
    run()
    while True:
        time.sleep(2)