|
@@ -2,10 +2,13 @@ package gorm
|
|
|
|
|
|
import (
|
|
|
"database/sql"
|
|
|
+ "database/sql/driver"
|
|
|
"errors"
|
|
|
"fmt"
|
|
|
"regexp"
|
|
|
+ "strconv"
|
|
|
"strings"
|
|
|
+ "time"
|
|
|
|
|
|
"reflect"
|
|
|
)
|
|
@@ -406,11 +409,15 @@ func (scope *Scope) SelectAttrs() []string {
|
|
|
return *scope.selectAttrs
|
|
|
}
|
|
|
|
|
|
-// OmitAttrs return omited attributes
|
|
|
+// OmitAttrs return omitted attributes
|
|
|
func (scope *Scope) OmitAttrs() []string {
|
|
|
return scope.Search.omits
|
|
|
}
|
|
|
|
|
|
+////////////////////////////////////////////////////////////////////////////////
|
|
|
+// Private Methods For *gorm.Scope
|
|
|
+////////////////////////////////////////////////////////////////////////////////
|
|
|
+
|
|
|
func (scope *Scope) scan(rows *sql.Rows, columns []string, fieldsMap map[string]*Field) {
|
|
|
var values = make([]interface{}, len(columns))
|
|
|
var ignored interface{}
|
|
@@ -441,3 +448,715 @@ func (scope *Scope) scan(rows *sql.Rows, columns []string, fieldsMap map[string]
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+func (scope *Scope) primaryCondition(value interface{}) string {
|
|
|
+ return fmt.Sprintf("(%v = %v)", scope.Quote(scope.PrimaryKey()), value)
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) buildWhereCondition(clause map[string]interface{}) (str string) {
|
|
|
+ switch value := clause["query"].(type) {
|
|
|
+ case string:
|
|
|
+ // if string is number
|
|
|
+ if regexp.MustCompile("^\\s*\\d+\\s*$").MatchString(value) {
|
|
|
+ return scope.primaryCondition(scope.AddToVars(value))
|
|
|
+ } else if value != "" {
|
|
|
+ str = fmt.Sprintf("(%v)", value)
|
|
|
+ }
|
|
|
+ case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64, sql.NullInt64:
|
|
|
+ return scope.primaryCondition(scope.AddToVars(value))
|
|
|
+ case []int, []int8, []int16, []int32, []int64, []uint, []uint8, []uint16, []uint32, []uint64, []string, []interface{}:
|
|
|
+ str = fmt.Sprintf("(%v IN (?))", scope.Quote(scope.PrimaryKey()))
|
|
|
+ clause["args"] = []interface{}{value}
|
|
|
+ case map[string]interface{}:
|
|
|
+ var sqls []string
|
|
|
+ for key, value := range value {
|
|
|
+ if value != nil {
|
|
|
+ sqls = append(sqls, fmt.Sprintf("(%v = %v)", scope.Quote(key), scope.AddToVars(value)))
|
|
|
+ } else {
|
|
|
+ sqls = append(sqls, fmt.Sprintf("(%v IS NULL)", scope.Quote(key)))
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return strings.Join(sqls, " AND ")
|
|
|
+ case interface{}:
|
|
|
+ var sqls []string
|
|
|
+ for _, field := range scope.New(value).Fields() {
|
|
|
+ if !field.IsIgnored && !field.IsBlank {
|
|
|
+ sqls = append(sqls, fmt.Sprintf("(%v = %v)", scope.Quote(field.DBName), scope.AddToVars(field.Field.Interface())))
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return strings.Join(sqls, " AND ")
|
|
|
+ }
|
|
|
+
|
|
|
+ args := clause["args"].([]interface{})
|
|
|
+ for _, arg := range args {
|
|
|
+ switch reflect.ValueOf(arg).Kind() {
|
|
|
+ case reflect.Slice: // For where("id in (?)", []int64{1,2})
|
|
|
+ if bytes, ok := arg.([]byte); ok {
|
|
|
+ str = strings.Replace(str, "?", scope.AddToVars(bytes), 1)
|
|
|
+ } else if values := reflect.ValueOf(arg); values.Len() > 0 {
|
|
|
+ var tempMarks []string
|
|
|
+ for i := 0; i < values.Len(); i++ {
|
|
|
+ tempMarks = append(tempMarks, scope.AddToVars(values.Index(i).Interface()))
|
|
|
+ }
|
|
|
+ str = strings.Replace(str, "?", strings.Join(tempMarks, ","), 1)
|
|
|
+ } else {
|
|
|
+ str = strings.Replace(str, "?", scope.AddToVars(Expr("NULL")), 1)
|
|
|
+ }
|
|
|
+ default:
|
|
|
+ if valuer, ok := interface{}(arg).(driver.Valuer); ok {
|
|
|
+ arg, _ = valuer.Value()
|
|
|
+ }
|
|
|
+
|
|
|
+ str = strings.Replace(str, "?", scope.AddToVars(arg), 1)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) buildNotCondition(clause map[string]interface{}) (str string) {
|
|
|
+ var notEqualSQL string
|
|
|
+ var primaryKey = scope.PrimaryKey()
|
|
|
+
|
|
|
+ switch value := clause["query"].(type) {
|
|
|
+ case string:
|
|
|
+ // is number
|
|
|
+ if regexp.MustCompile("^\\s*\\d+\\s*$").MatchString(value) {
|
|
|
+ id, _ := strconv.Atoi(value)
|
|
|
+ return fmt.Sprintf("(%v <> %v)", scope.Quote(primaryKey), id)
|
|
|
+ } else if regexp.MustCompile("(?i) (=|<>|>|<|LIKE|IS|IN) ").MatchString(value) {
|
|
|
+ str = fmt.Sprintf(" NOT (%v) ", value)
|
|
|
+ notEqualSQL = fmt.Sprintf("NOT (%v)", value)
|
|
|
+ } else {
|
|
|
+ str = fmt.Sprintf("(%v NOT IN (?))", scope.Quote(value))
|
|
|
+ notEqualSQL = fmt.Sprintf("(%v <> ?)", scope.Quote(value))
|
|
|
+ }
|
|
|
+ case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64, sql.NullInt64:
|
|
|
+ return fmt.Sprintf("(%v <> %v)", scope.Quote(primaryKey), value)
|
|
|
+ case []int, []int8, []int16, []int32, []int64, []uint, []uint8, []uint16, []uint32, []uint64, []string:
|
|
|
+ if reflect.ValueOf(value).Len() > 0 {
|
|
|
+ str = fmt.Sprintf("(%v NOT IN (?))", scope.Quote(primaryKey))
|
|
|
+ clause["args"] = []interface{}{value}
|
|
|
+ }
|
|
|
+ return ""
|
|
|
+ case map[string]interface{}:
|
|
|
+ var sqls []string
|
|
|
+ for key, value := range value {
|
|
|
+ if value != nil {
|
|
|
+ sqls = append(sqls, fmt.Sprintf("(%v <> %v)", scope.Quote(key), scope.AddToVars(value)))
|
|
|
+ } else {
|
|
|
+ sqls = append(sqls, fmt.Sprintf("(%v IS NOT NULL)", scope.Quote(key)))
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return strings.Join(sqls, " AND ")
|
|
|
+ case interface{}:
|
|
|
+ var sqls []string
|
|
|
+ for _, field := range scope.New(value).Fields() {
|
|
|
+ if !field.IsBlank {
|
|
|
+ sqls = append(sqls, fmt.Sprintf("(%v <> %v)", scope.Quote(field.DBName), scope.AddToVars(field.Field.Interface())))
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return strings.Join(sqls, " AND ")
|
|
|
+ }
|
|
|
+
|
|
|
+ args := clause["args"].([]interface{})
|
|
|
+ for _, arg := range args {
|
|
|
+ switch reflect.ValueOf(arg).Kind() {
|
|
|
+ case reflect.Slice: // For where("id in (?)", []int64{1,2})
|
|
|
+ if bytes, ok := arg.([]byte); ok {
|
|
|
+ str = strings.Replace(str, "?", scope.AddToVars(bytes), 1)
|
|
|
+ } else if values := reflect.ValueOf(arg); values.Len() > 0 {
|
|
|
+ var tempMarks []string
|
|
|
+ for i := 0; i < values.Len(); i++ {
|
|
|
+ tempMarks = append(tempMarks, scope.AddToVars(values.Index(i).Interface()))
|
|
|
+ }
|
|
|
+ str = strings.Replace(str, "?", strings.Join(tempMarks, ","), 1)
|
|
|
+ } else {
|
|
|
+ str = strings.Replace(str, "?", scope.AddToVars(Expr("NULL")), 1)
|
|
|
+ }
|
|
|
+ default:
|
|
|
+ if scanner, ok := interface{}(arg).(driver.Valuer); ok {
|
|
|
+ arg, _ = scanner.Value()
|
|
|
+ }
|
|
|
+ str = strings.Replace(notEqualSQL, "?", scope.AddToVars(arg), 1)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) buildSelectQuery(clause map[string]interface{}) (str string) {
|
|
|
+ switch value := clause["query"].(type) {
|
|
|
+ case string:
|
|
|
+ str = value
|
|
|
+ case []string:
|
|
|
+ str = strings.Join(value, ", ")
|
|
|
+ }
|
|
|
+
|
|
|
+ args := clause["args"].([]interface{})
|
|
|
+ for _, arg := range args {
|
|
|
+ switch reflect.ValueOf(arg).Kind() {
|
|
|
+ case reflect.Slice:
|
|
|
+ values := reflect.ValueOf(arg)
|
|
|
+ var tempMarks []string
|
|
|
+ for i := 0; i < values.Len(); i++ {
|
|
|
+ tempMarks = append(tempMarks, scope.AddToVars(values.Index(i).Interface()))
|
|
|
+ }
|
|
|
+ str = strings.Replace(str, "?", strings.Join(tempMarks, ","), 1)
|
|
|
+ default:
|
|
|
+ if valuer, ok := interface{}(arg).(driver.Valuer); ok {
|
|
|
+ arg, _ = valuer.Value()
|
|
|
+ }
|
|
|
+ str = strings.Replace(str, "?", scope.AddToVars(arg), 1)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) whereSQL() (sql string) {
|
|
|
+ var (
|
|
|
+ quotedTableName = scope.QuotedTableName()
|
|
|
+ primaryConditions, andConditions, orConditions []string
|
|
|
+ )
|
|
|
+
|
|
|
+ if !scope.Search.Unscoped && scope.HasColumn("deleted_at") {
|
|
|
+ sql := fmt.Sprintf("%v.deleted_at IS NULL", quotedTableName)
|
|
|
+ primaryConditions = append(primaryConditions, sql)
|
|
|
+ }
|
|
|
+
|
|
|
+ if !scope.PrimaryKeyZero() {
|
|
|
+ for _, field := range scope.PrimaryFields() {
|
|
|
+ sql := fmt.Sprintf("%v.%v = %v", quotedTableName, scope.Quote(field.DBName), scope.AddToVars(field.Field.Interface()))
|
|
|
+ primaryConditions = append(primaryConditions, sql)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for _, clause := range scope.Search.whereConditions {
|
|
|
+ if sql := scope.buildWhereCondition(clause); sql != "" {
|
|
|
+ andConditions = append(andConditions, sql)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for _, clause := range scope.Search.orConditions {
|
|
|
+ if sql := scope.buildWhereCondition(clause); sql != "" {
|
|
|
+ orConditions = append(orConditions, sql)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for _, clause := range scope.Search.notConditions {
|
|
|
+ if sql := scope.buildNotCondition(clause); sql != "" {
|
|
|
+ andConditions = append(andConditions, sql)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ orSQL := strings.Join(orConditions, " OR ")
|
|
|
+ combinedSQL := strings.Join(andConditions, " AND ")
|
|
|
+ if len(combinedSQL) > 0 {
|
|
|
+ if len(orSQL) > 0 {
|
|
|
+ combinedSQL = combinedSQL + " OR " + orSQL
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ combinedSQL = orSQL
|
|
|
+ }
|
|
|
+
|
|
|
+ if len(primaryConditions) > 0 {
|
|
|
+ sql = "WHERE " + strings.Join(primaryConditions, " AND ")
|
|
|
+ if len(combinedSQL) > 0 {
|
|
|
+ sql = sql + " AND (" + combinedSQL + ")"
|
|
|
+ }
|
|
|
+ } else if len(combinedSQL) > 0 {
|
|
|
+ sql = "WHERE " + combinedSQL
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) selectSQL() string {
|
|
|
+ if len(scope.Search.selects) == 0 {
|
|
|
+ if len(scope.Search.joinConditions) > 0 {
|
|
|
+ return fmt.Sprintf("%v.*", scope.QuotedTableName())
|
|
|
+ }
|
|
|
+ return "*"
|
|
|
+ }
|
|
|
+ return scope.buildSelectQuery(scope.Search.selects)
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) orderSQL() string {
|
|
|
+ if len(scope.Search.orders) == 0 || scope.Search.countingQuery {
|
|
|
+ return ""
|
|
|
+ }
|
|
|
+ return " ORDER BY " + strings.Join(scope.Search.orders, ",")
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) limitAndOffsetSQL() string {
|
|
|
+ return scope.Dialect().LimitAndOffsetSQL(scope.Search.limit, scope.Search.offset)
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) groupSQL() string {
|
|
|
+ if len(scope.Search.group) == 0 {
|
|
|
+ return ""
|
|
|
+ }
|
|
|
+ return " GROUP BY " + scope.Search.group
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) havingSQL() string {
|
|
|
+ if len(scope.Search.havingConditions) == 0 {
|
|
|
+ return ""
|
|
|
+ }
|
|
|
+
|
|
|
+ var andConditions []string
|
|
|
+ for _, clause := range scope.Search.havingConditions {
|
|
|
+ if sql := scope.buildWhereCondition(clause); sql != "" {
|
|
|
+ andConditions = append(andConditions, sql)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ combinedSQL := strings.Join(andConditions, " AND ")
|
|
|
+ if len(combinedSQL) == 0 {
|
|
|
+ return ""
|
|
|
+ }
|
|
|
+
|
|
|
+ return " HAVING " + combinedSQL
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) joinsSQL() string {
|
|
|
+ var joinConditions []string
|
|
|
+ for _, clause := range scope.Search.joinConditions {
|
|
|
+ if sql := scope.buildWhereCondition(clause); sql != "" {
|
|
|
+ joinConditions = append(joinConditions, strings.TrimSuffix(strings.TrimPrefix(sql, "("), ")"))
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return strings.Join(joinConditions, " ") + " "
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) prepareQuerySQL() {
|
|
|
+ if scope.Search.raw {
|
|
|
+ scope.Raw(strings.TrimSuffix(strings.TrimPrefix(scope.CombinedConditionSql(), " WHERE ("), ")"))
|
|
|
+ } else {
|
|
|
+ scope.Raw(fmt.Sprintf("SELECT %v FROM %v %v", scope.selectSQL(), scope.QuotedTableName(), scope.CombinedConditionSql()))
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) inlineCondition(values ...interface{}) *Scope {
|
|
|
+ if len(values) > 0 {
|
|
|
+ scope.Search.Where(values[0], values[1:]...)
|
|
|
+ }
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) callCallbacks(funcs []*func(s *Scope)) *Scope {
|
|
|
+ for _, f := range funcs {
|
|
|
+ (*f)(scope)
|
|
|
+ if scope.skipLeft {
|
|
|
+ break
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) updatedAttrsWithValues(values map[string]interface{}) (results map[string]interface{}, hasUpdate bool) {
|
|
|
+ if scope.IndirectValue().Kind() != reflect.Struct {
|
|
|
+ return values, true
|
|
|
+ }
|
|
|
+
|
|
|
+ results = map[string]interface{}{}
|
|
|
+ for key, value := range values {
|
|
|
+ if field, ok := scope.FieldByName(key); ok && scope.changeableField(field) {
|
|
|
+ if !reflect.DeepEqual(field.Field, reflect.ValueOf(value)) {
|
|
|
+ if _, ok := value.(*expr); ok {
|
|
|
+ hasUpdate = true
|
|
|
+ results[field.DBName] = value
|
|
|
+ } else if !equalAsString(field.Field.Interface(), value) {
|
|
|
+ field.Set(value)
|
|
|
+ if field.IsNormal {
|
|
|
+ hasUpdate = true
|
|
|
+ results[field.DBName] = field.Field.Interface()
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ field.Set(value)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) row() *sql.Row {
|
|
|
+ defer scope.trace(NowFunc())
|
|
|
+ scope.callCallbacks(scope.db.parent.callbacks.rowQueries)
|
|
|
+ scope.prepareQuerySQL()
|
|
|
+ return scope.SQLDB().QueryRow(scope.SQL, scope.SQLVars...)
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) rows() (*sql.Rows, error) {
|
|
|
+ defer scope.trace(NowFunc())
|
|
|
+ scope.callCallbacks(scope.db.parent.callbacks.rowQueries)
|
|
|
+ scope.prepareQuerySQL()
|
|
|
+ return scope.SQLDB().Query(scope.SQL, scope.SQLVars...)
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) initialize() *Scope {
|
|
|
+ for _, clause := range scope.Search.whereConditions {
|
|
|
+ scope.updatedAttrsWithValues(convertInterfaceToMap(clause["query"]))
|
|
|
+ }
|
|
|
+ scope.updatedAttrsWithValues(convertInterfaceToMap(scope.Search.initAttrs))
|
|
|
+ scope.updatedAttrsWithValues(convertInterfaceToMap(scope.Search.assignAttrs))
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) pluck(column string, value interface{}) *Scope {
|
|
|
+ dest := reflect.Indirect(reflect.ValueOf(value))
|
|
|
+ scope.Search.Select(column)
|
|
|
+ if dest.Kind() != reflect.Slice {
|
|
|
+ scope.Err(fmt.Errorf("results should be a slice, not %s", dest.Kind()))
|
|
|
+ return scope
|
|
|
+ }
|
|
|
+
|
|
|
+ rows, err := scope.rows()
|
|
|
+ if scope.Err(err) == nil {
|
|
|
+ defer rows.Close()
|
|
|
+ for rows.Next() {
|
|
|
+ elem := reflect.New(dest.Type().Elem()).Interface()
|
|
|
+ scope.Err(rows.Scan(elem))
|
|
|
+ dest.Set(reflect.Append(dest, reflect.ValueOf(elem).Elem()))
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) count(value interface{}) *Scope {
|
|
|
+ scope.Search.Select("count(*)")
|
|
|
+ scope.Search.countingQuery = true
|
|
|
+ scope.Err(scope.row().Scan(value))
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) typeName() string {
|
|
|
+ typ := scope.IndirectValue().Type()
|
|
|
+
|
|
|
+ for typ.Kind() == reflect.Slice || typ.Kind() == reflect.Ptr {
|
|
|
+ typ = typ.Elem()
|
|
|
+ }
|
|
|
+
|
|
|
+ return typ.Name()
|
|
|
+}
|
|
|
+
|
|
|
+// trace print sql log
|
|
|
+func (scope *Scope) trace(t time.Time) {
|
|
|
+ if len(scope.SQL) > 0 {
|
|
|
+ scope.db.slog(scope.SQL, t, scope.SQLVars...)
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) changeableField(field *Field) bool {
|
|
|
+ if selectAttrs := scope.SelectAttrs(); len(selectAttrs) > 0 {
|
|
|
+ for _, attr := range selectAttrs {
|
|
|
+ if field.Name == attr || field.DBName == attr {
|
|
|
+ return true
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return false
|
|
|
+ }
|
|
|
+
|
|
|
+ for _, attr := range scope.OmitAttrs() {
|
|
|
+ if field.Name == attr || field.DBName == attr {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return true
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) shouldSaveAssociations() bool {
|
|
|
+ if saveAssociations, ok := scope.Get("gorm:save_associations"); ok && !saveAssociations.(bool) {
|
|
|
+ return false
|
|
|
+ }
|
|
|
+ return true && !scope.HasError()
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) related(value interface{}, foreignKeys ...string) *Scope {
|
|
|
+ toScope := scope.db.NewScope(value)
|
|
|
+
|
|
|
+ for _, foreignKey := range append(foreignKeys, toScope.typeName()+"Id", scope.typeName()+"Id") {
|
|
|
+ fromField, _ := scope.FieldByName(foreignKey)
|
|
|
+ toField, _ := toScope.FieldByName(foreignKey)
|
|
|
+
|
|
|
+ if fromField != nil {
|
|
|
+ if relationship := fromField.Relationship; relationship != nil {
|
|
|
+ if relationship.Kind == "many_to_many" {
|
|
|
+ joinTableHandler := relationship.JoinTableHandler
|
|
|
+ scope.Err(joinTableHandler.JoinWith(joinTableHandler, toScope.db, scope.Value).Find(value).Error)
|
|
|
+ } else if relationship.Kind == "belongs_to" {
|
|
|
+ query := toScope.db
|
|
|
+ for idx, foreignKey := range relationship.ForeignDBNames {
|
|
|
+ if field, ok := scope.FieldByName(foreignKey); ok {
|
|
|
+ query = query.Where(fmt.Sprintf("%v = ?", scope.Quote(relationship.AssociationForeignDBNames[idx])), field.Field.Interface())
|
|
|
+ }
|
|
|
+ }
|
|
|
+ scope.Err(query.Find(value).Error)
|
|
|
+ } else if relationship.Kind == "has_many" || relationship.Kind == "has_one" {
|
|
|
+ query := toScope.db
|
|
|
+ for idx, foreignKey := range relationship.ForeignDBNames {
|
|
|
+ if field, ok := scope.FieldByName(relationship.AssociationForeignDBNames[idx]); ok {
|
|
|
+ query = query.Where(fmt.Sprintf("%v = ?", scope.Quote(foreignKey)), field.Field.Interface())
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if relationship.PolymorphicType != "" {
|
|
|
+ query = query.Where(fmt.Sprintf("%v = ?", scope.Quote(relationship.PolymorphicDBName)), scope.TableName())
|
|
|
+ }
|
|
|
+ scope.Err(query.Find(value).Error)
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ sql := fmt.Sprintf("%v = ?", scope.Quote(toScope.PrimaryKey()))
|
|
|
+ scope.Err(toScope.db.Where(sql, fromField.Field.Interface()).Find(value).Error)
|
|
|
+ }
|
|
|
+ return scope
|
|
|
+ } else if toField != nil {
|
|
|
+ sql := fmt.Sprintf("%v = ?", scope.Quote(toField.DBName))
|
|
|
+ scope.Err(toScope.db.Where(sql, scope.PrimaryKeyValue()).Find(value).Error)
|
|
|
+ return scope
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ scope.Err(fmt.Errorf("invalid association %v", foreignKeys))
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+// getTableOptions return the table options string or an empty string if the table options does not exist
|
|
|
+func (scope *Scope) getTableOptions() string {
|
|
|
+ tableOptions, ok := scope.Get("gorm:table_options")
|
|
|
+ if !ok {
|
|
|
+ return ""
|
|
|
+ }
|
|
|
+ return tableOptions.(string)
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) createJoinTable(field *StructField) {
|
|
|
+ if relationship := field.Relationship; relationship != nil && relationship.JoinTableHandler != nil {
|
|
|
+ joinTableHandler := relationship.JoinTableHandler
|
|
|
+ joinTable := joinTableHandler.Table(scope.db)
|
|
|
+ if !scope.Dialect().HasTable(joinTable) {
|
|
|
+ toScope := &Scope{Value: reflect.New(field.Struct.Type).Interface()}
|
|
|
+
|
|
|
+ var sqlTypes, primaryKeys []string
|
|
|
+ for idx, fieldName := range relationship.ForeignFieldNames {
|
|
|
+ if field, ok := scope.FieldByName(fieldName); ok {
|
|
|
+ foreignKeyStruct := field.clone()
|
|
|
+ foreignKeyStruct.IsPrimaryKey = false
|
|
|
+ foreignKeyStruct.TagSettings["IS_JOINTABLE_FOREIGNKEY"] = "true"
|
|
|
+ sqlTypes = append(sqlTypes, scope.Quote(relationship.ForeignDBNames[idx])+" "+scope.Dialect().DataTypeOf(foreignKeyStruct))
|
|
|
+ primaryKeys = append(primaryKeys, scope.Quote(relationship.ForeignDBNames[idx]))
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for idx, fieldName := range relationship.AssociationForeignFieldNames {
|
|
|
+ if field, ok := toScope.FieldByName(fieldName); ok {
|
|
|
+ foreignKeyStruct := field.clone()
|
|
|
+ foreignKeyStruct.IsPrimaryKey = false
|
|
|
+ foreignKeyStruct.TagSettings["IS_JOINTABLE_FOREIGNKEY"] = "true"
|
|
|
+ sqlTypes = append(sqlTypes, scope.Quote(relationship.AssociationForeignDBNames[idx])+" "+scope.Dialect().DataTypeOf(foreignKeyStruct))
|
|
|
+ primaryKeys = append(primaryKeys, scope.Quote(relationship.AssociationForeignDBNames[idx]))
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ scope.Err(scope.NewDB().Exec(fmt.Sprintf("CREATE TABLE %v (%v, PRIMARY KEY (%v)) %s", scope.Quote(joinTable), strings.Join(sqlTypes, ","), strings.Join(primaryKeys, ","), scope.getTableOptions())).Error)
|
|
|
+ }
|
|
|
+ scope.NewDB().Table(joinTable).AutoMigrate(joinTableHandler)
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) createTable() *Scope {
|
|
|
+ var tags []string
|
|
|
+ var primaryKeys []string
|
|
|
+ var primaryKeyInColumnType = false
|
|
|
+ for _, field := range scope.GetModelStruct().StructFields {
|
|
|
+ if field.IsNormal {
|
|
|
+ sqlTag := scope.Dialect().DataTypeOf(field)
|
|
|
+
|
|
|
+ // Check if the primary key constraint was specified as
|
|
|
+ // part of the column type. If so, we can only support
|
|
|
+ // one column as the primary key.
|
|
|
+ if strings.Contains(strings.ToLower(sqlTag), "primary key") {
|
|
|
+ primaryKeyInColumnType = true
|
|
|
+ }
|
|
|
+
|
|
|
+ tags = append(tags, scope.Quote(field.DBName)+" "+sqlTag)
|
|
|
+ }
|
|
|
+
|
|
|
+ if field.IsPrimaryKey {
|
|
|
+ primaryKeys = append(primaryKeys, scope.Quote(field.DBName))
|
|
|
+ }
|
|
|
+ scope.createJoinTable(field)
|
|
|
+ }
|
|
|
+
|
|
|
+ var primaryKeyStr string
|
|
|
+ if len(primaryKeys) > 0 && !primaryKeyInColumnType {
|
|
|
+ primaryKeyStr = fmt.Sprintf(", PRIMARY KEY (%v)", strings.Join(primaryKeys, ","))
|
|
|
+ }
|
|
|
+
|
|
|
+ scope.Raw(fmt.Sprintf("CREATE TABLE %v (%v %v) %s", scope.QuotedTableName(), strings.Join(tags, ","), primaryKeyStr, scope.getTableOptions())).Exec()
|
|
|
+
|
|
|
+ scope.autoIndex()
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) dropTable() *Scope {
|
|
|
+ scope.Raw(fmt.Sprintf("DROP TABLE %v", scope.QuotedTableName())).Exec()
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) modifyColumn(column string, typ string) {
|
|
|
+ scope.Raw(fmt.Sprintf("ALTER TABLE %v MODIFY %v %v", scope.QuotedTableName(), scope.Quote(column), typ)).Exec()
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) dropColumn(column string) {
|
|
|
+ scope.Raw(fmt.Sprintf("ALTER TABLE %v DROP COLUMN %v", scope.QuotedTableName(), scope.Quote(column))).Exec()
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) addIndex(unique bool, indexName string, column ...string) {
|
|
|
+ if scope.Dialect().HasIndex(scope.TableName(), indexName) {
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ var columns []string
|
|
|
+ for _, name := range column {
|
|
|
+ columns = append(columns, scope.quoteIfPossible(name))
|
|
|
+ }
|
|
|
+
|
|
|
+ sqlCreate := "CREATE INDEX"
|
|
|
+ if unique {
|
|
|
+ sqlCreate = "CREATE UNIQUE INDEX"
|
|
|
+ }
|
|
|
+
|
|
|
+ scope.Raw(fmt.Sprintf("%s %v ON %v(%v) %v", sqlCreate, indexName, scope.QuotedTableName(), strings.Join(columns, ", "), scope.whereSQL())).Exec()
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) addForeignKey(field string, dest string, onDelete string, onUpdate string) {
|
|
|
+ var keyName = fmt.Sprintf("%s_%s_%s_foreign", scope.TableName(), field, dest)
|
|
|
+ keyName = regexp.MustCompile("(_*[^a-zA-Z]+_*|_+)").ReplaceAllString(keyName, "_")
|
|
|
+
|
|
|
+ if scope.Dialect().HasForeignKey(scope.TableName(), keyName) {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ var query = `ALTER TABLE %s ADD CONSTRAINT %s FOREIGN KEY (%s) REFERENCES %s ON DELETE %s ON UPDATE %s;`
|
|
|
+ scope.Raw(fmt.Sprintf(query, scope.QuotedTableName(), scope.quoteIfPossible(keyName), scope.quoteIfPossible(field), dest, onDelete, onUpdate)).Exec()
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) removeIndex(indexName string) {
|
|
|
+ scope.Dialect().RemoveIndex(scope.TableName(), indexName)
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) autoMigrate() *Scope {
|
|
|
+ tableName := scope.TableName()
|
|
|
+ quotedTableName := scope.QuotedTableName()
|
|
|
+
|
|
|
+ if !scope.Dialect().HasTable(tableName) {
|
|
|
+ scope.createTable()
|
|
|
+ } else {
|
|
|
+ for _, field := range scope.GetModelStruct().StructFields {
|
|
|
+ if !scope.Dialect().HasColumn(tableName, field.DBName) {
|
|
|
+ if field.IsNormal {
|
|
|
+ sqlTag := scope.Dialect().DataTypeOf(field)
|
|
|
+ scope.Raw(fmt.Sprintf("ALTER TABLE %v ADD %v %v;", quotedTableName, scope.Quote(field.DBName), sqlTag)).Exec()
|
|
|
+ }
|
|
|
+ }
|
|
|
+ scope.createJoinTable(field)
|
|
|
+ }
|
|
|
+ scope.autoIndex()
|
|
|
+ }
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) autoIndex() *Scope {
|
|
|
+ var indexes = map[string][]string{}
|
|
|
+ var uniqueIndexes = map[string][]string{}
|
|
|
+
|
|
|
+ for _, field := range scope.GetStructFields() {
|
|
|
+ if name, ok := field.TagSettings["INDEX"]; ok {
|
|
|
+ if name == "INDEX" {
|
|
|
+ name = fmt.Sprintf("idx_%v_%v", scope.TableName(), field.DBName)
|
|
|
+ }
|
|
|
+ indexes[name] = append(indexes[name], field.DBName)
|
|
|
+ }
|
|
|
+
|
|
|
+ if name, ok := field.TagSettings["UNIQUE_INDEX"]; ok {
|
|
|
+ if name == "UNIQUE_INDEX" {
|
|
|
+ name = fmt.Sprintf("uix_%v_%v", scope.TableName(), field.DBName)
|
|
|
+ }
|
|
|
+ uniqueIndexes[name] = append(uniqueIndexes[name], field.DBName)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ for name, columns := range indexes {
|
|
|
+ scope.NewDB().Model(scope.Value).AddIndex(name, columns...)
|
|
|
+ }
|
|
|
+
|
|
|
+ for name, columns := range uniqueIndexes {
|
|
|
+ scope.NewDB().Model(scope.Value).AddUniqueIndex(name, columns...)
|
|
|
+ }
|
|
|
+
|
|
|
+ return scope
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) getColumnAsArray(columns []string, values ...interface{}) (results [][]interface{}) {
|
|
|
+ for _, value := range values {
|
|
|
+ indirectValue := reflect.ValueOf(value)
|
|
|
+ for indirectValue.Kind() == reflect.Ptr {
|
|
|
+ indirectValue = indirectValue.Elem()
|
|
|
+ }
|
|
|
+
|
|
|
+ switch indirectValue.Kind() {
|
|
|
+ case reflect.Slice:
|
|
|
+ for i := 0; i < indirectValue.Len(); i++ {
|
|
|
+ var result []interface{}
|
|
|
+ var object = indirect(indirectValue.Index(i))
|
|
|
+ for _, column := range columns {
|
|
|
+ result = append(result, object.FieldByName(column).Interface())
|
|
|
+ }
|
|
|
+ results = append(results, result)
|
|
|
+ }
|
|
|
+ case reflect.Struct:
|
|
|
+ var result []interface{}
|
|
|
+ for _, column := range columns {
|
|
|
+ result = append(result, indirectValue.FieldByName(column).Interface())
|
|
|
+ }
|
|
|
+ results = append(results, result)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+func (scope *Scope) getColumnAsScope(column string) *Scope {
|
|
|
+ indirectScopeValue := scope.IndirectValue()
|
|
|
+
|
|
|
+ switch indirectScopeValue.Kind() {
|
|
|
+ case reflect.Slice:
|
|
|
+ if fieldStruct, ok := scope.GetModelStruct().ModelType.FieldByName(column); ok {
|
|
|
+ fieldType := fieldStruct.Type
|
|
|
+ if fieldType.Kind() == reflect.Slice || fieldType.Kind() == reflect.Ptr {
|
|
|
+ fieldType = fieldType.Elem()
|
|
|
+ }
|
|
|
+
|
|
|
+ results := reflect.New(reflect.SliceOf(reflect.PtrTo(fieldType))).Elem()
|
|
|
+
|
|
|
+ for i := 0; i < indirectScopeValue.Len(); i++ {
|
|
|
+ result := indirect(indirect(indirectScopeValue.Index(i)).FieldByName(column))
|
|
|
+
|
|
|
+ if result.Kind() == reflect.Slice {
|
|
|
+ for j := 0; j < result.Len(); j++ {
|
|
|
+ if elem := result.Index(j); elem.CanAddr() {
|
|
|
+ results = reflect.Append(results, elem.Addr())
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } else if result.CanAddr() {
|
|
|
+ results = reflect.Append(results, result.Addr())
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return scope.New(results.Interface())
|
|
|
+ }
|
|
|
+ case reflect.Struct:
|
|
|
+ if field := indirectScopeValue.FieldByName(column); field.CanAddr() {
|
|
|
+ return scope.New(field.Addr().Interface())
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return nil
|
|
|
+}
|