position_repo.go 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
  1. package system
  2. import (
  3. "fmt"
  4. "time"
  5. "xorm.io/xorm"
  6. "xps/datamodels"
  7. "xps/viewmodels"
  8. )
  9. type PositionRepo struct {
  10. engine *xorm.Engine
  11. }
  12. func NewPositionRepo(engine *xorm.Engine) *PositionRepo {
  13. return &PositionRepo{
  14. engine: engine,
  15. }
  16. }
  17. func (d *PositionRepo) GetPage(m map[string]interface{}) (*viewmodels.PageResult, error) {
  18. ocId := m["ocId"].(int64)
  19. keywords := m["keywords"].(string)
  20. limit := m["limit"].(int)
  21. page := m["page"].(int)
  22. sqlSelect := ` SELECT A.* `
  23. sqlFrom := ` FROM s_position AS A `
  24. sqlWhere := fmt.Sprintf(` WHERE A.deleted_flag = 0 and A.oc_id = %d`, ocId)
  25. if keywords != "" {
  26. sqlWhere += ` AND A.position_name like ` + "'%" + keywords + "%'"
  27. }
  28. sqlCount := `SELECT COUNT(*) `
  29. total, err := d.engine.SQL(sqlCount + sqlFrom + sqlWhere).Count()
  30. if err != nil {
  31. return nil, err
  32. }
  33. var pageStart = (limit - 1) * limit
  34. sqlOrderBy := " ORDER BY position_id ASC "
  35. sqlLimit := fmt.Sprintf(" LIMIT %d OFFSET %d ", limit, pageStart)
  36. datalist := []viewmodels.PositionInfo{}
  37. err = d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy + sqlLimit).Find(&datalist)
  38. if err != nil {
  39. return nil, err
  40. }
  41. pageResult := &viewmodels.PageResult{}
  42. pageResult.Data = datalist
  43. pageResult.Total = total
  44. pageResult.PageSize = limit
  45. pageResult.Page = page
  46. return pageResult, nil
  47. }
  48. func (d *PositionRepo) GetList(m map[string]interface{}) ([]viewmodels.PositionInfo, error) {
  49. ocId := m["ocId"].(int64)
  50. keywords := m["keywords"].(string)
  51. sqlSelect := ` SELECT A.* `
  52. sqlFrom := ` FROM s_position AS A `
  53. sqlWhere := fmt.Sprintf(` WHERE A.deleted_flag = 0 and A.oc_id = %d`, ocId)
  54. if keywords != "" {
  55. sqlWhere += ` AND A.position_name like ` + "'%" + keywords + "%'"
  56. }
  57. sqlOrderBy := " ORDER BY position_id ASC "
  58. datalist := make([]viewmodels.PositionInfo, 0)
  59. err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist)
  60. if err != nil {
  61. return nil, err
  62. }
  63. return datalist, nil
  64. }
  65. func (d *PositionRepo) GetById(ocId int64, positionId int64) (*viewmodels.PositionInfo, error) {
  66. data := &viewmodels.PositionInfo{}
  67. data.OcId = ocId
  68. data.PositionId = positionId
  69. ok, err := d.engine.Get(data)
  70. if !ok || err != nil {
  71. return nil, err
  72. }
  73. return data, nil
  74. }
  75. func (d *PositionRepo) Create(data *datamodels.Position) error {
  76. data.CreatedAt = time.Now()
  77. _, err := d.engine.Insert(data)
  78. return err
  79. }
  80. func (d *PositionRepo) Update(data *datamodels.Position) error {
  81. ocId := data.OcId
  82. positionId := data.PositionId
  83. data.UpdatedAt = time.Now()
  84. _, err := d.engine.Where("oc_id = ?", ocId).Where("position_id =?", positionId).Update(data)
  85. return err
  86. }
  87. func (d *PositionRepo) Delete(m map[string]interface{}) error {
  88. ocId := m["ocId"].(int64)
  89. positionId := m["positionId"].(int64)
  90. deletedBy := m["deletedBy"].(int64)
  91. data := &datamodels.Position{}
  92. data.DeletedFlag = 1
  93. data.DeletedBy = deletedBy
  94. data.DeletedAt = time.Now()
  95. _, err := d.engine.Where("oc_id = ?", ocId).Where("position_id =?", positionId).Update(data)
  96. return err
  97. }