github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-07-02 04:00:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

change file storage, fix #412

Browse Source
This commit is contained in:
ffdfgdfg 2020-02-18 19:45:11 +08:00
parent 05c27b9365
commit 876d3ccc6f
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
lib/file/file.go
View File

@ -99,16 +99,28 @@ func (s *JsonDb) GetClient(id int) (c *Client, err error) {
return
}
var hostLock sync.Mutex
func (s *JsonDb) StoreHostToJsonFile() {
hostLock.Lock()
storeSyncMapToFile(s.Hosts, s.HostFilePath)
hostLock.Unlock()
}
var taskLock sync.Mutex
func (s *JsonDb) StoreTasksToJsonFile() {
taskLock.Lock()
storeSyncMapToFile(s.Tasks, s.TaskFilePath)
taskLock.Unlock()
}
var clientLock sync.Mutex
func (s *JsonDb) StoreClientsToJsonFile() {
clientLock.Lock()
storeSyncMapToFile(s.Clients, s.ClientFilePath)
clientLock.Unlock()
}
func (s *JsonDb) GetClientId() int32 {
@ -134,7 +146,8 @@ func loadSyncMapFromFile(filePath string, f func(value string)) {
}
func storeSyncMapToFile(m sync.Map, filePath string) {
file, err := os.Create(filePath)
file, err := os.Create(filePath + ".tmp")
// first create a temporary file to store
if err != nil {
panic(err)
}
@ -177,5 +190,7 @@ func storeSyncMapToFile(m sync.Map, filePath string) {
}
return true
})
file.Sync()
_ = file.Sync()
err = os.Rename(filePath+".tmp", filePath)
// replace the file, maybe provides atomic operation
}