526互联

python 多线程使用中关于daemon和join的用途

发布时间 2023-04-11 18:38:48作者: Marathon-Davis

我们在遇到 IO 耗时的时候，一般可以考虑使用到 python 的多线程操作，有的时候，我们主线程不必等待子线程运行结束，有的时候主线程需要等待子线程运行结束再执行主线程的逻辑，这里就涉及到 python 中的 daemon 和 join 的用途了。

在创建子线程时，考虑是否让子线程作为后台守护线程运行，有以下情况：

  • 如果我们想让子线程作为后台线程运行，即不必让主线程阻塞时，通常我们会设置 daemon=True
  • 如果我们想让子线程运行结束，就需要让主线程阻塞等待中，启动子线程时就需要设置 t.join()

下面我们看看一些示例：
daemon=True

from threading import Thread
import time
import datetime

def foo():
    print("foo start:", datetime.datetime.now())
    time.sleep(10)
    print("foo end:", datetime.datetime.now())

if __name__ == "__main__":
    print(datetime.datetime.now())
    t = Thread(target=foo, daemon=True)
    # t.setDaemon(True) # 与上同
    t.start()
    print(datetime.datetime.now())

    time.sleep(5)
    print("main thread end")
    print(datetime.datetime.now())

---
2023-04-11 18:20:02.611632
foo start: 2023-04-11 18:20:02.611632
2023-04-11 18:20:02.611632
main thread
2023-04-11 18:20:07.615428
---

示例比较简单，其实就是主线程只等待 5s 时间，而子线程中，需要运行 10s，显然在设置了后台守护线程后，主线程到时间就结束了，也就不管子线程的运行了。

join()

from threading import Thread
import time
import datetime

def foo():
    print("foo start:", datetime.datetime.now())
    time.sleep(10)
    print("foo end:", datetime.datetime.now())

if __name__ == "__main__":
    print(datetime.datetime.now())
    t = Thread(target=foo)
    t.start()
    t.join()
    print(datetime.datetime.now())

    time.sleep(5)
    print("main thread end")
    print(datetime.datetime.now())

---
2023-04-11 18:30:28.177553
foo start: 2023-04-11 18:30:28.178554
foo end: 2023-04-11 18:30:38.182320
2023-04-11 18:30:38.182320
main thread end
2023-04-11 18:30:43.184286
---

设置了 join 后，主线程首先阻塞住，等待子线程的运行结束，然后主线程的执行逻辑中还需要再休眠5s时间，故最后的时间是比刚开始时多了15s的时间。