Midnightar commited on
Commit
d06400b
·
verified ·
1 Parent(s): 5caa4d0

Update main.go

Browse files
Files changed (1) hide show
  1. main.go +15 -7
main.go CHANGED
@@ -32,15 +32,15 @@ func main() {
32
  r.POST("/token/getNew", func(c *gin.Context) {
33
  var req TokenRequest
34
  if err := c.BindJSON(&req); err != nil {
35
- c.JSON(400, gin.H{"error": "invalid JSON"})
36
  return
37
  }
38
  if appID == "" || appCert == "" {
39
- c.JSON(500, gin.H{"error": "missing APP_ID or APP_CERTIFICATE"})
40
  return
41
  }
42
  if req.Channel == "" {
43
- c.JSON(400, gin.H{"error": "missing channel"})
44
  return
45
  }
46
  if req.Uid == "" {
@@ -57,12 +57,20 @@ func main() {
57
 
58
  expireTime := uint32(time.Now().Unix() + req.Expire)
59
  if uidInt, err := strconv.ParseUint(req.Uid, 10, 32); err == nil {
60
- token, _ := rtctokenbuilder.BuildTokenWithUid(appID, appCert, req.Channel, uint32(uidInt), role, expireTime)
61
- c.JSON(200, gin.H{"token": token})
 
 
 
 
62
  return
63
  }
64
- token, _ := rtctokenbuilder.BuildTokenWithUserAccount(appID, appCert, req.Channel, req.Uid, role, expireTime)
65
- c.JSON(200, gin.H{"token": token})
 
 
 
 
66
  })
67
 
68
  r.Run(":" + port)
 
32
  r.POST("/token/getNew", func(c *gin.Context) {
33
  var req TokenRequest
34
  if err := c.BindJSON(&req); err != nil {
35
+ c.JSON(http.StatusBadRequest, gin.H{"error": "invalid JSON"})
36
  return
37
  }
38
  if appID == "" || appCert == "" {
39
+ c.JSON(http.StatusInternalServerError, gin.H{"error": "missing APP_ID or APP_CERTIFICATE"})
40
  return
41
  }
42
  if req.Channel == "" {
43
+ c.JSON(http.StatusBadRequest, gin.H{"error": "missing channel"})
44
  return
45
  }
46
  if req.Uid == "" {
 
57
 
58
  expireTime := uint32(time.Now().Unix() + req.Expire)
59
  if uidInt, err := strconv.ParseUint(req.Uid, 10, 32); err == nil {
60
+ token, err := rtctokenbuilder.BuildTokenWithUid(appID, appCert, req.Channel, uint32(uidInt), role, expireTime)
61
+ if err != nil {
62
+ c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
63
+ return
64
+ }
65
+ c.JSON(http.StatusOK, gin.H{"token": token})
66
  return
67
  }
68
+ token, err := rtctokenbuilder.BuildTokenWithUserAccount(appID, appCert, req.Channel, req.Uid, role, expireTime)
69
+ if err != nil {
70
+ c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
71
+ return
72
+ }
73
+ c.JSON(http.StatusOK, gin.H{"token": token})
74
  })
75
 
76
  r.Run(":" + port)