package repository import ( "fmt" "time" "xorm.io/xorm" "xps/pkg/common" "xps/pkg/system/datamodel" "xps/pkg/system/viewmodel" ) type RoleRepo struct { engine *xorm.Engine } func NewRoleRepo(engine *xorm.Engine) *RoleRepo { return &RoleRepo{ engine: engine, } } func (d *RoleRepo) GetPage(m map[string]interface{}) (*common.PageResult, error) { limit := m["limit"].(int) page := m["page"].(int) sqlSelect := ` SELECT A.role_id , A.role_name, A.role_desc, A.role_type_id, B.role_type_title, A.sort_no, A.status, A.is_fixed ` sqlFrom := ` FROM s_role AS A LEFT JOIN s_role_type AS B ON (A.role_type_id = B.role_type_id ) ` sqlWhere := ` WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.role_name like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } sqlCount := `SELECT COUNT(*) ` total, err := d.engine.SQL(sqlCount + sqlFrom + sqlWhere).Count(new(datamodel.Role)) if err != nil { return nil, err } sqlOrderBy := " ORDER BY role_id ASC " pageStart := (limit - 1) * limit sqlLimit := fmt.Sprintf(" LIMIT %d OFFSET %d ", limit, pageStart) datalist := make([]viewmodel.RoleInfo, 0) err = d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy + sqlLimit).Find(&datalist) if err != nil { return nil, err } pageResult := &common.PageResult{ Total: total, PageSize: limit, Page: page, Data: datalist, } return pageResult, nil } func (d *RoleRepo) GetList(m map[string]interface{}) ([]viewmodel.RoleInfo, error) { sqlSelect := ` SELECT A.role_id , A.role_name, A.role_desc, A.role_type_id, B.role_type_title, A.sort_no, A.status, A.is_fixed ` sqlFrom := ` FROM s_role AS A LEFT JOIN s_role_type AS B ON (A.role_type_id = B.role_type_id )` sqlWhere := `WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.role_name like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } sqlOrderBy := " ORDER BY role_id ASC " datalist := make([]viewmodel.RoleInfo, 0) err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist) if err != nil { return nil, err } else { return datalist, nil } } func (d *RoleRepo) GetById(roleId int64) (*viewmodel.RoleInfo, error) { sqlSelect := ` SELECT A.role_id , A.role_name, A.role_desc, A.role_type_id, B.role_type_title, A.sort_no, A.status, A.is_fixed ` sqlFrom := ` FROM s_role AS A LEFT JOIN s_role_type AS B ON (A.role_type_id = B.role_type_id )` sqlWhere := `WHERE A.deleted_flag = 0 ` sqlWhere += fmt.Sprintf(" AND A.role_id = %d ", roleId) data := &viewmodel.RoleInfo{RoleId: roleId} has, err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere).Get(data) if !has || err != nil { return nil, err } else { return data, nil } } func (d *RoleRepo) Create(data *datamodel.Role) error { _, err := d.engine.Insert(data) return err } func (d *RoleRepo) Update(data *datamodel.Role) error { roleId := data.RoleId data.UpdatedAt = time.Now() _, err := d.engine.Where("role_id = ?", roleId).Update(data) return err } func (d *RoleRepo) Delete(m map[string]interface{}) error { roleId := m["roleId"].(int64) deletedBy := m["deletedBy"].(int64) data := &datamodel.Role{} data.DeletedFlag = 1 data.DeletedBy = deletedBy data.DeletedAt = time.Now() _, err := d.engine.Where("role_id = ?", roleId).Update(data) return err }