serde_json.go

package pgsql

import (
	"database/sql"
	"errors"
	"github.com/gogo/protobuf/jsonpb"
	"github.com/gogo/protobuf/proto"
	"strings"
)

type SerdeJsonFunc func (input proto.Message, output proto.Message, callback func(jsonString string) *sql.Row) error

func SerdeJson(marshaller *jsonpb.Marshaler, unmarshaller *jsonpb.Unmarshaler) SerdeJsonFunc {
	return func (input proto.Message, output proto.Message, callback func(jsonString string) *sql.Row) error {
		jsonString, err := marshaller.MarshalToString(input)
		if err != nil {
			return err
		}

		var data sql.NullString
		if err := callback(jsonString).Scan(&data); err != nil {
			return err
		}

		if !data.Valid {
			return errors.New("no result returned")
		}

		if err := unmarshaller.Unmarshal(strings.NewReader(data.String), output); err != nil {
			return err
		}

		return nil
	}
}