api_log_repo.go 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. package repository
  2. import (
  3. "fmt"
  4. "xorm.io/xorm"
  5. "xps/pkg/api/datamodel"
  6. "xps/pkg/api/viewmodel"
  7. "xps/pkg/common"
  8. )
  9. type ApiLogRepo struct {
  10. engine *xorm.Engine
  11. }
  12. func NewApiLogRepo(engine *xorm.Engine) *ApiLogRepo {
  13. return &ApiLogRepo{
  14. engine: engine,
  15. }
  16. }
  17. func (d *ApiLogRepo) GetPage(m map[string]interface{}) (*common.PageResult, error) {
  18. limit := m["limit"].(int)
  19. page := m["page"].(int)
  20. sqlSelect := ` SELECT A.*,
  21. B.app_title,
  22. C.api_title `
  23. sqlFrom := ` FROM api_log AS A
  24. LEFT JOIN app AS B ON (A.app_id = B.app_id)
  25. LEFT JOIN api AS C ON (A.api_id = C.api_id)`
  26. sqlWhere := ` WHERE 1 = 1 `
  27. if apiId, ok := m["apiId"]; ok {
  28. sqlWhere += ` AND A.api_id = ` + fmt.Sprintf("%d", apiId)
  29. }
  30. if accountId, ok := m["accountId"]; ok {
  31. sqlWhere += ` AND A.account_id = ` + fmt.Sprintf("%d", accountId)
  32. }
  33. if keywords, ok := m["keywords"]; ok {
  34. sqlWhere += ` AND A.api_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'"
  35. }
  36. sqlCount := ` SELECT COUNT(*) `
  37. total, err1 := d.engine.SQL(sqlCount + sqlFrom + sqlWhere).Count(new(datamodel.ApiLog))
  38. if err1 != nil {
  39. return nil, err1
  40. }
  41. sqlOrderBy := " ORDER BY A.req_id ASC "
  42. pageStart := (page - 1) * limit
  43. sqlLimit := fmt.Sprintf(" LIMIT %d OFFSET %d ", limit, pageStart)
  44. datalist := make([]datamodel.ApiLog, 0)
  45. err2 := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy + sqlLimit).Find(&datalist)
  46. if err2 != nil {
  47. return nil, err2
  48. }
  49. pageResult := &common.PageResult{}
  50. pageResult.Data = datalist
  51. pageResult.Total = total
  52. pageResult.PageSize = limit
  53. pageResult.Page = page
  54. return pageResult, nil
  55. }
  56. func (d *ApiLogRepo) GetList(m map[string]interface{}) ([]viewmodel.ApiLogInfo, error) {
  57. sqlSelect := ` SELECT A.*,
  58. B.app_title,
  59. C.api_title `
  60. sqlFrom := ` FROM api_log AS A
  61. LEFT JOIN app AS B ON (A.app_id = B.app_id)
  62. LEFT JOIN api AS C ON (A.api_id = C.api_id)`
  63. sqlWhere := ` WHERE 1 = 1 `
  64. if apiId, ok := m["apiId"]; ok {
  65. sqlWhere += ` AND A.api_id = ` + fmt.Sprintf("%d", apiId)
  66. }
  67. if accountId, ok := m["accountId"]; ok {
  68. sqlWhere += ` AND A.account_id = ` + fmt.Sprintf("%d", accountId)
  69. }
  70. if keywords, ok := m["keywords"]; ok {
  71. sqlWhere += ` AND A.api_title like ` + "'%" + fmt.Sprintf("%s", keywords) + "%'"
  72. }
  73. sqlOrderBy := " ORDER BY A.req_id ASC "
  74. datalist := make([]viewmodel.ApiLogInfo, 0)
  75. err := d.engine.SQL(sqlSelect + sqlFrom + sqlWhere + sqlOrderBy).Find(&datalist)
  76. if err != nil {
  77. return nil, err
  78. } else {
  79. return datalist, nil
  80. }
  81. }
  82. func (d *ApiLogRepo) GetById(reqId int64) (*viewmodel.ApiLogInfo, error) {
  83. data := &viewmodel.ApiLogInfo{}
  84. ok, err := d.engine.
  85. Where("req_id = ?", reqId).
  86. Get(data)
  87. if !ok && err == nil {
  88. return nil, err
  89. } else {
  90. return data, nil
  91. }
  92. }
  93. func (d *ApiLogRepo) Create(data *datamodel.ApiLog) error {
  94. _, err := d.engine.Insert(data)
  95. return err
  96. }