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
}