forked from couchbase/cbft
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cfg_simple.go
99 lines (81 loc) · 1.96 KB
/
cfg_simple.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Copyright (c) 2014 Couchbase, Inc.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the
// License. You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an "AS
// IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
// express or implied. See the License for the specific language
// governing permissions and limitations under the License.
package main
import (
"encoding/json"
"io/ioutil"
"sync"
)
// Local-only, persisted (in a single file) implementation of Cfg.
type CfgSimple struct {
m sync.Mutex
path string
cfgMem *CfgMem
}
func NewCfgSimple(path string) *CfgSimple {
return &CfgSimple{
path: path,
cfgMem: NewCfgMem(),
}
}
func (c *CfgSimple) Get(key string, cas uint64) (
[]byte, uint64, error) {
c.m.Lock()
defer c.m.Unlock()
return c.cfgMem.Get(key, cas)
}
func (c *CfgSimple) Set(key string, val []byte, cas uint64) (
uint64, error) {
c.m.Lock()
defer c.m.Unlock()
cas, err := c.cfgMem.Set(key, val, cas)
if err != nil {
return 0, err
}
err = c.unlockedSave()
if err != nil {
return 0, err
}
return cas, err
}
func (c *CfgSimple) Del(key string, cas uint64) error {
c.m.Lock()
defer c.m.Unlock()
err := c.cfgMem.Del(key, cas)
if err != nil {
return err
}
return c.unlockedSave()
}
func (c *CfgSimple) Load() error {
c.m.Lock()
defer c.m.Unlock()
return c.unlockedLoad()
}
func (c *CfgSimple) unlockedLoad() error {
buf, err := ioutil.ReadFile(c.path)
if err != nil {
return err
}
cfgMem := &CfgMem{}
if err := json.Unmarshal(buf, cfgMem); err != nil {
return err
}
c.cfgMem = cfgMem
return nil
}
func (c *CfgSimple) unlockedSave() error {
buf, err := json.Marshal(c.cfgMem)
if err != nil {
return err
}
return ioutil.WriteFile(c.path, buf, 0600)
}