tool/lxutils
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

支持多个db

Browse Source
This commit is contained in:
liuchangshun 2024-03-06 15:02:11 +08:00
parent 4fb99540c5
commit 8597d83657
1 changed files with 12 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

91
lxDb/sql.go
View File

@ -11,13 +11,8 @@ import (
// 带事务的, 和不带事务的 说明:
// 如果需要支持事务请调aaaTx方法, 并传开启事务的DB
// OneById 通过ID查询一条
func OneById(m interface{}, id uint) (err error) {
return OneByIdTx(DB, m, id)
}
// OneByIdTx 通过ID查询一条
func OneByIdTx(tx *gorm.DB, m interface{}, id uint) (err error) {
func OneById(tx *gorm.DB, m interface{}, id uint) (err error) {
if id == 0 {
return errors.New("主键ID不能为空")
}
@ -27,13 +22,8 @@ func OneByIdTx(tx *gorm.DB, m interface{}, id uint) (err error) {
return nil
}
// One 查询一条 使用m的有值属性作为条件, m必须是Model结构体. 条件为空时报错.
func One(m interface{}) (err error) {
return OneTx(DB, m)
}
// OneTx 查询一条 使用m的有值属性作为条件, m必须是Model结构体. 条件为空时报错.
func OneTx(tx *gorm.DB, m interface{}) error {
func One(tx *gorm.DB, m interface{}) error {
return oneTx(tx, m, "")
}
@ -59,23 +49,13 @@ func oneTx(tx *gorm.DB, m interface{}, Type string) (err error) {
return nil
}
// First 查询第一条 使用m的有值属性作为条件, m必须是Model结构体. 条件为空时报错.
func First(m interface{}) (err error) {
return FirstTx(DB, m)
}
// FirstTx 查询第一条 使用m的有值属性作为条件, m必须是Model结构体. 条件为空时报错.
func FirstTx(tx *gorm.DB, m interface{}) error {
func First(tx *gorm.DB, m interface{}) error {
return oneTx(tx, m, "first")
}
// Last 查询最后一条 使用m的有值属性作为条件, m必须是Model结构体. 条件为空时报错.
func Last(m interface{}) (err error) {
return LastTx(DB, m)
}
// LastTx 查询最后一条 使用m的有值属性作为条件, m必须是Model结构体. 条件为空时报错.
func LastTx(tx *gorm.DB, m interface{}) error {
func Last(tx *gorm.DB, m interface{}) error {
return oneTx(tx, m, "last")
}
@ -88,23 +68,13 @@ func LastTx(tx *gorm.DB, m interface{}) error {
// return nil
//}
// Create 新增
func Create(m interface{}) error {
return CreateTx(DB, m)
}
// CreateTx 带事务的, 新增
func CreateTx(tx *gorm.DB, m interface{}) error {
func Create(tx *gorm.DB, m interface{}) error {
return tx.Create(m).Error
}
// Update 更新一条数据的单个字段, m.ID必须有值
func Update(m interface{}, field string, value interface{}) error {
return UpdateTx(DB, m, field, value)
}
// UpdateTx 带事务的, 更新一条数据的单个字段, m.ID必须有值
func UpdateTx(tx *gorm.DB, m interface{}, field string, value interface{}) error {
func Update(tx *gorm.DB, m interface{}, field string, value interface{}) error {
db := tx.Model(m).Update(field, value)
if err := db.Error; err != nil {
return err
@ -115,17 +85,8 @@ func UpdateTx(tx *gorm.DB, m interface{}, field string, value interface{}) error
return nil
}
// Updates 更新一条数据的多个字段, m.ID必须有值
// FIXME: 只会更新非零值的字段!!
// 尽量不要让m和fields是同一个对象, fields只赋值需要更新的字段
// TODO 支持更新成零值, 支持结构体 map, 放在表结构体里? 不通过ID更新?
// TODO: fields 里不能ID有值, 否则也更新ID, 这是不行的
func Updates(m interface{}, fields interface{}) error {
return UpdatesTx(DB, m, fields)
}
// UpdatesTx 带事务的, 更新一条数据的多个字段, m.ID必须有值
func UpdatesTx(tx *gorm.DB, m interface{}, fields interface{}) error {
func Updates(tx *gorm.DB, m interface{}, fields interface{}) error {
db := tx.Model(m).Updates(fields)
if err := db.Error; err != nil {
return err
@ -136,15 +97,8 @@ func UpdatesTx(tx *gorm.DB, m interface{}, fields interface{}) error {
return nil
}
// Delete 删除, m.ID必须有值.
func Delete(m interface{}) error {
//func Delete(m interface{}, conds ...interface{}) error { // 不做这种支持, 由ma.DB.Delete()去支持
return DeleteTx(DB, m)
//return DeleteTx(nil, m, conds...)
}
// DeleteTx 删除, m.ID必须有值. tx不为空就是带事务的
func DeleteTx(tx *gorm.DB, m interface{}) error {
func Delete(tx *gorm.DB, m interface{}) error {
//func DeleteTx(tx *gorm.DB, m interface{}, conds ...interface{}) error {
db := tx.Delete(m)
//db := tx.Delete(m, conds...)
@ -157,13 +111,8 @@ func DeleteTx(tx *gorm.DB, m interface{}) error {
return nil
}
// List 查询数据列表, m是库表结构体, m的有值属性会作为查询条件, 且必须有条件, list里个体的类型可以与m的类型不同
func List(m interface{}, list interface{}) (err error) {
return ListTx(DB, m, list)
}
// ListTx 查询数据列表, m是库表结构体, m的有值属性会作为查询条件, 且必须有条件, list里个体的类型可以与m的类型不同
func ListTx(tx *gorm.DB, m interface{}, list interface{}) (err error) {
func List(tx *gorm.DB, m interface{}, list interface{}) (err error) {
if lxUtil.IsZeroOfUnderlyingType(m) {
return errors.New("条件不能为空")
}
@ -174,13 +123,8 @@ func ListTx(tx *gorm.DB, m interface{}, list interface{}) (err error) {
return nil
}
// ListAll 查询所有数据, m是库表结构体, m的有值属性不会作为查询条件, list里个体的类型可以与m的类型不同
func ListAll(m interface{}, list interface{}) (err error) {
return ListAllTx(DB, m, list)
}
// ListAllTx 查询所有数据, m是库表结构体, m的有值属性不会作为查询条件, list里个体的类型可以与m的类型不同
func ListAllTx(tx *gorm.DB, m interface{}, list interface{}) (err error) {
func ListAll(tx *gorm.DB, m interface{}, list interface{}) (err error) {
if tx.Model(m).Find(list).Error != nil {
// if tx.Where(m).Find(list).Error != nil {
return errors.New("查询出现错误")
@ -188,14 +132,8 @@ func ListAllTx(tx *gorm.DB, m interface{}, list interface{}) (err error) {
return nil
}
// Query 查询数据列表, 支持分页、总数、汇总, 见PaginationQuery属性. PaginationQuery不可为nil
func Query(m interface{}, list interface{}, q *PaginationQuery) (err error) {
QueryTx(DB, m, list, q)
return
}
// QueryTx 带事务的查询数据列表. tx为空就是不带事务, 否则认为是开启了事务. 你应该避免使用次方法, 而是使用Query
func QueryTx(tx *gorm.DB, m interface{}, list interface{}, q *PaginationQuery) (err error) {
func Query(tx *gorm.DB, m interface{}, list interface{}, q *PaginationQuery) (err error) {
// !! 关于会话 新的Statement实例 及复用 ref: https://gorm.io/zh_CN/docs/method_chaining.html
if q == nil {
@ -268,12 +206,7 @@ func QueryTx(tx *gorm.DB, m interface{}, list interface{}, q *PaginationQuery) (
//
//}
// SqlQuery 原生SQL查询列表, 支持分页. PaginationQuery可为nil
func SqlQuery(sql string, list interface{}, q *PaginationQuery, params ...interface{}) (err error) {
return SqlQueryTx(DB, sql, list, q, params...)
}
func SqlQueryTx(tx *gorm.DB, sql string, list interface{}, q *PaginationQuery, params ...interface{}) (err error) {
func SqlQuery(tx *gorm.DB, sql string, list interface{}, q *PaginationQuery, params ...interface{}) (err error) {
var builder strings.Builder
builder.WriteString(sql)