92 lines
1.9 KiB
Go
92 lines
1.9 KiB
Go
|
package mySql
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
"reflect"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 因为有大量的是简单的通过where查询一条数据的，例如查询某个用户的资料，所以封装这个工具类简化操作，用法与query保持一致
|
|||
|
|
type QueryManager struct {
|
|||
|
|
tableName string
|
|||
|
|
wheres []*QueryManagerWhere
|
|||
|
|
errorText string
|
|||
|
|
error error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type QueryManagerWhere struct {
|
|||
|
|
colName string
|
|||
|
|
value interface{}
|
|||
|
|
emptyCheck bool
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewQueryManager(tableName string) *QueryManager {
|
|||
|
|
return &QueryManager{
|
|||
|
|
tableName: tableName,
|
|||
|
|
wheres: make([]*QueryManagerWhere, 0),
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *QueryManager) SetErrorText(text string) *QueryManager {
|
|||
|
|
t.errorText = text
|
|||
|
|
return t
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *QueryManager) Where(colName string, value interface{}) *QueryManager {
|
|||
|
|
t.wheres = append(t.wheres, &QueryManagerWhere{
|
|||
|
|
colName: colName,
|
|||
|
|
value: value,
|
|||
|
|
emptyCheck: true,
|
|||
|
|
})
|
|||
|
|
return t
|
|||
|
|
}
|
|||
|
|
func (t *QueryManager) WhereRow(colName string, value interface{}, emptyCheck bool) *QueryManager {
|
|||
|
|
t.wheres = append(t.wheres, &QueryManagerWhere{
|
|||
|
|
colName: colName,
|
|||
|
|
value: value,
|
|||
|
|
emptyCheck: emptyCheck,
|
|||
|
|
})
|
|||
|
|
return t
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *QueryManager) Take(data interface{}) error {
|
|||
|
|
if len(t.wheres) == 0 {
|
|||
|
|
return errors.New("not where")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for colName := range t.wheres {
|
|||
|
|
if isBlank(reflect.ValueOf(t.wheres[colName])) {
|
|||
|
|
//如果是空值
|
|||
|
|
return t.getEmptyError()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
query := NewQuery(t.tableName)
|
|||
|
|
for _, where := range t.wheres {
|
|||
|
|
if where.emptyCheck {
|
|||
|
|
//检查空值
|
|||
|
|
if isBlank(reflect.ValueOf(where.value)) {
|
|||
|
|
return t.getEmptyError()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
query.Where(where.colName, where.value)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err := query.Take(data)
|
|||
|
|
if err != nil {
|
|||
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|||
|
|
return t.getEmptyError()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *QueryManager) getEmptyError() error {
|
|||
|
|
if t.errorText != "" {
|
|||
|
|
return errors.New(t.errorText)
|
|||
|
|
} else {
|
|||
|
|
return errors.New("不存在")
|
|||
|
|
}
|
|||
|
|
}
|