mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-02 23:40:24 +08:00
62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
package persistence
|
|
|
|
import (
|
|
"fmt"
|
|
"mayfly-go/internal/devops/domain/entity"
|
|
"mayfly-go/internal/devops/domain/repository"
|
|
"mayfly-go/pkg/biz"
|
|
"mayfly-go/pkg/model"
|
|
)
|
|
|
|
type mongoRepo struct{}
|
|
|
|
var MongoDao repository.Mongo = &mongoRepo{}
|
|
|
|
// 分页获取数据库信息列表
|
|
func (d *mongoRepo) GetList(condition *entity.Mongo, pageParam *model.PageParam, toEntity interface{}, orderBy ...string) *model.PageResult {
|
|
sql := "SELECT d.* FROM t_mongo d JOIN t_project_member pm ON d.project_id = pm.project_id WHERE 1 = 1 "
|
|
if condition.CreatorId != 0 {
|
|
// 使用创建者id模拟项目成员id
|
|
sql = fmt.Sprintf("%s AND pm.account_id = %d", sql, condition.CreatorId)
|
|
}
|
|
if condition.ProjectId != 0 {
|
|
sql = fmt.Sprintf("%s AND d.project_id = %d", sql, condition.ProjectId)
|
|
}
|
|
if condition.EnvId != 0 {
|
|
sql = fmt.Sprintf("%s AND d.env_id = %d", sql, condition.EnvId)
|
|
}
|
|
sql = sql + " ORDER BY d.create_time DESC"
|
|
return model.GetPageBySql(sql, pageParam, toEntity)
|
|
}
|
|
|
|
func (d *mongoRepo) Count(condition *entity.Mongo) int64 {
|
|
return model.CountBy(condition)
|
|
}
|
|
|
|
// 根据条件获取
|
|
func (d *mongoRepo) Get(condition *entity.Mongo, cols ...string) error {
|
|
return model.GetBy(condition, cols...)
|
|
}
|
|
|
|
// 根据id获取
|
|
func (d *mongoRepo) GetById(id uint64, cols ...string) *entity.Mongo {
|
|
db := new(entity.Mongo)
|
|
if err := model.GetById(db, id, cols...); err != nil {
|
|
return nil
|
|
|
|
}
|
|
return db
|
|
}
|
|
|
|
func (d *mongoRepo) Insert(db *entity.Mongo) {
|
|
biz.ErrIsNil(model.Insert(db), "新增mongo信息失败")
|
|
}
|
|
|
|
func (d *mongoRepo) Update(db *entity.Mongo) {
|
|
biz.ErrIsNil(model.UpdateById(db), "更新mongo信息失败")
|
|
}
|
|
|
|
func (d *mongoRepo) Delete(id uint64) {
|
|
model.DeleteById(new(entity.Mongo), id)
|
|
}
|