package repository import ( "errors" "fmt" "time" "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" ) type EvaluationCycleRepository struct { transactionContext *pgTransaction.TransactionContext } func NewEvaluationCycleRepository(transactionContext *pgTransaction.TransactionContext) *EvaluationCycleRepository { return &EvaluationCycleRepository{transactionContext: transactionContext} } func (repo *EvaluationCycleRepository) TransformToDomain(m *models.EvaluationCycle) domain.EvaluationCycle { return domain.EvaluationCycle{ Id: m.Id, Name: m.Name, TimeStart: m.TimeStart, TimeEnd: m.TimeEnd, CompanyId: m.CompanyId, CreatorId: m.CreatorId, KpiCycle: m.KpiCycle, SummaryState: domain.ProjectSummaryState(m.SummaryState), CreatedAt: m.CreatedAt.Local(), UpdatedAt: m.UpdatedAt.Local(), DeletedAt: m.DeletedAt, } } func (repo *EvaluationCycleRepository) TransformToModel(d *domain.EvaluationCycle) models.EvaluationCycle { return models.EvaluationCycle{ Id: d.Id, Name: d.Name, TimeStart: d.TimeStart, TimeEnd: d.TimeEnd, CompanyId: d.CompanyId, CreatorId: d.CreatorId, KpiCycle: d.KpiCycle, SummaryState: int(d.SummaryState), CreatedAt: d.CreatedAt, UpdatedAt: d.UpdatedAt, DeletedAt: d.DeletedAt, } } func (repo *EvaluationCycleRepository) nextIdentify() (int64, error) { return utils.NewSnowflakeId() } func (repo *EvaluationCycleRepository) Insert(d *domain.EvaluationCycle) (*domain.EvaluationCycle, 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 *EvaluationCycleRepository) Remove(d *domain.EvaluationCycle) (*domain.EvaluationCycle, 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 *EvaluationCycleRepository) FindOne(queryOptions map[string]interface{}) (*domain.EvaluationCycle, error) { tx := repo.transactionContext.PgTx m := new(models.EvaluationCycle) 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 *EvaluationCycleRepository) Find(queryOptions map[string]interface{}) (int64, []*domain.EvaluationCycle, error) { tx := repo.transactionContext.PgTx var m []*models.EvaluationCycle query := tx.Model(&m).Where("deleted_at isnull") if v, ok := queryOptions["name"].(string); ok && len(v) > 0 { query.Where("name LIKE ?", v) } if v, ok := queryOptions["ids"]; ok { query.Where("id in (?)", pg.In(v)) } if v, ok := queryOptions["id"]; ok { query.Where("id=?", v) } if v, ok := queryOptions["companyId"]; ok { query.Where("company_id = ?", v) } if v, ok := queryOptions["limit"].(int64); ok { query.Limit(int(v)) } if v, ok := queryOptions["offset"].(int64); ok { query.Offset(int(v)) } if v, ok := queryOptions["timeStart"]; ok { t := v.(time.Time) query.Where("time_start<=?", t) } if v, ok := queryOptions["timeEnd"]; ok { t := v.(time.Time) query.Where("time_end>=?", t) } // 按创建时间降序 query.Order("created_at DESC") count, err := query.SelectAndCount() if err != nil { return 0, nil, err } var arrays []*domain.EvaluationCycle for _, v := range m { d := repo.TransformToDomain(v) arrays = append(arrays, &d) } return int64(count), arrays, nil } func (repo *EvaluationCycleRepository) Count(queryOptions map[string]interface{}) (int64, error) { tx := repo.transactionContext.PgTx m := new(models.EvaluationCycle) query := sqlbuilder.BuildQuery(tx.Model(m), queryOptions) query.Where("deleted_at isnull") if id, ok := queryOptions["id"]; ok { query.Where("id = ?", id) } if notId, ok := queryOptions["notId"]; ok { query.Where("id != ?", notId) } if v, ok := queryOptions["name"].(string); ok && len(v) > 0 { query.Where("name = ?", v) } if v, ok := queryOptions["companyId"]; ok { query.Where("company_id = ?", v) } count, err := query.Count() if err != nil { return 0, err } return int64(count), nil } // 获取已结束的周期 func (repo *EvaluationCycleRepository) FindCycleEnd(limit int) ([]*domain.EvaluationCycle, error) { tx := repo.transactionContext.PgTx var m []*models.EvaluationCycle query := tx.Model(&m). Where("deleted_at isnull"). Where("time_end<=?", time.Now()). Where("summary_state=0"). Limit(limit) err := query.Select() if err != nil { return nil, err } var arrays []*domain.EvaluationCycle for _, v := range m { d := repo.TransformToDomain(v) arrays = append(arrays, &d) } return arrays, nil }