sqlService.go 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. package service
  2. import (
  3. "app.yhyue.com/moapp/jybase/common"
  4. "app.yhyue.com/moapp/jybase/encrypt"
  5. elastic "app.yhyue.com/moapp/jybase/es"
  6. MC "bp.jydev.jianyu360.cn/BaseService/biCenter/api/common"
  7. "bp.jydev.jianyu360.cn/BaseService/biCenter/api/internal/types"
  8. "fmt"
  9. )
  10. type SqlService struct {
  11. }
  12. func (l SqlService) SqlManage(in *types.SqlManageReq) *[]map[string]interface{} {
  13. sqlData := MC.BaseMysql.FindOne("sql_manage", map[string]interface{}{
  14. "id": common.Int64All(in.Id),
  15. }, "", "")
  16. if sqlData == nil || len(*sqlData) == 0 {
  17. return &[]map[string]interface{}{}
  18. }
  19. dbType := (*sqlData)["db_type"]
  20. rs := &[]map[string]interface{}{}
  21. switch dbType {
  22. case "es":
  23. rs = elastic.Get(common.InterfaceToStr((*sqlData)["db_name"]), common.InterfaceToStr((*sqlData)["db_name"]), Sqlformat(common.InterfaceToStr((*sqlData)["content"]), *in))
  24. }
  25. for k, v := range *rs {
  26. (*rs)[k]["id"] = encrypt.EncodeArticleId2ByCheck(common.InterfaceToStr(v["id"]))
  27. delete((*rs)[k], "_id")
  28. }
  29. return rs
  30. }
  31. func Sqlformat(sqlStr string, sqlData types.SqlManageReq) string {
  32. data := []interface{}{}
  33. for _, v := range sqlData.Params {
  34. switch v.Type {
  35. case "string":
  36. data = append(data, v.Value)
  37. case "int":
  38. data = append(data, common.Int64All(v.Value))
  39. }
  40. }
  41. return fmt.Sprintf(sqlStr, data...)
  42. }