在 linux 环境下用 Python 进行项目开发过程中经常会遇到多个进程对同一个文件进行读写问题,而此时就要对文件进行加锁控制,在 Python 的 linux 版本下有个 fcntl 模块可以方便的对文件进行加、解锁控制。
import fcntl
file_path = "/home/ubuntu/aaa.json"
f = open(file_path, 'w')
fcntl.flock(f.fileno(), fcntl.LOCK_EX) # 加锁,其它进程对文件操作则不能成功
f.write("something")
fcntl.flock(f.fileno(), fcntl.LOCK_UN) # 解锁
f.close()
fcntl.flock(f.fileno(), operation) operation 的操作包括以下选项:
-
fcntl.LOCK_EX
排他锁: 除加锁进程外其他进程没有对已加锁文件读写访问权限
-
fcntl.LOCK_UN
解锁: 对加锁文件进行解锁
-
fcntl.LOCK_SH
共享锁: 所有进程都没有写权限,即使加锁进程也没有,但所有进程都有读权限
-
fcntl.LOCK_NB
非阻塞锁: 如果指定此参数,函数不能获得文件锁就立即返回,否则,函数会等待获得文件锁。
LOCK_NB可以同LOCK_SH或LOCK_NB进行按位或(|)运算操作。
fcnt.flock(f.fileno(),fcntl.LOCK_EX|fcntl.LOCK_NB)