forked from orofarne/gopnik-couchbase-kv
/
couchbasekv.go
119 lines (102 loc) · 2.48 KB
/
couchbasekv.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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package couchbasekv
import (
"fmt"
"strings"
"plugins"
"github.com/couchbaselabs/go-couchbase"
"github.com/op/go-logging"
json "github.com/orofarne/strict-json"
)
var log = logging.MustGetLogger("global")
type couchbaseKVConf struct {
Addrs []string
Pool string
Bucket string
Expire int
Retries int
}
type CouchbaseKV struct {
config couchbaseKVConf
bucket *couchbase.Bucket
}
type CouchbaseKVFactory struct {
}
func (self *CouchbaseKVFactory) Name() string {
return "CouchbaseKVPlugin"
}
func (self *CouchbaseKVFactory) New(cfg json.RawMessage) (interface{}, error) {
var res = new(CouchbaseKV)
err := res.Configure(cfg)
if err != nil {
return nil, err
}
return res, nil
}
func init() {
plugins.DefaultPluginStore.AddPlugin(new(CouchbaseKVFactory))
}
func (self *CouchbaseKV) Configure(cfg json.RawMessage) error {
// Unmarshal config
err := json.Unmarshal(cfg, &self.config)
if err != nil {
return err
}
// Defaults
if self.config.Pool == "" {
self.config.Pool = "default"
}
// Connect
var errs []error
for _, addr := range self.config.Addrs {
self.bucket, err = couchbase.GetBucket(addr, self.config.Pool, self.config.Bucket)
if err == nil {
return nil
} else {
errs = append(errs, err)
}
}
return fmt.Errorf("Failed to connect to couchbase: %v", errs)
}
func (self *CouchbaseKV) Get(key string) (data []byte, err error) {
log.Debug("Request data by key '%v' from couchbase...", key)
for i := 0; i < self.config.Retries+1; i++ {
if i != 0 {
log.Debug("Retry request tile by key '%v' from couchbase...", key)
}
data, err = self.bucket.GetRaw(key)
if err == nil {
return
}
if err != nil && strings.Contains(err.Error(), "KEY_ENOENT") {
log.Debug("Key '%v' not found", key)
return nil, nil
}
}
return
}
func (self *CouchbaseKV) Set(key string, value []byte) (err error) {
log.Debug("Save data by key '%v' to couchbase", key)
for i := 0; i < self.config.Retries+1; i++ {
if i != 0 {
log.Debug("Retry save data by key '%v' to couchbase", key)
}
err = self.bucket.SetRaw(key, self.config.Expire, value)
if err == nil {
return
}
}
return
}
func (self *CouchbaseKV) Delete(key string) (err error) {
log.Debug("Delete data by key '%v' from couchbase", key)
for i := 0; i < self.config.Retries+1; i++ {
if i != 0 {
log.Debug("Retry delete data by key '%v' from couchbase", key)
}
err = self.bucket.Delete(key)
if err == nil || strings.Contains(err.Error(), "KEY_ENOENT") {
return nil
}
}
return
}