在Python中,可以使用multiprocessing
模塊中的Lock
類來實現多進程鎖。
下面是一個簡單的例子,展示了如何在多進程中使用鎖:
import multiprocessing
# 共享變量
counter = multiprocessing.Value('i', 0)
lock = multiprocessing.Lock()
def increment():
for _ in range(100000):
with lock:
counter.value += 1
def main():
processes = []
for _ in range(3):
process = multiprocessing.Process(target=increment)
process.start()
processes.append(process)
for process in processes:
process.join()
print(f"Counter value: {counter.value}")
if __name__ == '__main__':
main()
在上面的例子中,我們定義了一個共享變量counter
,它使用了multiprocessing.Value
來創建一個可被不同進程共享的整數值。然后我們創建了一個鎖lock
,用于在多個進程之間進行同步。
increment
函數是我們要在每個進程中運行的邏輯,它使用了with lock
語句來獲取鎖,然后對counter.value
進行自增操作。
在main
函數中,我們創建了3個進程,并將它們添加到processes
列表中。然后通過process.join()
等待所有進程執行完畢。
最后,我們打印出counter.value
的值,以驗證多進程對共享變量的修改是否正確。
注意,鎖的使用應該盡可能地減少,因為它會引入額外的開銷。只有在需要對共享資源進行互斥訪問時,才應該使用鎖。