func beforeQuery(scope *gorm.Scope) { if isLocalizable(scope) { quotedTableName := scope.QuotedTableName() quotedPrimaryKey := scope.Quote(scope.PrimaryKey()) locale, isLocale := getLocale(scope) switch mode, _ := scope.DB().Get("l10n:mode"); mode { case "unscoped": case "global": scope.Search.Where(fmt.Sprintf("%v.language_code = ?", quotedTableName), Global) case "locale": scope.Search.Where(fmt.Sprintf("%v.language_code = ?", quotedTableName), locale) case "reverse": if !scope.Search.Unscoped && scope.Fields()["deleted_at"] != nil { scope.Search.Where(fmt.Sprintf("(%v NOT IN (SELECT DISTINCT(%v) FROM %v t2 WHERE t2.language_code = ? AND t2.deleted_at IS NULL) AND language_code = ?)", quotedPrimaryKey, quotedPrimaryKey, quotedTableName), locale, Global) } else { scope.Search.Where(fmt.Sprintf("(%v NOT IN (SELECT DISTINCT(%v) FROM %v t2 WHERE t2.language_code = ?) AND language_code = ?)", quotedPrimaryKey, quotedPrimaryKey, quotedTableName), locale, Global) } default: if isLocale { if !scope.Search.Unscoped && scope.Fields()["deleted_at"] != nil { scope.Search.Where(fmt.Sprintf("((%v NOT IN (SELECT DISTINCT(%v) FROM %v t2 WHERE t2.language_code = ? AND t2.deleted_at IS NULL) AND language_code = ?) OR language_code = ?) AND deleted_at IS NULL", quotedPrimaryKey, quotedPrimaryKey, quotedTableName), locale, Global, locale) } else { scope.Search.Where(fmt.Sprintf("(%v NOT IN (SELECT DISTINCT(%v) FROM %v t2 WHERE t2.language_code = ?) AND language_code = ?) OR (language_code = ?)", quotedPrimaryKey, quotedPrimaryKey, quotedTableName), locale, Global, locale) } } else { scope.Search.Where(fmt.Sprintf("%v.language_code = ?", quotedTableName), Global) } } } }
func scopePrimaryKeys(scope *gorm.Scope, tableName string) string { var primaryKeys []string for _, field := range scope.PrimaryFields() { key := fmt.Sprintf("%v.%v", scope.Quote(tableName), scope.Quote(field.DBName)) primaryKeys = append(primaryKeys, key) } if len(primaryKeys) > 1 { return fmt.Sprintf("(%v)", strings.Join(primaryKeys, ",")) } return strings.Join(primaryKeys, "") }
func toQueryCondition(scope *gorm.Scope, columns []string) string { var newColumns []string for _, column := range columns { newColumns = append(newColumns, scope.Quote(column)) } if len(columns) > 1 { return fmt.Sprintf("(%v)", strings.Join(newColumns, ",")) } return strings.Join(columns, ",") }