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
}

  

posted @ 2019-01-06 12:18  queqp  阅读(1115)  评论(0编辑  收藏  举报