95 lines
1.8 KiB
Go
95 lines
1.8 KiB
Go
package orm
|
|
|
|
import (
|
|
"reflect"
|
|
|
|
"github.com/go-pg/pg/types"
|
|
)
|
|
|
|
const (
|
|
PrimaryKeyFlag = uint8(1) << iota
|
|
ForeignKeyFlag
|
|
NotNullFlag
|
|
UniqueFlag
|
|
ArrayFlag
|
|
customTypeFlag
|
|
)
|
|
|
|
type Field struct {
|
|
Type reflect.Type
|
|
|
|
GoName string // struct field name, e.g. Id
|
|
SQLName string // SQL name, .e.g. id
|
|
Column types.Q // escaped SQL name
|
|
SQLType string
|
|
Index []int
|
|
Default types.Q
|
|
|
|
flags uint8
|
|
|
|
append types.AppenderFunc
|
|
scan types.ScannerFunc
|
|
|
|
isZero func(reflect.Value) bool
|
|
}
|
|
|
|
func (f *Field) Copy() *Field {
|
|
copy := *f
|
|
copy.Index = copy.Index[:len(f.Index):len(f.Index)]
|
|
return ©
|
|
}
|
|
|
|
func (f *Field) SetFlag(flag uint8) {
|
|
f.flags |= flag
|
|
}
|
|
|
|
func (f *Field) HasFlag(flag uint8) bool {
|
|
return f.flags&flag != 0
|
|
}
|
|
|
|
func (f *Field) Value(strct reflect.Value) reflect.Value {
|
|
return strct.FieldByIndex(f.Index)
|
|
}
|
|
|
|
func (f *Field) IsZero(strct reflect.Value) bool {
|
|
fv := f.Value(strct)
|
|
return f.isZero(fv)
|
|
}
|
|
|
|
func (f *Field) OmitZero(strct reflect.Value) bool {
|
|
return !f.HasFlag(NotNullFlag) && f.isZero(f.Value(strct))
|
|
}
|
|
|
|
func (f *Field) AppendValue(b []byte, strct reflect.Value, quote int) []byte {
|
|
fv := f.Value(strct)
|
|
if !f.HasFlag(NotNullFlag) && f.isZero(fv) {
|
|
return types.AppendNull(b, quote)
|
|
}
|
|
return f.append(b, fv, quote)
|
|
}
|
|
|
|
func (f *Field) ScanValue(strct reflect.Value, b []byte) error {
|
|
fv := fieldByIndex(strct, f.Index)
|
|
return f.scan(fv, b)
|
|
}
|
|
|
|
type Method struct {
|
|
Index int
|
|
|
|
flags int8
|
|
|
|
appender func([]byte, reflect.Value, int) []byte
|
|
}
|
|
|
|
func (m *Method) Has(flag int8) bool {
|
|
return m.flags&flag != 0
|
|
}
|
|
|
|
func (m *Method) Value(strct reflect.Value) reflect.Value {
|
|
return strct.Method(m.Index).Call(nil)[0]
|
|
}
|
|
|
|
func (m *Method) AppendValue(dst []byte, strct reflect.Value, quote int) []byte {
|
|
mv := m.Value(strct)
|
|
return m.appender(dst, mv, quote)
|
|
}
|