Python使用多线程并获取函数返回值

 没有返回值的代码:

check_code = self._creat_check(3)
check_id = self._select_check_id(check_code)

one_thread = threading.Thread(target=self._rob, kwargs={'check_id': check_id})
two_thread = threading.Thread(target=self._rob, kwargs={'check_id': check_id})
one_thread.start()
two_thread.start()

可以接收返回值的代码:

class MyTread(threading.Thread):
    def __init__(self, func, args=()):
        super(MyTread, self).__init__()
        self.func = func
        self.args = args

    def run(self):
        self.result = self.func(*self.args)

    def get(self):
        try:
            return self.result  # 如果子线程不使用join方法,此处可能会报没有self.result的错误
        except Exception:
            return None



 # 抢单:只生成一个订单,2人同时抢,断言一个成功,另一个失败(测试就剩一个检查单时是否加锁)
    def test_05(self):
        # 创建2个线程,同时抢单
        check_code = self._creat_check(3)
        check_id = self._select_check_id(check_code)
        li = []
        for i in range(1, 3):
            if i == 1:
                t = MyTread(self._rob, args=(check_id,))
            if i == 2:
                t = MyTread(self._rob2, args=(check_id,))
            li.append(t)
            t.start()

        code_li = []
        for t in li:
            t.join()
            code_li.append(t.result.get('code'))
            print(t.result)
        print(code_li)
        if (code_li[0] == 500 and code_li[1] == 0) or (code_li[1] == 500 and code_li[0] == 0):
            pytest.assume(1 == 1)
        else:
            pytest.assume(1 > 1, f'两人同时抢单断言失败,返回码:{code_li}')

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值