6a60650668
We can't return an error in case a field is added to the UDT,
otherwise existing code would break by simply altering the UDT in the
database. For extra fields at the end of the UDT put nulls to be in
line with gocql, but also python-driver and java-driver.
In gocql it was fixed in d2ed1bb74f
79 lines
1.8 KiB
Go
79 lines
1.8 KiB
Go
// Copyright (C) 2017 ScyllaDB
|
|
// Use of this source code is governed by a ALv2-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package gocqlx
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/gocql/gocql"
|
|
"github.com/scylladb/go-reflectx"
|
|
)
|
|
|
|
// UDT is a marker interface that needs to be embedded in a struct if you want
|
|
// to marshal or unmarshal it as a User Defined Type.
|
|
type UDT interface {
|
|
udt()
|
|
}
|
|
|
|
var (
|
|
_ gocql.UDTMarshaler = udt{}
|
|
_ gocql.UDTUnmarshaler = udt{}
|
|
)
|
|
|
|
type udt struct {
|
|
field map[string]reflect.Value
|
|
value reflect.Value
|
|
unsafe bool
|
|
}
|
|
|
|
func makeUDT(value reflect.Value, mapper *reflectx.Mapper, unsafe bool) udt {
|
|
return udt{
|
|
value: value,
|
|
field: mapper.FieldMap(value),
|
|
unsafe: unsafe,
|
|
}
|
|
}
|
|
|
|
func (u udt) MarshalUDT(name string, info gocql.TypeInfo) ([]byte, error) {
|
|
value, ok := u.field[name]
|
|
if ok {
|
|
return gocql.Marshal(info, value.Interface())
|
|
}
|
|
if u.unsafe {
|
|
return nil, nil
|
|
}
|
|
return nil, fmt.Errorf("missing name %q in %s", name, u.value.Type())
|
|
}
|
|
|
|
func (u udt) UnmarshalUDT(name string, info gocql.TypeInfo, data []byte) error {
|
|
value, ok := u.field[name]
|
|
if ok {
|
|
return gocql.Unmarshal(info, data, value.Addr().Interface())
|
|
}
|
|
if u.unsafe {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("missing name %q in %s", name, u.value.Type())
|
|
}
|
|
|
|
// udtWrapValue adds UDT wrapper if needed.
|
|
func udtWrapValue(value reflect.Value, mapper *reflectx.Mapper, unsafe bool) interface{} {
|
|
if value.Type().Implements(autoUDTInterface) {
|
|
return makeUDT(value, mapper, unsafe)
|
|
}
|
|
return value.Interface()
|
|
}
|
|
|
|
// udtWrapSlice adds UDT wrapper if needed.
|
|
func udtWrapSlice(mapper *reflectx.Mapper, unsafe bool, v []interface{}) []interface{} {
|
|
for i := range v {
|
|
if _, ok := v[i].(UDT); ok {
|
|
v[i] = makeUDT(reflect.ValueOf(v[i]), mapper, unsafe)
|
|
}
|
|
}
|
|
return v
|
|
}
|