|
|
@@ -0,0 +1,124 @@
|
|
|
+package repositories
|
|
|
+
|
|
|
+import (
|
|
|
+ "fmt"
|
|
|
+ "time"
|
|
|
+ "xorm.io/xorm"
|
|
|
+ "xps/datamodels"
|
|
|
+ "xps/viewmodels"
|
|
|
+)
|
|
|
+
|
|
|
+type TaskRepo struct {
|
|
|
+ engine *xorm.Engine
|
|
|
+}
|
|
|
+
|
|
|
+func NewTaskRepo(engine *xorm.Engine) *TaskRepo {
|
|
|
+ return &TaskRepo{
|
|
|
+ engine: engine,
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (d *TaskRepo) GetPage(m map[string]interface{}) (*viewmodels.PageResult, error) {
|
|
|
+ ocId := m["ocId"].(int64)
|
|
|
+ limit := m["limit"].(int)
|
|
|
+ page := m["page"].(int)
|
|
|
+
|
|
|
+ sqlSelect := ` SELECT A.* `
|
|
|
+ sqlFrom := ` FROM task AS A `
|
|
|
+ sqlWhere := ` WHERE A.deleted_flag = 0 and A.oc_id = ` + fmt.Sprintf("%d", ocId)
|
|
|
+
|
|
|
+ if keywords, ok := m["keywords"]; ok {
|
|
|
+ sqlWhere += ` AND A.task_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'"
|
|
|
+ }
|
|
|
+
|
|
|
+ sqlCount := ` SELECT COUNT(*) `
|
|
|
+ total, err1 := d.engine.SQL(sqlCount + sqlFrom + sqlWhere).Count(new(datamodels.Task))
|
|
|
+ if err1 != nil {
|
|
|
+ return nil, err1
|
|
|
+ }
|
|
|
+
|
|
|
+ sqlOrderBy := " ORDER BY A.task_id DESC "
|
|
|
+ pageStart := (page - 1) * limit
|
|
|
+ sqlLimit := fmt.Sprintf(" LIMIT %d OFFSET %d ", limit, pageStart)
|
|
|
+
|
|
|
+ datalist := make([]datamodels.Task, 0)
|
|
|
+ err2 := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy + sqlLimit).Find(&datalist)
|
|
|
+ if err2 != nil {
|
|
|
+ return nil, err2
|
|
|
+ }
|
|
|
+
|
|
|
+ pageResult := &viewmodels.PageResult{}
|
|
|
+ pageResult.Data = datalist
|
|
|
+ pageResult.Total = total
|
|
|
+ pageResult.PageSize = limit
|
|
|
+ pageResult.Page = page
|
|
|
+ return pageResult, nil
|
|
|
+
|
|
|
+}
|
|
|
+
|
|
|
+func (d *TaskRepo) GetList(m map[string]interface{}) ([]viewmodels.TaskInfo, error) {
|
|
|
+ ocId := m["ocId"].(int64)
|
|
|
+
|
|
|
+ sqlSelect := ` SELECT A.* `
|
|
|
+ sqlFrom := ` FROM task AS A `
|
|
|
+ sqlWhere := ` WHERE A.deleted_flag = 0 AND A.oc_id =` + fmt.Sprintf("%d", ocId)
|
|
|
+ if keywords, ok := m["keywords"]; ok {
|
|
|
+ sqlWhere += ` AND A.task_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'"
|
|
|
+ }
|
|
|
+
|
|
|
+ sqlOrderBy := " ORDER BY A.task_id DESC "
|
|
|
+ datalist := make([]viewmodels.TaskInfo, 0)
|
|
|
+ err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ } else {
|
|
|
+ return datalist, nil
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (d *TaskRepo) GetById(ocId int64, taskId int64) (*viewmodels.TaskInfo, error) {
|
|
|
+ sqlSelect := ` SELECT A.* `
|
|
|
+ sqlFrom := ` FROM task AS A `
|
|
|
+ sqlWhere := ` WHERE A.deleted_flag = 0`
|
|
|
+ sqlWhere += fmt.Sprintf(" AND A.oc_id = %d", ocId)
|
|
|
+ sqlWhere += fmt.Sprintf(" AND A.task_id = %d", taskId)
|
|
|
+
|
|
|
+ data := &viewmodels.TaskInfo{}
|
|
|
+ has, err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere).Get(data)
|
|
|
+ if !has || err != nil {
|
|
|
+ return nil, err
|
|
|
+ } else {
|
|
|
+ return data, nil
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func (d *TaskRepo) Create(t *datamodels.Task) error {
|
|
|
+ t.CreatedAt = time.Now()
|
|
|
+ _, err := d.engine.Insert(t)
|
|
|
+ return err
|
|
|
+}
|
|
|
+
|
|
|
+func (d *TaskRepo) Update(t *datamodels.Task) error {
|
|
|
+ ocId := t.OcId
|
|
|
+ taskId := t.TaskId
|
|
|
+
|
|
|
+ t.UpdatedAt = time.Now()
|
|
|
+ _, err := d.engine.
|
|
|
+ Where("oc_id = ?", ocId).
|
|
|
+ Where("task_id = ?", taskId).
|
|
|
+ Update(t)
|
|
|
+ return err
|
|
|
+}
|
|
|
+
|
|
|
+func (d *TaskRepo) Delete(m map[string]interface{}) error {
|
|
|
+ ocId := m["ocId"].(int64)
|
|
|
+ taskId := m["taskId"].(int64)
|
|
|
+ deletedBy := m["deletedBy"].(int64)
|
|
|
+
|
|
|
+ data := &datamodels.Task{}
|
|
|
+ data.DeletedFlag = 1
|
|
|
+ data.DeletedBy = deletedBy
|
|
|
+ data.DeletedAt = time.Now()
|
|
|
+ _, err := d.engine.Where("oc_id = ?", ocId).Where("task_id = ?", taskId).Update(data)
|
|
|
+ return err
|
|
|
+}
|