79 lines
1.4 KiB
Go
79 lines
1.4 KiB
Go
|
package mySql
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
type SqlMapper struct {
|
||
|
|
Sql string
|
||
|
|
Args []interface{}
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *SqlMapper) setDB(db *gorm.DB) *SqlMapper {
|
||
|
|
t.db = db
|
||
|
|
return t
|
||
|
|
}
|
||
|
|
|
||
|
|
// 查询
|
||
|
|
func (t *SqlMapper) Query() *gorm.DB {
|
||
|
|
return t.db.Raw(t.Sql, t.Args...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t *SqlMapper) Exec() *gorm.DB {
|
||
|
|
return t.db.Exec(t.Sql, t.Args...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewSqlMapper(sql string, args []interface{}) *SqlMapper {
|
||
|
|
return &SqlMapper{Sql: sql, Args: args}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Mapper(sql string, args []interface{}, err error) *SqlMapper {
|
||
|
|
if err != nil {
|
||
|
|
panic(err.Error())
|
||
|
|
}
|
||
|
|
return NewSqlMapper(sql, args)
|
||
|
|
}
|
||
|
|
|
||
|
|
type SqlMappers []*SqlMapper
|
||
|
|
|
||
|
|
func Mappers(sqlMappers ...*SqlMapper) (list SqlMappers) {
|
||
|
|
list = sqlMappers
|
||
|
|
return
|
||
|
|
}
|
||
|
|
func (t SqlMappers) apply(tx *gorm.DB) {
|
||
|
|
for _, sql := range t {
|
||
|
|
sql.setDB(tx)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func (t SqlMappers) Exec(f func() error) error {
|
||
|
|
if len(t) == 0 {
|
||
|
|
return errors.New("无Mapper")
|
||
|
|
}
|
||
|
|
//其实是以第一个为准
|
||
|
|
return t[0].db.Transaction(func(tx *gorm.DB) error {
|
||
|
|
t.apply(tx)
|
||
|
|
return f()
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (t SqlMappers) ExecTransaction() error {
|
||
|
|
//其实是以第一个为准
|
||
|
|
return t[0].db.Transaction(func(tx *gorm.DB) error {
|
||
|
|
fmt.Println("事务开始")
|
||
|
|
for _, sql := range t {
|
||
|
|
sql.setDB(tx)
|
||
|
|
err := sql.Exec().Error
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println("事务结束(失败)")
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fmt.Println("事务成功")
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
}
|