generated from gitpod-samples/template-golang-cli
-
Notifications
You must be signed in to change notification settings - Fork 0
/
crud.go
91 lines (77 loc) · 1.91 KB
/
crud.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
package store
import (
"fmt"
badger "github.com/dgraph-io/badger/v3"
"github.com/rs/zerolog/log"
)
func AddEntryToDb(db *badger.DB, clientname string, m []byte) error {
err := db.Update(func(txn *badger.Txn) error {
err := txn.Set([]byte(clientname), m)
return err
})
return err
}
func DeleteEntryToDb(db *badger.DB, keys []string) []error {
var errorlist []error
for _, v := range keys {
err := db.Update(func(txn *badger.Txn) error {
err := txn.Delete([]byte(v))
return err
},
)
errorlist = append(errorlist, err)
}
return errorlist
}
func ViewEntryDb(db *badger.DB, key string) ([]byte, error) {
var val []byte
err := db.View(func(txn *badger.Txn) error {
item, err := txn.Get([]byte(key))
if err != nil {
return err
}
val, err = item.ValueCopy(nil)
if err != nil {
return err
}
/*
var studentDecode auth.ClientData
d := gob.NewDecoder(bytes.NewReader(val))
if err := d.Decode(&studentDecode); err != nil {
panic(err)
}
log.Printf("Decoded Struct from badger : name [%s] age [%s]\n", studentDecode.ClientID, studentDecode.RedirectURI) */
return nil
})
return val, err
}
func PrefixDb(db *badger.DB, key string) (map[string][]byte, error) {
clientList := make(map[string][]byte)
err := db.View(func(txn *badger.Txn) error {
it := txn.NewIterator(badger.DefaultIteratorOptions)
defer it.Close()
prefix := []byte(key)
for it.Seek(prefix); it.ValidForPrefix(prefix); it.Next() {
item := it.Item()
k := item.Key()
err := item.Value(func(v []byte) error {
fmt.Printf("key=%s, value=%s\n", k, v)
return nil
})
if err != nil {
return err
}
keyc := string(item.KeyCopy(nil))
val, err := item.ValueCopy(nil)
if err != nil {
log.Error().Err(err).Msg("Unable to copy db item")
}
clientList[keyc] = val
}
return nil
})
if err != nil {
log.Error().Err(err).Msg("Unmarshal body")
}
return clientList, nil
}