python线程运行完通知

原创

mob64ca12f2c96c 2024-08-16 07:29:07 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12f2c96c的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 线程运行完通知

在 Python 编程中，线程是执行并发操作的基本单位。多线程编程可以显著提高程序的性能和响应能力。然而，当我们启动多个线程时，往往需要在某个线程运行完后进行通知，比如更新 UI 或者打开某个资源。在本篇文章中，我们将介绍如何在 Python 中实现线程的运行完通知，并提供相应的代码示例。

线程基本概念

线程是程序执行的最小单位，它代表一个独立的执行路径。Python 中的 threading 模块为我们提供了一种简单的方式来创建和管理线程。以下是一个简单的线程创建示例：

import threading
import time

def worker():
    print("Worker thread is running...")
    time.sleep(2)
    print("Worker thread has finished.")

# 创建并启动线程
t = threading.Thread(target=worker)
t.start()

# 等待线程完成
t.join()
print("Main thread is done.")

在上面的代码中，我们定义了一个 worker 函数作为线程的运行目标。t.join() 方法确保主线程在 worker 线程结束前不会退出。

使用事件进行通知

为了在一个线程结束后通知其他线程或主线程，我们可以使用 threading.Event 类。事件对象可以用于线程之间的通信。

以下是一个使用事件机制进行线程通知的示例：

import threading
import time

# 创建一个事件对象
event = threading.Event()

def worker():
    print("Worker thread is running...")
    time.sleep(2)
    print("Worker thread has finished.")
    event.set()  # 通知主线程

# 创建并启动线程
t = threading.Thread(target=worker)
t.start()

# 等待通知
event.wait()
print("Main thread has been notified.")

在上面的代码中，主线程通过调用 event.wait() 方法进行等待，而工作线程在完成任务后通过 event.set() 方法通知主线程，主线程随即被唤醒。

使用回调函数实现通知

除了使用事件，我们还可以通过回调函数来实现线程运行完通知。回调函数是一个在某个操作完成后被调用的函数。以下是使用回调函数的一个示例：

import threading
import time

def worker(callback):
    print("Worker thread is running...")
    time.sleep(2)
    print("Worker thread has finished.")
    callback()  # 调用回调函数

def thread_done():
    print("Main thread has been notified via callback.")

# 创建并启动线程
t = threading.Thread(target=worker, args=(thread_done,))
t.start()

# 主线程可以继续执行其他任务
t.join()

在这个例子中，主线程定义了一个回调函数 thread_done，该函数会在工作线程完成时被调用。通过将回调函数作为参数传递给工作线程，我们实现了通知的效果。

使用条件变量

条件变量允许线程以某种条件的方式进行等待，我们可以利用条件变量来实现线程的协作。

import threading
import time

# 创建一个条件变量
condition = threading.Condition()
data = []

def producer():
    global data
    time.sleep(2)
    with condition:
        data.append('item')
        print("Producer has produced an item.")
        condition.notify()  # 通知消费者线程

def consumer():
    with condition:
        print("Consumer is waiting for an item...")
        condition.wait()  # 等待通知
        print("Consumer has consumed:", data.pop())

# 创建并启动生产者和消费者线程
prod_thread = threading.Thread(target=producer)
cons_thread = threading.Thread(target=consumer)

cons_thread.start()
prod_thread.start()

prod_thread.join()
cons_thread.join()

在这个例子中，生产者线程在生产一个项目后会通知消费者线程。消费者线程在没有项目时会被阻塞，直到它接收到通知。

旅行图：线程的生命周期

在多线程编程中，线程的生命周期也是一个重要的概念。以下是一个使用 mermaid 的旅行图，它展示了线程的生命周期流程：

journey
    title 线程的生命周期
    section 创建
      主线程发起线程 : 5: 所有线程
      创建工作线程 : 5: 所有线程
    section 运行
      工作线程开始执行 : 3: 工作线程
      工作线程完成 : 3: 主线程
    section 结束
      工作线程通知主线程 : 5: 所有线程
      主线程继续执行 : 5: 所有线程