对Python多线程读写文件加锁的实例详解

最新推荐文章于 2024-04-19 11:08:56 发布

原创

最新推荐文章于 2024-04-19 11:08:56 发布 · 2.5k 阅读

本文详细介绍了Python中如何在多线程环境下对文件进行读写操作时使用文件锁和线程锁。通过fcntl模块的flock()函数实现文件锁，以及threading.Lock类来创建线程锁，确保了多线程访问文件的安全性。示例代码展示了加锁、解锁的过程，并解释了锁的工作原理和状态转换。

Python的多线程在io方面比单线程还是有优势，但是在多线程开发时，少不了对文件的读写操作。在管理多个线程对同一文件的读写操作时，就少不了文件锁了。

使用fcntl

在linux下，python的标准库有现成的文件锁，来自于fcntl模块。这个模块提供了unix系统fcntl()和ioctl()的接口。

对于文件锁的操作，主要需要使用 fcntl.flock(fd, operation)这个函数。

其中，参数 fd 表示文件描述符；参数 operation 指定要进行的锁操作，该参数的取值有如下几种：

LOCK_SH：表示要创建一个共享锁，在任意时间内，一个文件的共享锁可以被多个进程拥有

LOCK_EX：表示创建一个排他锁，在任意时间内，一个文件的排他锁只能被一个进程拥有

LOCK_UN：表示删除该进程创建的锁

LOCK_MAND：它主要是用于共享模式强制锁，它可以与 LOCK_READ 或者 LOCK_WRITE联合起来使用，从而表示是否允许并发的读操作或者并发的写操作

demo

import fcntl
import threading
import time
  
  
def writetoTxt(txtFile):
 id = threading.currentThread().getName()
 with open(txtFile, 'a') as f:
  fcntl.flock(f.fileno(), fcntl.LOCK_EX) #加锁
  print "{0} acquire lock".format(id)
  f.write("write from {0} \r\n".format(id))
  time.sleep(3)
 # 在with块外，文件关闭，自动解锁
 print "{0} exit".format(id)
  
  
for i in range(5):
 myThread = threading.Thread(target=writetoTxt, args=("test.txt",))
 myThread.start()

标签