Python使用文件锁实现进程间同步功能【基于fcntl模块】


Posted in Python onOctober 16, 2017

本文实例讲述了Python使用文件锁实现进程间同步功能。分享给大家供大家参考,具体如下:

简介

在实际应用中,会出现这种应用场景:希望shell下执行的脚本对某些竞争资源提供保护,避免出现冲突。本文将通过fcntl模块的文件整体上锁机制来实现这种进程间同步功能。

fcntl系统函数介绍

Linux系统提供了文件整体上锁(flock)和更细粒度的记录上锁(fcntl)功能,底层功能均可由fcntl函数实现。

首先来了解记录上锁。记录上锁是读写锁的一种扩展类型,它可用于有亲缘关系或无亲缘关系的进程间共享某个文件的读与写。被锁住的文件通过其描述字访问,执行上锁操作的函数是fcntl。这种类型的锁在内核中维护,其宿主标识为fcntl调用进程的进程ID。这意味着这些锁用于不同进程间的上锁,而不是同一进程内不同线程间的上锁。

fcntl记录上锁即可用于读也可用于写,对于文件的任意字节,最多只能存在一种类型的锁(读锁或写锁)。而且,一个给定字节可以有多个读写锁,但只能有一个写入锁。

对于一个打开着某个文件的给定进程来说,当它关闭该文件的任何一个描述字或者终止时,与该文件关联的所有锁都被删除。锁不能通过fork由子进程继承。

NAME
    fcntl - manipulate file descriptor
SYNOPSIS
    #include <unistd.h>
    #include <fcntl.h>
    int fcntl(int fd, int cmd, ... /* arg */ );
DESCRIPTION
    fcntl() performs one of the operations described below on the open file descriptor fd. The operation is determined by cmd.
    fcntl() can take an optional third argument. Whether or not this argument is required is determined by cmd. The required argument type
    is indicated in parentheses after each cmd name (in most cases, the required type is int, and we identify the argument using the name
    arg), or void is specified if the argument is not required.
    Advisory record locking
    Linux implements traditional ("process-associated") UNIX record locks, as standardized by POSIX. For a Linux-specific alternative with
    better semantics, see the discussion of open file description locks below.
    F_SETLK, F_SETLKW, and F_GETLK are used to acquire, release, and test for the existence of record locks (also known as byte-range, file-
    segment, or file-region locks). The third argument, lock, is a pointer to a structure that has at least the following fields (in
    unspecified order).
      struct flock {
        ...
        short l_type;  /* Type of lock: F_RDLCK,
                  F_WRLCK, F_UNLCK */
        short l_whence; /* How to interpret l_start:
                  SEEK_SET, SEEK_CUR, SEEK_END */
        off_t l_start;  /* Starting offset for lock */
        off_t l_len;   /* Number of bytes to lock */
        pid_t l_pid;   /* PID of process blocking our lock
                  (set by F_GETLK and F_OFD_GETLK) */
        ...
      };

其次,文件上锁源自Berkeley的Unix实现支持给整个文件上锁或解锁的文件上锁(file locking),但没有给文件内的字节范围上锁或解锁的能力。

fcntl模块及基于文件锁的同步功能。

Python fcntl模块提供了基于文件描述符的文件和I/O控制功能。它是Unix系统调用fcntl()和ioctl()的接口。因此,我们可以基于文件锁来提供进程同步的功能。

import fcntl
class Lock(object):
  def __init__(self, file_name):
    self.file_name = file_name
    self.handle = open(file_name, 'w')
  def lock(self):
    fcntl.flock(self.handle, fcntl.LOCK_EX)
  def unlock(self):
    fcntl.flock(self.handle, fcntl.LOCK_UN)
  def __del__(self):
    try:
      self.handle.close()
    except:
      pass

应用

我们做一个简单的场景应用:需要从指定的服务器上下载软件版本到/exports/images目录下,因为这个脚本可以在多用户环境执行。我们不希望下载出现冲突,并仅在该目录下保留一份指定的软件版本。下面是基于文件锁的参考实现:

if __name__ == "__main__":
  parser = OptionParser()
  group = OptionGroup(parser, "FTP download tool", "Download build from ftp server")
  group.add_option("--server", type="string", help="FTP server's IP address")
  group.add_option("--username", type="string", help="User name")
  group.add_option("--password", type="string", help="User's password")
  group.add_option("--buildpath", type="string", help="Build path in the ftp server")
  group.add_option("--buildname", type="string", help="Build name to be downloaded")
  parser.add_option_group(group)
  (options, args) = parser.parse_args()
  local_dir = "/exports/images"
  lock_file = "/var/tmp/flock.txt"
  flock = Lock(lock_file)
  flock.lock()
  if os.path.isfile(os.path.join(local_dir, options.buildname)):
    log.info("build exists, nothing needs to be done")
    log.info("Download completed")
    flock.unlock()
    exit(0)
  log.info("start to download build " + options.buildname)
  t = paramiko.Transport((options.server, 22))
  t.connect(username=options.username, password=options.password)
  sftp = paramiko.SFTPClient.from_transport(t)
  sftp.get(os.path.join(options.buildpath, options.buildname),
       os.path.join(local_dir, options.buildname))
  sftp.close()
  t.close()
  log.info("Download completed")
  flock.unlock()

