python,django,多线程,事务

Question

问题：如何操作使多线程达到普通方法调用的效果，使一次子线程失败，所有子线程及主线程保持一致,不去操作数据库

    @action(methods=['post'], detail=False)
    @transaction.atomic
    def uploadTest(self, request):
        data = {'code': 0, 'msg': '', 'data': []}
        connection.cursor().execute("INSERT INTO bby_test (id, name,age) VALUES (%s, %s, %s)", (1, 'name', 1))
        for i in range(0, 5):
            # 1.多线程调用 X
             pool.submit(Test,i)
            # 2.方法调用 √
             Test(i)
            # 问题：如何操作使多线程达到普通方法调用的效果，使一次子线程失败，所有子线程及主线程保持一致,不去操作数据库
        return JsonResponse(data)
 
def Test(i):
    print(i)
    if i % 2 == 0:
        a = 1/0
    connection.cursor().execute("INSERT INTO bby_test (id, name,age) VALUES (%s, %s, %s)", (2, 'name2', 2))
 

Answer 1

基于Monster 组和GPT的调写：
要让多线程达到普通方法调用的效果，使一次子线程失败，所有子线程及主线程保持一致，可以使用以下步骤：

import threading

lock = threading.Lock()  # 创建线程锁

@action(methods=['post'], detail=False)
def uploadTest(self, request):
    data = {'code': 0, 'msg': '', 'data': []}
    connection.cursor().execute("INSERT INTO bby_test (id, name,age) VALUES (%s, %s, %s)", (1, 'name', 1))
    threads = []
    for i in range(0, 5):
        t = threading.Thread(target=Test, args=(i,))
        threads.append(t)
        t.start()

    # 等待所有子线程结束
    for t in threads:
        t.join()

    return JsonResponse(data)

def Test(i):
    try:
        print(i)
        if i % 2 == 0:
            raise Exception("Exception in thread {}".format(i))
        with lock:
            # 线程锁保证多个线程不会同时修改数据库
            connection.cursor().execute("INSERT INTO bby_test (id, name,age) VALUES (%s, %s, %s)", (2, 'name2', 2))
    except Exception as e:
        # 异常捕获，保证子线程和主线程都能处理异常
        with lock:
            print(e)

在这个例子中，我们使用ThreadPoolExecutor创建一个具有最大工作线程数为5的线程池。然后，我们将上传数据的任务添加到线程池中，使用as_completed方法迭代所有线程，并捕获可能抛出的异常。如果有任何一个线程抛出异常，则向所有线程发送一个停止信号。当所有线程完成时，我们检查是否有任何线程抛出了异常。如果有，则将异常重新抛出，以便在主线程中处理。