fcntl与文件锁

最近要加一个文件同步的服务,以daemon的方式部署在每台机器上,其他服务进程如果需要访问本机磁盘文件时,如果访问不到需要告诉文件同步服务去远端服务器把文件同步过来,同步文件的服务我们直接对rsync做了一层包装,文件同步存在一个中间状态,即文件已经存在但内容还没同步完成,这个中间状态时其他的进程不能来读取文件,这里我们使用了建议性文件锁来协调进程间文件的读取。

文件锁(File Lock)是一种在特定的时间内只允许一个 进程进行访问文件的机制,通过使用文件锁,可以使得多进程访问文件更加安全

一个锁结构如下

struct flock {
    short l_type;        /*锁类型*/
    short l_whence;      /*锁的起始位置*/
    off_t l_start;       /*锁区域开始位置的字节偏移量*/  
    off_t l_len;         /*锁长度*/
    pid_t l_pid;         /*持有锁的进程*/

}

l_type的三个取值分别是

  • F_RDLCK 共享读锁
  • F_WRLCK 独占性写锁
  • F_UNLCK 解锁一个区域

l_whence有三个取值

  • SEEK_SET 文件头
  • SEEK_CUR 当前位置
  • SEEK_END 文件尾

在linux操作系统中,我们可以使用fcntl支持的三种命令F_GETLK,F_SETLK和F_SETLKW,通过第三个参数指向一个锁结构的指针(flockptr),用于操作文件描述符实现文件锁。

int fcntl(int fd, int cmd, ... /* args */ );
  • F_GETLK 判断flockptr指向的锁是否被另外一把锁排斥,如果已经存在一把锁,这现有的flockptr指向的锁将被阻止创建,并且flockptr指向的锁结构内容将被该锁替换,如果不存在则flockptr指向的锁类型会被设置为 F_UNLOCK。
  • F_SETLK 设置flockptr所指向的锁,如果设置的是读锁或者写锁被兼容性规则阻止的话,fcntl会立即返回错误,如果设置的是F_UNLCK,则可以清除文件锁。
  • F_SETLKW 是F_SETLK的阻塞版本。

需要注意的是F_GETLK测试能否建立一把锁后,使用F_SETLK加锁仍有可能失败,因为两者不是一个原子操作,需要处理F_SETLK是fcntl的返回值。

理解了上面的API后,使用lua对fcntl做了一层封装,代码在lua-fie-lock, 效果如下

[lua-file-lock]$ lua sample/set_lock.lua 
1539930102   : locked
1539930107   : unlocked

[lua-file-lock]$ lua sample/try_lock.lua 
1539930103  : locked, retry after 1s
1539930105  : locked, retry after 1s
1539930107  : lockable

wechat
微信扫一扫,订阅我的博客动态^_^