package repositories import ( "fmt" "time" "xorm.io/xorm" "xps/datamodel" "xps/viewmodel" ) type BizRuleRepo struct { engine *xorm.Engine } func NewBizRuleRepo(engine *xorm.Engine) *BizRuleRepo { return &BizRuleRepo{ engine: engine, } } func (d *BizRuleRepo) GetPage(m map[string]interface{}) (*viewmodel.PageResult, error) { limit := m["limit"].(int) page := m["page"].(int) sqlSelect := ` SELECT A.* ` sqlFrom := ` FROM biz_rule AS A ` sqlWhere := ` WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.rule_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } sqlCount := ` SELECT COUNT(*) ` total, err1 := d.engine.SQL(sqlCount + sqlFrom + sqlWhere).Count(new(datamodel.BizRule)) if err1 != nil { return nil, err1 } sqlOrderBy := " ORDER BY A.rule_id ASC " pageStart := (page - 1) * limit sqlLimit := fmt.Sprintf(" LIMIT %d OFFSET %d ", limit, pageStart) datalist := make([]datamodel.BizRule, 0) err2 := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy + sqlLimit).Find(&datalist) if err2 != nil { return nil, err2 } pageResult := &viewmodel.PageResult{} pageResult.Data = datalist pageResult.Total = total pageResult.PageSize = limit pageResult.Page = page return pageResult, nil } func (d *BizRuleRepo) GetList(m map[string]interface{}) ([]viewmodel.BizRuleInfo, error) { sqlSelect := ` SELECT A.* ` sqlFrom := ` FROM biz_rule AS A ` sqlWhere := ` WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.rule_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } sqlOrderBy := " ORDER BY A.rule_id ASC " datalist := make([]viewmodel.BizRuleInfo, 0) err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist) if err != nil { return nil, err } return datalist, nil } func (d *BizRuleRepo) GetById(ruleId int64) (*viewmodel.BizRuleInfo, error) { sqlSelect := ` SELECT A.* ` sqlFrom := ` FROM biz_rule AS A ` sqlWhere := ` WHERE A.deleted_flag = 0 ` sqlWhere += fmt.Sprintf(" AND A.rule_id = %d ", ruleId) data := &viewmodel.BizRuleInfo{} has, err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere).Get(data) if !has || err != nil { return nil, err } else { return data, nil } } func (d *BizRuleRepo) Create(data *datamodel.BizRule) error { data.CreatedAt = time.Now() _, err := d.engine.Insert(data) return err } func (d *BizRuleRepo) Update(data *datamodel.BizRule) error { ruleId := data.RuleId data.UpdatedAt = time.Now() _, err := d.engine.Where("rule_id = ?", ruleId).Update(data) return err } func (d *BizRuleRepo) Delete(m map[string]interface{}) error { ruleId := m["ruleId"].(int64) deletedBy := m["deletedBy"].(int64) data := &datamodel.BizRule{} data.DeletedFlag = 1 data.DeletedBy = deletedBy data.DeletedAt = time.Now() _, err := d.engine.Where("rule_id = ?", ruleId).Update(data) return err }