go中利用常驻内存特性,包变量实现简单的文件锁

go中包变量常驻内存,刚好最近有个场景,需要在读写文件的时候锁定该文件,不让其他人操作。此时,就可以利用go中包变量常驻内存的特点,非常简单实现这个功能;

包定义

package initial

import (

)



var FielLock map[string]int

func init() {

        FielLock = make(map[string]int)
}

  

具体实现(这里有细节需要注意,如果在加锁之后,有一些操作,比如文件读取错误导致程序意外退出,没有解锁成功。此刻如果不加超时限制的话,那么就无法解锁了,所以时间限制还是要有的)

func (this *BaseController) AddFileLock(path string) error {
        timestamp, err := strconv.Atoi(strconv.FormatInt(time.Now().Unix(), 10))
        if err != nil {
                return err
        }

        _, ok := initial.FielLock[path]

        //设置60秒超时
        if true == ok && initial.FielLock[path] == 1 && initial.FielLock[path+currentTime]+60 > timestamp {
                return errors.New("当前文件正在被使用,请勿操作")
        }

        initial.FielLock[path] = 1
        initial.FielLock[path+currentTime] = timestamp

        return nil
}

//解锁操作
func (this *BaseController) deleteFileLock(path string) {
        initial.FielLock[path] = 0
}