多进程的目的是并发执行程序,提高程序执行效率,但有时候我们想要在某一时间,或者满足某一条件时,只有一个进程在执行,就需要使用Lock锁机制。

示例:

import multiprocessing
import time


def task1(lock):
    with lock:  # with上下文语句使用锁,会自动释放锁
        n = 5
        while n > 1:
            print(f"{time.strftime('%H:%M:%S')} task1 输出信息")
            time.sleep(1)
            n -= 1


def task2(lock):
    lock.acquire()
    n = 5
    while n > 1:
        print(f"{time.strftime('%H:%M:%S')} task2 输出信息")
        time.sleep(1)
        n -= 1
    lock.release()


def task3(lock):
    lock.acquire()
    n = 5
    while n > 1:
        print(f"{time.strftime('%H:%M:%S')} task3 输出信息")
        time.sleep(1)
        n -= 1
    lock.release()


if __name__ == "__main__":
    lock = multiprocessing.Lock()
    p1 = multiprocessing.Process(target=task1, args=(lock,))
    p2 = multiprocessing.Process(target=task2, args=(lock,))
    p3 = multiprocessing.Process(target=task3, args=(lock,))
    p1.start()
    p2.start()
    p3.start()

输出:

20:13:22 task1 输出信息
20:13:23 task1 输出信息
20:13:24 task1 输出信息
20:13:25 task1 输出信息
20:13:26 task2 输出信息
20:13:27 task2 输出信息
20:13:28 task2 输出信息
20:13:29 task2 输出信息
20:13:30 task3 输出信息
20:13:31 task3 输出信息
20:13:32 task3 输出信息
20:13:33 task3 输出信息

说明:从结果上看,由于锁的机制,同一时刻只有一个进程在执行。

使用lock = multiprocess.Lock()可以得到一个锁的实例,在上面的代码中,可以使用with语句来使用锁,也可以在要执行的代码块前使用lock.acquire()方法,在执行完后再释放lock.release()锁,需要注意的是lock.acquire()后面的语句不能阻塞,否则会发生死锁的情况