package repository import ( "errors" "fmt" "github.com/go-pg/pg/v10" "github.com/linmadan/egglib-go/persistent/pg/sqlbuilder" pgTransaction "github.com/linmadan/egglib-go/transaction/pg" "gitlab.fjmaimaimai.com/allied-creation/performance/pkg/domain" "gitlab.fjmaimaimai.com/allied-creation/performance/pkg/infrastructure/pg/models" "gitlab.fjmaimaimai.com/allied-creation/performance/pkg/utils" "time" ) type EvaluationProjectRepository struct { transactionContext *pgTransaction.TransactionContext } func NewEvaluationProjectRepository(transactionContext *pgTransaction.TransactionContext) *EvaluationProjectRepository { return &EvaluationProjectRepository{transactionContext: transactionContext} } func (repo *EvaluationProjectRepository) TransformToDomain(m *models.EvaluationProject) domain.EvaluationProject { return domain.EvaluationProject{ Id: m.Id, Name: m.Name, Describe: m.Describe, CompanyId: m.CompanyId, CycleId: m.CycleId, CreatorId: m.CreatorId, State: m.State, HrBp: m.HrBp, Pmp: m.Pmp, PmpIds: m.PmpIds, Recipients: m.Recipients, Template: m.Template, CreatedAt: m.CreatedAt, UpdatedAt: m.UpdatedAt, DeletedAt: m.DeletedAt, } } func (repo *EvaluationProjectRepository) TransformToModel(d *domain.EvaluationProject) models.EvaluationProject { return models.EvaluationProject{ Id: d.Id, Name: d.Name, Describe: d.Describe, CompanyId: d.CompanyId, CycleId: d.CycleId, CreatorId: d.CreatorId, State: d.State, HrBp: d.HrBp, Pmp: d.Pmp, PmpIds: d.PmpIds, Recipients: d.Recipients, Template: d.Template, CreatedAt: d.CreatedAt, UpdatedAt: d.UpdatedAt, DeletedAt: d.DeletedAt, } } func (repo *EvaluationProjectRepository) nextIdentify() (int64, error) { return utils.NewSnowflakeId() } func (repo *EvaluationProjectRepository) Insert(d *domain.EvaluationProject) (*domain.EvaluationProject, error) { var isCreate = d.Id == 0 if isCreate { id, err := repo.nextIdentify() if err != nil { return d, err } d.Id = id d.CreatedAt = time.Now() d.UpdatedAt = d.CreatedAt } else { d.UpdatedAt = time.Now() } m := repo.TransformToModel(d) tx := repo.transactionContext.PgTx var err error if isCreate { _, err = tx.Model(&m).Returning("id").Insert() } else { _, err = tx.Model(&m).Returning("id").WherePK().Update() // 更新和删除必须增加条件 } if err != nil { return nil, err } d.Id = m.Id return d, nil } func (repo *EvaluationProjectRepository) Remove(d *domain.EvaluationProject) (*domain.EvaluationProject, error) { tx := repo.transactionContext.PgTx nowTime := time.Now() m := repo.TransformToModel(d) m.DeletedAt = &nowTime if _, err := tx.Model(&m).WherePK().Update(); err != nil { return d, err } return d, nil } func (repo *EvaluationProjectRepository) FindOne(queryOptions map[string]interface{}) (*domain.EvaluationProject, error) { tx := repo.transactionContext.PgTx m := new(models.EvaluationProject) query := tx.Model(m) query.Where("deleted_at isnull") if id, ok := queryOptions["id"]; ok { query.Where("id=?", id) } if err := query.First(); err != nil { if errors.Is(err, pg.ErrNoRows) { return nil, fmt.Errorf("没有此资源") } else { return nil, err } } u := repo.TransformToDomain(m) return &u, nil } func (repo *EvaluationProjectRepository) Find(queryOptions map[string]interface{}, excludeColumns ...string) (int64, []*domain.EvaluationProject, error) { tx := repo.transactionContext.PgTx var m []*models.EvaluationProject query := tx.Model(&m).Where("deleted_at isnull") if len(excludeColumns) > 0 { query.ExcludeColumn(excludeColumns...) } if v, ok := queryOptions["name"]; ok { query.Where("name = ?", v) } if v, ok := queryOptions["companyId"]; ok { query.Where("company_id = ?", v) } if v, ok := queryOptions["cycleId"]; ok { query.Where("cycle_id = ?", v) } if v, ok := queryOptions["state"]; ok && v.(int) >= 0 { query.Where("state = ?", v) } if v, ok := queryOptions["pmpIds"]; ok { pmpIds := v.([]string) if len(pmpIds) > 0 { query.WhereGroup(func(query *pg.Query) (*pg.Query, error) { for i := range pmpIds { query.WhereOr("pmp_ids @> ?", pmpIds[i]) } return query, nil }) } } if v, ok := queryOptions["limit"].(int); ok { query.Limit(v) } if v, ok := queryOptions["offset"].(int); ok { query.Offset(v) } count, err := query.SelectAndCount() if err != nil { return 0, nil, err } var arrays []*domain.EvaluationProject for _, v := range m { d := repo.TransformToDomain(v) arrays = append(arrays, &d) } return int64(count), arrays, nil } func (repo *EvaluationProjectRepository) Count(queryOptions map[string]interface{}) (int64, error) { tx := repo.transactionContext.PgTx m := new(models.EvaluationProject) query := sqlbuilder.BuildQuery(tx.Model(m), queryOptions) query.Where("deleted_at isnull") if v, ok := queryOptions["id"]; ok { query.Where("id = ?", v) } if v, ok := queryOptions["notId"]; ok { query.Where("id != ?", v) } if v, ok := queryOptions["name"]; ok { query.Where("name = ?", v) } if v, ok := queryOptions["companyId"]; ok { query.Where("company_id = ?", v) } if v, ok := queryOptions["cycleId"]; ok { query.Where("cycle_id = ?", v) } count, err := query.Count() if err != nil { return 0, err } return int64(count), nil }