gorm/finisher_api.go

175 lines
4.0 KiB
Go
Raw Normal View History

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
}