12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697 |
- package callbacks
- import (
- "sort"
- "github.com/jinzhu/gorm"
- "github.com/jinzhu/gorm/clause"
- )
- // SelectAndOmitColumns get select and omit columns, select -> true, omit -> false
- func SelectAndOmitColumns(stmt *gorm.Statement) (map[string]bool, bool) {
- results := map[string]bool{}
- // select columns
- for _, column := range stmt.Selects {
- if field := stmt.Schema.LookUpField(column); field != nil {
- results[field.DBName] = true
- } else {
- results[column] = true
- }
- }
- // omit columns
- for _, omit := range stmt.Omits {
- if field := stmt.Schema.LookUpField(omit); field != nil {
- results[field.DBName] = false
- } else {
- results[omit] = false
- }
- }
- return results, len(stmt.Selects) > 0
- }
- // ConvertMapToValues convert map to values
- func ConvertMapToValues(stmt *gorm.Statement, mapValue map[string]interface{}) (values clause.Values) {
- columns := make([]string, 0, len(mapValue))
- selectColumns, restricted := SelectAndOmitColumns(stmt)
- var keys []string
- for k, _ := range mapValue {
- keys = append(keys, k)
- }
- sort.Strings(keys)
- for _, k := range keys {
- if field := stmt.Schema.LookUpField(k); field != nil {
- k = field.DBName
- }
- if v, ok := selectColumns[k]; (ok && v) || (!ok && !restricted) {
- columns = append(columns, k)
- values.Values[0] = append(values.Values[0], mapValue[k])
- }
- }
- return
- }
- // ConvertSliceOfMapToValues convert slice of map to values
- func ConvertSliceOfMapToValues(stmt *gorm.Statement, mapValues []map[string]interface{}) (values clause.Values) {
- var (
- columns = []string{}
- result = map[string][]interface{}{}
- selectColumns, restricted = SelectAndOmitColumns(stmt)
- )
- for idx, mapValue := range mapValues {
- for k, v := range mapValue {
- if field := stmt.Schema.LookUpField(k); field != nil {
- k = field.DBName
- }
- if _, ok := result[k]; !ok {
- if v, ok := selectColumns[k]; (ok && v) || (!ok && !restricted) {
- result[k] = make([]interface{}, len(mapValues))
- columns = append(columns, k)
- } else {
- continue
- }
- }
- result[k][idx] = v
- }
- }
- sort.Strings(columns)
- values.Values = make([][]interface{}, len(mapValues))
- for idx, column := range columns {
- for i, v := range result[column] {
- if i == 0 {
- values.Values[i] = make([]interface{}, len(columns))
- }
- values.Values[i][idx] = v
- }
- }
- return
- }
|