wangshan пре 2 месеци
родитељ
комит
2f61478202
1 измењених фајлова са 4 додато и 3 уклоњено
  1. 4 3
      services/sse/sse.go

+ 4 - 3
services/sse/sse.go

@@ -8,7 +8,6 @@ import (
 	"app.yhyue.com/moapp/message/model"
 	"app.yhyue.com/moapp/message/util"
 	"fmt"
-	"log"
 	"net/http"
 	"strings"
 	"time"
@@ -30,12 +29,12 @@ func (s *ServerSentRouter) Notify() {
 
 	sessVal := s.Session().GetMultiple()
 	userId := common.ObjToString(sessVal["userId"])
-	log.Println("message  userId:", userId)
+	logger.Info("message  userId:", userId)
 	if userId == "" {
 		userId = common.InterfaceToStr(s.Session().Id())
 	}
 	//userId = s.GetString("userId")
-	log.Println("params  userId:", s.GetString("userId"))
+	logger.Info("params  userId:", s.GetString("userId"))
 	if userId != "" {
 		// 创建用户专属的消息通道
 		messageChan := make(chan string)
@@ -57,8 +56,10 @@ func (s *ServerSentRouter) Notify() {
 		for {
 			select {
 			case msg := <-messageChan:
+				logger.Info("--msg--:", msg, "--userId--:", userId)
 				fmt.Fprintf(s.ResponseWriter, "data: %s\n\n", msg)
 			case <-ticker.C:
+				logger.Info("--heartbeat--:", userId)
 				fmt.Fprintf(s.ResponseWriter, ": heartbeat\n\n")
 			case <-s.Request.Context().Done():
 				return // 客户端断开连接