package repositories import ( "fmt" "time" "xorm.io/xorm" "xps/datamodels" "xps/viewmodels" ) type ModuleRepo struct { engine *xorm.Engine } func NewModuleRepo(engine *xorm.Engine) *ModuleRepo { return &ModuleRepo{ engine: engine, } } func (d *ModuleRepo) GetPage(m map[string]interface{}) (*viewmodels.PageResult, error) { limit := m["limit"].(int) page := m["page"].(int) sqlSelect := ` SELECT A.* ` sqlFrom := ` FROM api_module AS A ` sqlWhere := ` WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.module_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } sqlCount := ` SELECT COUNT(*) ` total, err1 := d.engine.SQL(sqlCount + sqlFrom + sqlWhere).Count(new(datamodels.ApiModule)) if err1 != nil { return nil, err1 } sqlOrderBy := " ORDER BY A.module_id ASC " pageStart := (page - 1) * limit sqlLimit := fmt.Sprintf(" LIMIT %d OFFSET %d ", limit, pageStart) datalist := make([]viewmodels.ApiModuleInfo, 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 *ModuleRepo) GetList(m map[string]interface{}) ([]viewmodels.ApiModuleInfo, error) { sqlSelect := ` SELECT A.* ` sqlFrom := ` FROM api_module AS A ` sqlWhere := ` WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.module_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } sqlOrderBy := " ORDER BY A.module_id ASC " datalist := make([]viewmodels.ApiModuleInfo, 0) err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist) if err != nil { return nil, err } else { return datalist, nil } } func (d *ModuleRepo) GetById(moduleId int64) (*viewmodels.ApiModuleInfo, error) { sqlSelect := ` SELECT A.* ` sqlFrom := ` FROM api_module AS A ` sqlWhere := ` WHERE A.deleted_flag = 0 ` sqlWhere += fmt.Sprintf(" AND A.module_id = %d", moduleId) data := &viewmodels.ApiModuleInfo{} has, err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere).Get(data) if !has || err != nil { return nil, err } else { return data, nil } } func (d *ModuleRepo) Create(data *datamodels.ApiModule) error { data.CreatedAt = time.Now() _, err := d.engine.Insert(data) return err } func (d *ModuleRepo) Update(data *datamodels.ApiModule) error { moduleId := data.ModuleId data.UpdatedAt = time.Now() _, err := d.engine. Where("module_id = ?", moduleId). Update(data) return err } func (d *ModuleRepo) Delete(m map[string]interface{}) error { moduleId := m["moduleId"].(int64) deletedBy := m["deletedBy"].(int64) data := &datamodels.ApiModule{} data.DeletedFlag = 1 data.DeletedBy = deletedBy data.DeletedAt = time.Now() _, err := d.engine. Where("module_id = ?", moduleId). Update(data) return err }