72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package repositoryimpl
|
||
|
||
import (
|
||
"center-api/internal/biz/bo"
|
||
"center-api/internal/pkg/mapstructure"
|
||
"context"
|
||
)
|
||
|
||
type Query[Q any] interface {
|
||
Limit(limit int) *Q
|
||
Offset(offset int) *Q
|
||
CountX(ctx context.Context) int
|
||
Clone() *Q
|
||
}
|
||
|
||
// Base 基础类型，提供了 po与do之间的转换等共用方法，在do中嵌入此类型创建，大部分情况下可不用再重复写 ToEntity 和 ToEntities 方法
|
||
type Base[P, D, Q any] struct {
|
||
}
|
||
|
||
// ToEntity 转换成实体
|
||
// 支持基本类型的值对象
|
||
// 支持 edge 的实体转换，注意名称要保持一致，否则只能在组合类覆盖此此方法
|
||
func (b *Base[P, D, Q]) ToEntity(p *P) *D {
|
||
if p == nil {
|
||
return nil
|
||
}
|
||
var d *D
|
||
_ = mapstructure.Decode(p, &d)
|
||
return d
|
||
}
|
||
|
||
// ToEntities 转换成实体
|
||
// 支持基本类型的值对象
|
||
// 支持 edge 的实体转换，注意名称要保持一致，否则只能在组合类覆盖此此方法
|
||
func (b *Base[P, D, Q]) ToEntities(ps []*P) []*D {
|
||
if ps == nil {
|
||
return nil
|
||
}
|
||
var entities []*D
|
||
//e := b.MapstructureDeepDecode(ps, &entities)
|
||
_ = mapstructure.Decode(ps, &entities)
|
||
return entities
|
||
}
|
||
|
||
// SetPageByBo 设置分页
|
||
func (b *Base[P, D, Q]) SetPageByBo(query Query[Q], pageBo *bo.ReqPageBo) {
|
||
if query == nil || pageBo == nil {
|
||
return
|
||
}
|
||
if pageBo.Limit > 0 {
|
||
query.Limit(pageBo.Limit)
|
||
}
|
||
if pageBo.Page > 0 {
|
||
query.Offset(pageBo.GetOffset())
|
||
}
|
||
}
|
||
|
||
// QueryRespPage 获取分页数据
|
||
func (b *Base[P, D, Q]) QueryRespPage(ctx context.Context, query Query[Q], pageBo *bo.ReqPageBo) *bo.RespPageBo {
|
||
cloneQ := ((any)(query.Clone())).(Query[Q])
|
||
cloneQ.Offset(0)
|
||
cloneQ.Limit(0)
|
||
//查询总数
|
||
total := cloneQ.CountX(ctx)
|
||
resp := &bo.RespPageBo{Total: total}
|
||
if pageBo != nil {
|
||
resp.Page = pageBo.Page
|
||
resp.Limit = pageBo.Limit
|
||
}
|
||
return resp
|
||
}
|