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

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

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
}