You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package builderx
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/go-xorm/builder"
|
|
)
|
|
|
|
const dbTag = "db"
|
|
|
|
func NewEq(in interface{}) builder.Eq {
|
|
return builder.Eq(ToMap(in))
|
|
}
|
|
|
|
func NewGt(in interface{}) builder.Gt {
|
|
return builder.Gt(ToMap(in))
|
|
}
|
|
|
|
func ToMap(in interface{}) map[string]interface{} {
|
|
out := make(map[string]interface{})
|
|
v := reflect.ValueOf(in)
|
|
if v.Kind() == reflect.Ptr {
|
|
v = v.Elem()
|
|
}
|
|
// we only accept structs
|
|
if v.Kind() != reflect.Struct {
|
|
panic(fmt.Errorf("ToMap only accepts structs; got %T", v))
|
|
}
|
|
typ := v.Type()
|
|
for i := 0; i < v.NumField(); i++ {
|
|
// gets us a StructField
|
|
fi := typ.Field(i)
|
|
if tagv := fi.Tag.Get(dbTag); tagv != "" {
|
|
// set key of map to value in struct field
|
|
val := v.Field(i)
|
|
zero := reflect.Zero(val.Type()).Interface()
|
|
current := val.Interface()
|
|
|
|
if reflect.DeepEqual(current, zero) {
|
|
continue
|
|
}
|
|
out[tagv] = current
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
func FieldNames(in interface{}) []string {
|
|
out := make([]string, 0)
|
|
v := reflect.ValueOf(in)
|
|
if v.Kind() == reflect.Ptr {
|
|
v = v.Elem()
|
|
}
|
|
// we only accept structs
|
|
if v.Kind() != reflect.Struct {
|
|
panic(fmt.Errorf("ToMap only accepts structs; got %T", v))
|
|
}
|
|
typ := v.Type()
|
|
for i := 0; i < v.NumField(); i++ {
|
|
// gets us a StructField
|
|
fi := typ.Field(i)
|
|
if tagv := fi.Tag.Get(dbTag); tagv != "" {
|
|
out = append(out, fmt.Sprintf("`%v`", tagv))
|
|
} else {
|
|
out = append(out, fmt.Sprintf("`%v`", fi.Name))
|
|
}
|
|
}
|
|
return out
|
|
}
|