123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228 |
- package gorm
- import (
- "fmt"
- "strings"
- "github.com/jinzhu/gorm/clause"
- )
- // Model specify the model you would like to run db operations
- // // update all users's name to `hello`
- // db.Model(&User{}).Update("name", "hello")
- // // if user's primary key is non-blank, will use it as condition, then will only update the user's name to `hello`
- // db.Model(&user).Update("name", "hello")
- func (db *DB) Model(value interface{}) (tx *DB) {
- tx = db.getInstance()
- tx.Statement.Model = value
- return
- }
- // Clauses Add clauses
- func (db *DB) Clauses(conds ...clause.Expression) (tx *DB) {
- tx = db.getInstance()
- var whereConds []interface{}
- for _, cond := range conds {
- if c, ok := cond.(clause.Interface); ok {
- tx.Statement.AddClause(c)
- } else {
- whereConds = append(whereConds, cond)
- }
- }
- if len(whereConds) > 0 {
- tx.Statement.AddClause(clause.Where{Exprs: tx.Statement.BuildCondtion(whereConds[0], whereConds[1:]...)})
- }
- return
- }
- // Table specify the table you would like to run db operations
- func (db *DB) Table(name string) (tx *DB) {
- tx = db.getInstance()
- tx.Statement.Table = name
- return
- }
- // Select specify fields that you want when querying, creating, updating
- func (db *DB) Select(query interface{}, args ...interface{}) (tx *DB) {
- tx = db.getInstance()
- switch v := query.(type) {
- case []string:
- tx.Statement.Selects = v
- for _, arg := range args {
- switch arg := arg.(type) {
- case string:
- tx.Statement.Selects = append(tx.Statement.Selects, arg)
- case []string:
- tx.Statement.Selects = append(tx.Statement.Selects, arg...)
- default:
- tx.AddError(fmt.Errorf("unsupported select args %v %v", query, args))
- return
- }
- }
- case string:
- fields := strings.FieldsFunc(v, isChar)
- // normal field names
- if len(fields) == 1 || (len(fields) == 3 && strings.ToUpper(fields[1]) == "AS") {
- tx.Statement.Selects = fields
- for _, arg := range args {
- switch arg := arg.(type) {
- case string:
- tx.Statement.Selects = append(tx.Statement.Selects, arg)
- case []string:
- tx.Statement.Selects = append(tx.Statement.Selects, arg...)
- default:
- tx.Statement.AddClause(clause.Select{
- Expression: clause.Expr{SQL: v, Vars: args},
- })
- return
- }
- }
- } else {
- tx.Statement.AddClause(clause.Select{
- Expression: clause.Expr{SQL: v, Vars: args},
- })
- }
- default:
- tx.AddError(fmt.Errorf("unsupported select args %v %v", query, args))
- }
- return
- }
- // Omit specify fields that you want to ignore when creating, updating and querying
- func (db *DB) Omit(columns ...string) (tx *DB) {
- tx = db.getInstance()
- if len(columns) == 1 && strings.ContainsRune(columns[0], ',') {
- tx.Statement.Omits = strings.FieldsFunc(columns[0], isChar)
- } else {
- tx.Statement.Omits = columns
- }
- return
- }
- func (db *DB) Where(query interface{}, args ...interface{}) (tx *DB) {
- tx = db.getInstance()
- tx.Statement.AddClause(clause.Where{Exprs: tx.Statement.BuildCondtion(query, args...)})
- return
- }
- // Not add NOT condition
- func (db *DB) Not(query interface{}, args ...interface{}) (tx *DB) {
- tx = db.getInstance()
- tx.Statement.AddClause(clause.Where{Exprs: []clause.Expression{clause.Not(tx.Statement.BuildCondtion(query, args...)...)}})
- return
- }
- // Or add OR conditions
- func (db *DB) Or(query interface{}, args ...interface{}) (tx *DB) {
- tx = db.getInstance()
- tx.Statement.AddClause(clause.Where{Exprs: []clause.Expression{clause.Or(tx.Statement.BuildCondtion(query, args...)...)}})
- return
- }
- // Joins specify Joins conditions
- // db.Joins("JOIN emails ON emails.user_id = users.id AND emails.email = ?", "jinzhu@example.org").Find(&user)
- func (db *DB) Joins(query string, args ...interface{}) (tx *DB) {
- tx = db.getInstance()
- return
- }
- // Group specify the group method on the find
- func (db *DB) Group(column string) (tx *DB) {
- tx = db.getInstance()
- return
- }
- // Having specify HAVING conditions for GROUP BY
- func (db *DB) Having(query interface{}, args ...interface{}) (tx *DB) {
- tx = db.getInstance()
- return
- }
- // Order specify order when retrieve records from database
- // db.Order("name DESC")
- // db.Order(gorm.Expr("name = ? DESC", "first")) // sql expression
- func (db *DB) Order(value interface{}) (tx *DB) {
- tx = db.getInstance()
- switch v := value.(type) {
- case clause.OrderByColumn:
- tx.Statement.AddClause(clause.OrderBy{
- Columns: []clause.OrderByColumn{v},
- })
- default:
- tx.Statement.AddClause(clause.OrderBy{
- Columns: []clause.OrderByColumn{{
- Column: clause.Column{Name: fmt.Sprint(value), Raw: true},
- }},
- })
- }
- return
- }
- // Limit specify the number of records to be retrieved
- func (db *DB) Limit(limit int64) (tx *DB) {
- tx = db.getInstance()
- return
- }
- // Offset specify the number of records to skip before starting to return the records
- func (db *DB) Offset(offset int64) (tx *DB) {
- tx = db.getInstance()
- return
- }
- // Scopes pass current database connection to arguments `func(*DB) *DB`, which could be used to add conditions dynamically
- // func AmountGreaterThan1000(db *gorm.DB) *gorm.DB {
- // return db.Where("amount > ?", 1000)
- // }
- //
- // func OrderStatus(status []string) func (db *gorm.DB) *gorm.DB {
- // return func (db *gorm.DB) *gorm.DB {
- // return db.Scopes(AmountGreaterThan1000).Where("status in (?)", status)
- // }
- // }
- //
- // db.Scopes(AmountGreaterThan1000, OrderStatus([]string{"paid", "shipped"})).Find(&orders)
- func (db *DB) Scopes(funcs ...func(*DB) *DB) *DB {
- for _, f := range funcs {
- db = f(db)
- }
- return db
- }
- // Preload preload associations with given conditions
- // db.Preload("Orders", "state NOT IN (?)", "cancelled").Find(&users)
- func (db *DB) Preload(column string, conditions ...interface{}) (tx *DB) {
- tx = db.getInstance()
- return
- }
- func (db *DB) Assign(attrs ...interface{}) (tx *DB) {
- tx = db.getInstance()
- return
- }
- func (db *DB) Attrs(attrs ...interface{}) (tx *DB) {
- tx = db.getInstance()
- return
- }
- func (db *DB) Unscoped() (tx *DB) {
- tx = db.getInstance()
- return
- }
- func (db *DB) Raw(sql string, values ...interface{}) (tx *DB) {
- tx = db.getInstance()
- tx.Statement.SQL = strings.Builder{}
- clause.Expr{SQL: sql, Vars: values}.Build(tx.Statement)
- return
- }
|