2020-01-29 14:22:44 +03:00
|
|
|
package gorm
|
|
|
|
|
|
|
|
import (
|
|
|
|
"database/sql"
|
2020-02-22 15:57:29 +03:00
|
|
|
"strings"
|
2020-02-04 03:56:15 +03:00
|
|
|
|
|
|
|
"github.com/jinzhu/gorm/clause"
|
2020-01-29 14:22:44 +03:00
|
|
|
)
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
// Create insert the value into database
|
|
|
|
func (db *DB) Create(value interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
tx.Statement.Dest = value
|
|
|
|
tx.callbacks.Create().Execute(tx)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Save update value in database, if the value doesn't have primary key, will insert it
|
|
|
|
func (db *DB) Save(value interface{}) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// First find first record that match given conditions, order by primary key
|
|
|
|
func (db *DB) First(out interface{}, where ...interface{}) (tx *DB) {
|
2020-02-07 18:45:35 +03:00
|
|
|
tx = db.getInstance().Limit(1).Order(clause.OrderByColumn{
|
2020-02-04 03:56:15 +03:00
|
|
|
Column: clause.Column{Table: clause.CurrentTable, Name: clause.PrimaryKey},
|
|
|
|
Desc: true,
|
|
|
|
})
|
2020-02-02 09:40:44 +03:00
|
|
|
tx.Statement.Dest = out
|
|
|
|
tx.callbacks.Query().Execute(tx)
|
2020-01-29 14:22:44 +03:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Take return a record that match given conditions, the order will depend on the database implementation
|
|
|
|
func (db *DB) Take(out interface{}, where ...interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Last find last record that match given conditions, order by primary key
|
|
|
|
func (db *DB) Last(out interface{}, where ...interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Find find records that match given conditions
|
|
|
|
func (db *DB) Find(out interface{}, where ...interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
func (db *DB) FirstOrInit(out interface{}, where ...interface{}) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
func (db *DB) FirstOrCreate(out interface{}, where ...interface{}) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Update update attributes with callbacks, refer: https://jinzhu.github.io/gorm/crud.html#update
|
|
|
|
func (db *DB) Update(column string, value interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Updates update attributes with callbacks, refer: https://jinzhu.github.io/gorm/crud.html#update
|
|
|
|
func (db *DB) Updates(values interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
func (db *DB) UpdateColumn(column string, value interface{}) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *DB) UpdateColumns(values interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
// Delete delete value match given conditions, if the value has primary key, then will including the primary key as condition
|
|
|
|
func (db *DB) Delete(value interface{}, where ...interface{}) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
func (db *DB) Related(value interface{}, foreignKeys ...string) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
//Preloads only preloads relations, don`t touch out
|
|
|
|
func (db *DB) Preloads(out interface{}) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
func (db *DB) Association(column string) *Association {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *DB) Count(value interface{}) (tx *DB) {
|
2020-01-29 14:22:44 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
func (db *DB) Row() *sql.Row {
|
2020-02-22 14:41:01 +03:00
|
|
|
tx := db.getInstance()
|
|
|
|
tx.callbacks.Row().Execute(tx)
|
|
|
|
return tx.Statement.Dest.(*sql.Row)
|
2020-02-03 05:40:03 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
func (db *DB) Rows() (*sql.Rows, error) {
|
2020-02-22 14:41:01 +03:00
|
|
|
tx := db.Set("rows", true)
|
|
|
|
tx.callbacks.Row().Execute(tx)
|
|
|
|
return tx.Statement.Dest.(*sql.Rows), tx.Error
|
2020-02-03 05:40:03 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// Scan scan value to a struct
|
|
|
|
func (db *DB) Scan(dest interface{}) (tx *DB) {
|
2020-01-30 10:14:48 +03:00
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2020-02-03 05:40:03 +03:00
|
|
|
func (db *DB) ScanRows(rows *sql.Rows, result interface{}) error {
|
2020-01-30 10:14:48 +03:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-01-29 14:22:44 +03:00
|
|
|
func (db *DB) Transaction(fc func(tx *DB) error, opts ...*sql.TxOptions) (err error) {
|
|
|
|
panicked := true
|
|
|
|
tx := db.Begin(opts...)
|
|
|
|
defer func() {
|
|
|
|
// Make sure to rollback when panic, Block error or Commit error
|
|
|
|
if panicked || err != nil {
|
|
|
|
tx.Rollback()
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2020-02-22 18:08:20 +03:00
|
|
|
err = fc(tx.Session(&Session{}))
|
2020-01-29 14:22:44 +03:00
|
|
|
|
|
|
|
if err == nil {
|
|
|
|
err = tx.Commit().Error
|
|
|
|
}
|
|
|
|
|
|
|
|
panicked = false
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *DB) Begin(opts ...*sql.TxOptions) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *DB) Commit() (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *DB) Rollback() (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *DB) Exec(sql string, values ...interface{}) (tx *DB) {
|
|
|
|
tx = db.getInstance()
|
2020-02-22 15:57:29 +03:00
|
|
|
tx.Statement.SQL = strings.Builder{}
|
|
|
|
clause.Expr{SQL: sql, Vars: values}.Build(tx.Statement)
|
2020-02-22 14:41:01 +03:00
|
|
|
tx.callbacks.Raw().Execute(tx)
|
2020-01-29 14:22:44 +03:00
|
|
|
return
|
|
|
|
}
|