You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.2 KiB
91 lines
2.2 KiB
package main |
|
|
|
import ( |
|
"reflect" |
|
|
|
"github.com/ugorji/go/codec" |
|
"samhofi.us/x/keybase/v2/types/chat1" |
|
) |
|
|
|
// KvStorePutStruct marshals an interface to JSON and sends to kvstore |
|
func (b *bot) KVStorePutStruct(convIDstr chat1.ConvIDStr, v interface{}) error { |
|
// marshal the struct to JSON |
|
kvstoreDataString, err := encodeStructToJSONString(v) |
|
if err != nil { |
|
return err |
|
} |
|
// put the string in kvstore |
|
err = b.KVStorePut(string(convIDstr), getTypeName(v), kvstoreDataString) |
|
if err != nil { |
|
return err |
|
} |
|
return nil |
|
} |
|
|
|
// KVStoreGetStruct gets a string from kvstore and unmarshals the JSON to a struct |
|
func (b *bot) KVStoreGetStruct(convIDstr chat1.ConvIDStr, v interface{}) error { |
|
// get the string from kvstore |
|
result, err := b.KVStoreGet(string(convIDstr), getTypeName(v)) |
|
if err != nil { |
|
return err |
|
} |
|
// if there was no result just return and the struct is unmodified |
|
if result == "" { |
|
return nil |
|
} |
|
// unmarshal the string into JSON |
|
err = decodeJSONStringToStruct(v, result) |
|
if err != nil { |
|
return err |
|
} |
|
return nil |
|
} |
|
|
|
// KVStorePut puts a string into kvstore given a key and namespace |
|
func (b *bot) KVStorePut(namespace string, key string, value string) error { |
|
_, err := b.k.KVPut(&b.config.KVStoreTeam, namespace, key, value) |
|
if err != nil { |
|
return err |
|
} |
|
return nil |
|
} |
|
|
|
// KVStoreGet gets a string from kvstore given a key and namespace |
|
func (b *bot) KVStoreGet(namespace string, key string) (string, error) { |
|
kvResult, err := b.k.KVGet(&b.config.KVStoreTeam, namespace, key) |
|
if err != nil { |
|
return "", err |
|
} |
|
return kvResult.EntryValue, nil |
|
} |
|
|
|
// getTypeName returns the name of a type, regardless of if its a pointer or not |
|
func getTypeName(v interface{}) string { |
|
t := reflect.TypeOf(v) |
|
if t.Kind() == reflect.Ptr { |
|
return t.Elem().Name() |
|
} |
|
return t.Name() |
|
} |
|
|
|
func encodeStructToJSONString(v interface{}) (string, error) { |
|
jh := codecHandle() |
|
var bytes []byte |
|
err := codec.NewEncoderBytes(&bytes, jh).Encode(v) |
|
if err != nil { |
|
return "", err |
|
} |
|
result := string(bytes) |
|
return result, nil |
|
} |
|
|
|
func decodeJSONStringToStruct(v interface{}, src string) error { |
|
bytes := []byte(src) |
|
jh := codecHandle() |
|
return codec.NewDecoderBytes(bytes, jh).Decode(v) |
|
} |
|
|
|
func codecHandle() *codec.JsonHandle { |
|
var jh codec.JsonHandle |
|
return &jh |
|
}
|
|
|