package repository import ( "fmt" "time" "xorm.io/xorm" "xps/pkg/api/datamodel" "xps/pkg/api/viewmodel" "xps/pkg/base" ) type ApiRepo struct { engine *xorm.Engine } func NewApiRepo(engine *xorm.Engine) *ApiRepo { return &ApiRepo{ engine: engine, } } func (d *ApiRepo) GetPage(m map[string]interface{}) (*base.PageResult, error) { limit := m["limit"].(int) page := m["page"].(int) sqlSelect := ` SELECT A.*, B.biz_title, C.object_title, C.object_code, C.object_type, D.oam_title, D.oam_type, D.oam_code, D.sql_type, D.sql_statement, D.sql_count, D.oam_desc ` sqlFrom := ` FROM api AS A LEFT JOIN biz AS B ON (A.biz_id = B.biz_id) LEFT JOIN biz_object AS C ON (A.biz_id = C.biz_id AND A.object_id = C.object_id) LEFT JOIN biz_object_oam AS D ON (A.biz_id = D.biz_id AND A.object_id = D.object_id AND A.oam_id = D.oam_id)` sqlWhere := ` WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.api_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } if bizId, ok := m["bizId"]; ok { sqlWhere += fmt.Sprintf(" AND A.biz_id = %d ", bizId) } if objectId, ok := m["objectId"]; ok { sqlWhere += fmt.Sprintf(" AND A.object_id = %d ", objectId) } sqlCount := ` SELECT COUNT(*) ` total, err1 := d.engine.SQL(sqlCount + sqlFrom + sqlWhere).Count(new(viewmodel.ApiInfo)) if err1 != nil { return nil, err1 } sqlOrderBy := " ORDER BY A.module_id, A.api_id ASC " pageStart := (page - 1) * limit sqlLimit := fmt.Sprintf(" LIMIT %d OFFSET %d ", limit, pageStart) datalist := make([]viewmodel.ApiInfo, 0) err2 := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy + sqlLimit).Find(&datalist) if err2 != nil { return nil, err2 } pageResult := &base.PageResult{} pageResult.Data = datalist pageResult.Total = total pageResult.PageSize = limit pageResult.Page = page return pageResult, nil } func (d *ApiRepo) GetList(m map[string]interface{}) ([]viewmodel.ApiInfo, error) { sqlSelect := ` SELECT A.*, B.biz_title, C.object_title, C.object_code, C.object_type, D.oam_title, D.oam_type, D.oam_code, D.sql_type, D.sql_statement, D.sql_count, D.oam_desc ` sqlFrom := ` FROM api AS A LEFT JOIN biz AS B ON (A.biz_id = B.biz_id) LEFT JOIN biz_object AS C ON (A.biz_id = C.biz_id AND A.object_id = C.object_id) LEFT JOIN biz_object_oam AS D ON (A.biz_id = D.biz_id AND A.object_id = D.object_id AND A.oam_id = D.oam_id)` sqlWhere := ` WHERE A.deleted_flag = 0 ` if keywords, ok := m["keywords"]; ok { sqlWhere += ` AND A.api_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'" } if bizId, ok := m["bizId"]; ok { sqlWhere += fmt.Sprintf(" AND A.biz_id = %d ", bizId) } if objectId, ok := m["objectId"]; ok { sqlWhere += fmt.Sprintf(" AND A.object_id = %d ", objectId) } sqlOrderBy := " ORDER BY A.biz_id, A.object_id, A.api_id ASC" datalist := make([]viewmodel.ApiInfo, 0) err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist) if err != nil { return nil, err } else { return datalist, nil } } func (d *ApiRepo) GetAll() ([]viewmodel.ApiInfo, error) { sqlSelect := ` SELECT A.*, B.biz_title, C.object_title, C.object_code, C.object_type, D.oam_title, D.oam_type, D.oam_code, D.sql_type, D.sql_statement, D.sql_count, D.oam_desc ` sqlFrom := ` FROM api AS A LEFT JOIN biz AS B ON (A.biz_id = B.biz_id) LEFT JOIN biz_object AS C ON (A.biz_id = C.biz_id AND A.object_id = C.object_id) LEFT JOIN biz_object_oam AS D ON (A.biz_id = D.biz_id AND A.object_id = D.object_id AND A.oam_id = D.oam_id)` sqlWhere := ` WHERE A.deleted_flag = 0 ` sqlOrderBy := " ORDER BY A.biz_id, A.object_id, A.api_id ASC" datalist := make([]viewmodel.ApiInfo, 0) err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist) if err != nil { return nil, err } else { return datalist, nil } } func (d *ApiRepo) GetById(apiId int64) (*viewmodel.ApiInfo, error) { sqlSelect := ` SELECT A.*, B.biz_title, C.object_title, C.object_code, C.object_type, D.oam_title, D.oam_type, D.oam_code, D.sql_type, D.sql_statement, D.sql_count, D.oam_desc ` sqlFrom := ` FROM api AS A LEFT JOIN biz AS B ON (A.biz_id = B.biz_id) LEFT JOIN biz_object AS C ON (A.biz_id = C.biz_id AND A.object_id = C.object_id) LEFT JOIN biz_object_oam AS D ON (A.biz_id = D.biz_id AND A.object_id = D.object_id AND A.oam_id = D.oam_id)` sqlWhere := ` WHERE A.deleted_flag = 0 ` sqlWhere += fmt.Sprintf(" AND A.api_id = %d", apiId) data := &viewmodel.ApiInfo{} has, err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere).Get(data) if !has || err != nil { return nil, err } else { return data, nil } } func (d *ApiRepo) GetByCode(apiCode string) (*viewmodel.ApiInfo, error) { data := &viewmodel.ApiInfo{} ok, err := d.engine. Where("api_code = ?", apiCode). Get(data) if !ok && err == nil { return nil, err } else { return data, nil } } func (d *ApiRepo) Create(data *datamodel.Api) error { data.CreatedAt = time.Now() _, err := d.engine.Insert(data) return err } func (d *ApiRepo) Update(data *datamodel.Api) error { apiId := data.ApiId data.UpdatedAt = time.Now() _, err := d.engine. Where("api_id = ?", apiId). Update(data) return err } func (d *ApiRepo) Delete(m map[string]interface{}) error { apiId := m["apiId"].(int64) deletedBy := m["deletedBy"].(int64) data := &datamodel.Api{} data.DeletedFlag = 1 data.DeletedBy = deletedBy data.DeletedAt = time.Now() _, err := d.engine. Where("api_id = ?", apiId). Update(data) return err }