Explorar el Código

fix: playground chat

feitianbubu hace 5 meses
padre
commit
a100baf57f
Se han modificado 2 ficheros con 14 adiciones y 3 borrados
  1. 2 3
      controller/playground.go
  2. 12 0
      middleware/auth.go

+ 2 - 3
controller/playground.go

@@ -65,9 +65,8 @@ func Playground(c *gin.Context) {
 		Group:  group,
 	}
 	_ = middleware.SetupContextForToken(c, tempToken)
-	_, err = getChannel(c, group, playgroundRequest.Model, 0)
-	if err != nil {
-		newAPIError = types.NewError(err, types.ErrorCodeGetChannelFailed)
+	_, newAPIError = getChannel(c, group, playgroundRequest.Model, 1)
+	if newAPIError != nil {
 		return
 	}
 	//middleware.SetupContextForSelectedChannel(c, channel, playgroundRequest.Model)

+ 12 - 0
middleware/auth.go

@@ -123,6 +123,18 @@ func authHelper(c *gin.Context, minRole int) {
 	c.Set("id", id)
 	c.Set("group", session.Get("group"))
 	c.Set("use_access_token", useAccessToken)
+
+	userCache, err := model.GetUserCache(id.(int))
+	if err != nil {
+		c.JSON(http.StatusOK, gin.H{
+			"success": false,
+			"message": err.Error(),
+		})
+		c.Abort()
+		return
+	}
+	userCache.WriteContext(c)
+
 	c.Next()
 }