fix couchbase serialization (#552)

This commit is contained in:
Yaron Schneider 2020-12-07 09:43:46 -08:00 committed by GitHub
parent 9391dc7864
commit 10b3dc16fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 13 deletions

View File

@ -11,6 +11,7 @@ import (
"strconv"
"github.com/dapr/components-contrib/state"
"github.com/dapr/components-contrib/state/utils"
"github.com/dapr/dapr/pkg/logger"
jsoniter "github.com/json-iterator/go"
"gopkg.in/couchbase/gocb.v1"
@ -131,14 +132,7 @@ func (cbs *Couchbase) Set(req *state.SetRequest) error {
if err != nil {
return err
}
var value string
b, ok := req.Value.([]byte)
if ok {
value = string(b)
} else {
value, err = cbs.json.MarshalToString(req.Value)
}
value, err := utils.Marshal(req.Value, cbs.json.Marshal)
if err != nil {
return fmt.Errorf("couchbase error: failed to convert value %v", err)
}
@ -183,13 +177,9 @@ func (cbs *Couchbase) Get(req *state.GetRequest) (*state.GetResponse, error) {
return nil, fmt.Errorf("couchbase error: failed to get value for key %s - %v", req.Key, err)
}
value, err := cbs.json.Marshal(&data)
if err != nil {
return nil, fmt.Errorf("couchbase error: failed to convert value to byte[] - %v", err)
}
return &state.GetResponse{
Data: value,
Data: data.([]byte),
ETag: fmt.Sprintf("%d", cas),
}, nil
}