完成一版 setTxCondition

This commit is contained in:
白茶清欢 2022-05-15 18:04:06 +08:00
parent 43588655ef
commit 670c7c7af9

100
base.go
View File

@ -5,26 +5,19 @@
// Date : 8:04 下午 2021/8/8 // Date : 8:04 下午 2021/8/8
package mysql package mysql
import "gorm.io/gorm" import (
"gorm.io/gorm"
)
type BaseDao struct { type BaseDao struct {
} }
// CreateNewData 创建新的数据 // CreateNewRecord 创建新的数据
// //
// Author : go_developer@163.com<白茶清欢> // Author : go_developer@163.com<白茶清欢>
// //
// Date : 8:06 下午 2021/8/8 // Date : 8:06 下午 2021/8/8
func (b *BaseDao) CreateNewData(dbInstance *gorm.DB, table string, data interface{}, o *Option) error { func (b *BaseDao) CreateNewRecord(dbInstance *gorm.DB, table string, data interface{}) error {
if nil != o {
if o.Limit > 0 {
dbInstance.Limit(o.Limit)
}
if o.Offset > 0 {
dbInstance.Offset(o.Offset)
}
}
return dbInstance.Table(table).Create(data).Error return dbInstance.Table(table).Create(data).Error
} }
@ -33,18 +26,8 @@ func (b *BaseDao) CreateNewData(dbInstance *gorm.DB, table string, data interfac
// Author : go_developer@163.com<白茶清欢> // Author : go_developer@163.com<白茶清欢>
// //
// Date : 8:12 下午 2021/8/8 // Date : 8:12 下午 2021/8/8
func (b *BaseDao) UpdateData(dbInstance *gorm.DB, table string, data interface{}, condition interface{}, o *Option) error { func (b *BaseDao) UpdateData(dbInstance *gorm.DB, table string, updateDate interface{}, optionFuncList ...SetOption) error {
if nil != o { return dbInstance.Table(table).Updates(updateDate).Error
if o.Limit > 0 {
dbInstance.Limit(o.Limit)
}
if o.Offset > 0 {
dbInstance.Offset(o.Offset)
}
}
return dbInstance.Table(table).Where(condition).Updates(data).Error
} }
// Select 查询数据列表 // Select 查询数据列表
@ -128,3 +111,72 @@ func (b *BaseDao) Commit(db *gorm.DB) error {
func (b *BaseDao) Rollback(db *gorm.DB) error { func (b *BaseDao) Rollback(db *gorm.DB) error {
return db.Rollback().Error return db.Rollback().Error
} }
// setTxCondition 设置查询条件
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 17:38 2022/5/15
func (b *BaseDao) setTxCondition(tx *gorm.DB, table string, optionFuncList ...SetOption) {
// 指定查询的表
tx.Table(table)
// 构建查询条件
o := &Option{}
for _, fn := range optionFuncList {
fn(o)
}
// 设置where条件
if nil != o.Where {
tx.Where(o.Where)
}
// 设置 limit offset
if o.Limit > 0 {
tx.Limit(o.Limit)
}
if o.Offset >= 0 {
tx.Offset(o.Offset)
}
// in 语句
if nil != o.In {
tx.Where(o.In)
}
// not in 语句
if nil != o.NotIn {
for field, value := range o.NotIn {
tx.Where(field+" NOT IN ? ", value)
}
}
// like 语句
if nil != o.Like {
for field, value := range o.NotIn {
tx.Where(field+" LIKE ? ", value)
}
}
// NOT LIKE 语句
if nil != o.NotLike {
for field, value := range o.NotLike {
tx.Where(field+" NOT LIKE ? ", value)
}
}
// >=
if nil != o.Start {
for field, value := range o.Start {
tx.Where(field+" >= ?", value)
}
}
// <
if nil != o.End {
for field, value := range o.End {
tx.Where(field+" < ?", value)
}
}
}