在Python中,可以使用threading
模塊來實現多線程操作,使用threading.Lock()
來創建一個鎖對象。
下面是一個示例代碼,演示了如何使用多線程寫文件并加鎖:
import threading
def write_file(lock, filename, content):
# 獲取鎖
lock.acquire()
try:
with open(filename, 'a') as f:
f.write(content + '\n')
finally:
# 釋放鎖
lock.release()
if __name__ == '__main__':
# 創建一個鎖對象
lock = threading.Lock()
# 創建多個線程,并啟動
threads = []
for i in range(5):
t = threading.Thread(target=write_file, args=(lock, 'test.txt', f'Thread {i}'))
threads.append(t)
t.start()
# 等待所有線程結束
for t in threads:
t.join()
在上面的代碼中,write_file()
函數用于寫入文件,其中lock.acquire()
用于獲取鎖,lock.release()
用于釋放鎖。在每個線程中,先獲取鎖,執行寫文件操作,然后釋放鎖,以確保每個線程在寫文件時都是互斥的。
注意,在使用鎖時,需要確保在獲取鎖后的代碼塊中,如果發生異常也能正常釋放鎖,通常使用try...finally
來確保鎖的釋放。