token.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. package client
  2. import (
  3. "fmt"
  4. "git.sxidc.com/go-tools/api_binding/http_binding/response"
  5. "net/url"
  6. "strconv"
  7. )
  8. const (
  9. generateTokenUrl = "/ds/api/admin/token/generate"
  10. deleteTokenUrl = "/ds/api/admin/token/delete"
  11. getTokensUrl = "/ds/api/admin/token/query"
  12. )
  13. func (c *Client) GenerateToken(token string, baseUrl string, name string) (string, error) {
  14. fullUrl, err := url.JoinPath(baseUrl, generateTokenUrl)
  15. if err != nil {
  16. return "", err
  17. }
  18. resp := new(struct {
  19. response.MsgResponse
  20. Token string `json:"token"`
  21. })
  22. err = c.post(token, fullUrl, map[string]any{
  23. "name": name,
  24. }, resp)
  25. if err != nil {
  26. return "", err
  27. }
  28. if !resp.Success {
  29. return "", fmt.Errorf(resp.Msg)
  30. }
  31. return resp.Token, nil
  32. }
  33. func (c *Client) DeleteToken(token string, baseUrl string, name string) error {
  34. fullUrl, err := url.JoinPath(baseUrl, deleteTokenUrl)
  35. if err != nil {
  36. return err
  37. }
  38. resp := new(response.MsgResponse)
  39. err = c.post(token, fullUrl, map[string]any{
  40. "name": name,
  41. }, resp)
  42. if err != nil {
  43. return err
  44. }
  45. if !resp.Success {
  46. return fmt.Errorf(resp.Msg)
  47. }
  48. return nil
  49. }
  50. func (c *Client) GetTokens(token string, baseUrl string, name string, pageNo int, pageSize int) ([]TokenInfo, error) {
  51. fullUrl, err := url.JoinPath(baseUrl, getTokensUrl)
  52. if err != nil {
  53. return nil, err
  54. }
  55. resp := new(response.InfosResponse[TokenInfo])
  56. err = c.get(token, fullUrl, map[string]string{
  57. "name": name,
  58. "pageNo": strconv.Itoa(pageNo),
  59. "pageSize": strconv.Itoa(pageSize),
  60. }, resp)
  61. if err != nil {
  62. return nil, err
  63. }
  64. if !resp.Success {
  65. return nil, fmt.Errorf(resp.Msg)
  66. }
  67. return resp.Infos, nil
  68. }