希望本文所述对大家Python程序设计有所帮助。

Python 相关文章推荐
python实现自动重启本程序的方法
Jul 09 Python
Python使用统计函数绘制简单图形实例代码
May 15 Python
Python pip替换为阿里源的方法步骤
Jul 02 Python
Pycharm远程调试原理及具体配置详解
Aug 08 Python
Django项目主urls导入应用中views的红线问题解决
Aug 10 Python
python中web框架的自定义创建
Sep 08 Python
Python 中如何实现参数化测试的方法示例
Dec 10 Python
Python SSL证书验证问题解决方案
Jan 13 Python
python内打印变量之%和f的实例
Feb 19 Python
pytorch:model.train和model.eval用法及区别详解
Feb 20 Python
Python安装第三方库攻略(pip和Anaconda)
Oct 15 Python
python 实现两个变量值进行交换的n种操作
Jun 02 Python
python利用paramiko连接远程服务器执行命令的方法
Oct 16 #Python
基于使用paramiko执行远程linux主机命令(详解)
Oct 16 #Python
python中文件变化监控示例(watchdog)
Oct 16 #Python
python中import reload __import__的区别详解
Oct 16 #Python
使用Python操作excel文件的实例代码
Oct 15 #Python
python出现&quot;IndentationError: unexpected indent&quot;错误解决办法
Oct 15 #Python
python 二分查找和快速排序实例详解
Oct 13 #Python
You might like
php面向对象全攻略 (五) 封装性
2009/09/30 PHP
PHP数据类型的总结分析
2013/06/13 PHP
PHP 使用pcntl和libevent 实现Timer功能
2013/10/27 PHP
PHP获取时间排除周六、周日的两个方法
2014/06/30 PHP
php邮件发送的两种方式
2020/04/28 PHP
硬盘浏览程序,保存成网页格式便可使用
2006/12/03 Javascript
JavaScript中链式调用之研习
2011/04/07 Javascript
window.showModalDialog()返回值的学习心得总结
2014/01/07 Javascript
JQuery限制复选框checkbox可选中个数的方法
2015/04/20 Javascript
jquery实现点击label的同时触发文本框点击事件的方法
2015/06/05 Javascript
原生js实现模拟滚动条
2015/06/15 Javascript
AngularJS基础教程之简单介绍
2015/09/27 Javascript
js实现div在页面拖动效果
2016/05/04 Javascript
深入理解Javascript中的自执行匿名函数
2016/06/03 Javascript
javascript replace()第二个参数为函数时的参数用法
2016/12/26 Javascript
Vue组件开发之LeanCloud带图形校验码的短信发送功能
2017/11/07 Javascript
微信小程序实现获取自己所处位置的经纬度坐标功能示例
2017/11/30 Javascript
js实现选项卡效果
2020/03/07 Javascript
python判断字符串编码的简单实现方法(使用chardet)
2016/07/01 Python
Django教程笔记之中间件middleware详解
2018/08/01 Python
Python实现的IP端口扫描工具类示例
2019/02/15 Python
python向字符串中添加元素的实例方法
2019/06/28 Python
python判断两个序列的成员是否一样的实例代码
2020/03/01 Python
JupyterNotebook 输出窗口的显示效果调整方法
2020/04/13 Python
浅谈Pycharm的项目文件名是红色的原因及解决方式
2020/06/01 Python
python3中TQDM库安装及使用详解
2020/11/18 Python
全球知名提供各类营养保健品的零售商:Vitamin Shoppe
2016/10/09 全球购物
新西兰演唱会和体育门票网站:Ticketmaster新西兰
2017/10/07 全球购物
金鑫耀Java笔试题
2014/09/06 面试题
文艺晚会策划方案
2014/06/11 职场文书
升学宴演讲稿
2014/09/01 职场文书
优秀班集体事迹材料
2014/12/25 职场文书
高中生思想道德自我评价
2015/03/09 职场文书
楚门的世界观后感
2015/06/03 职场文书
省级三好学生主要事迹材料
2015/11/03 职场文书
Java 数组内置函数toArray详解
2021/06/28 Java/Android