handler/im.go

@@ -12,7 +12,6 @@ import (
 	"strconv"
 	"sync"
 	"time"
-	"videoplayer/dao"
 	"videoplayer/proto"
 	"videoplayer/service"
 	"videoplayer/worker"
@@ -65,7 +64,6 @@ func SetUpIMGroup(router *gin.Engine) {
 	//接受邀请,确认好友关系
 	imGroup.POST("/accept_invite", AcceptInvite)
 	imGroup.POST("/create_group", CreateGroup)
-	imGroup.GET("/sse_msg", ServerSendMsg)
 }
 func generateRandomHexString(length int) (string, error) {
 	bytes := make([]byte, length/2) // 16字节的字符串需要32个十六进制字符，即16个字节
@@ -328,50 +326,3 @@ func SRMessage(c *gin.Context) {
 		}
 	}
 }
-
-func ServerSendMsg(c *gin.Context) {
-	//sse
-	c.Writer.Header().Set("Content-Type", "text/event-stream")
-	c.Writer.Header().Set("Cache-Control", "no-cache")
-	c.Writer.Header().Set("Connection", "keep-alive")
-	c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
-
-	id, _ := c.Get("id")
-	user_id := int(id.(float64))
-
-	//设置用户在线状态
-	worker.SetRedisWithExpire("user_"+strconv.Itoa(user_id)+"_status_v2", "1", time.Second*60)
-	//发送消息
-
-	key := "user_" + strconv.Itoa(user_id) + "_msg_ids"
-
-	for {
-		msg_id := worker.PopRedisListLeft(key)
-		if msg_id != "" {
-			msg_id_num, _ := strconv.ParseInt(msg_id, 10, 64)
-			msgs := dao.FindMessageByID(uint(msg_id_num))
-			if len(msgs) > 0 {
-				msg := msgs[0]
-				//发送消息
-				_, err := c.Writer.Write([]byte("data: " + msg.Msg + "\n\n"))
-				if err != nil {
-					worker.SetRedisWithExpire("user_"+strconv.Itoa(user_id)+"_status_v2", "0", time.Second*3600)
-					break
-				}
-			}
-		} else {
-			var msg proto.Message
-			msg.Type = "check"
-			msg.Msg = "check"
-			msg.From_user_id = -1
-			//发送心跳包
-			res3, _ := json.Marshal(msg)
-			_, err := c.Writer.Write([]byte("data: " + string(res3) + "\n\n"))
-			if err != nil {
-				worker.SetRedisWithExpire("user_"+strconv.Itoa(user_id)+"_status_v2", "0", time.Second*3600)
-				break
-			}
-		}
-		time.Sleep(time.Second * 1)
-	}
-}

service/imService.go

@@ -2,10 +2,7 @@ package service
 
 import (
 	"errors"
-	"strconv"
-	"time"
 	"videoplayer/dao"
-	"videoplayer/worker"
 )
 
 func CreateGeneralMessageService(from_id, to_id, msg_type int, content string) (error, uint) {
@@ -20,11 +17,6 @@ func CreateGeneralMessageService(from_id, to_id, msg_type int, content string) (
 			return errors.New("not a friend"), 0
 		}
 		err, id = dao.CreateSimpleMessage(from_id, to_id, content)
-		res := worker.GetRedis("user_" + strconv.Itoa(to_id) + "_status_v2")
-		if res == "1" {
-			//在线,存入redis
-			worker.PushRedisListWithExpire("user_"+strconv.Itoa(to_id)+"_msg_ids", strconv.Itoa(int(id)), time.Second*300)
-		}
 	case 2:
 		err, id = dao.CreateGeneralMessage(from_id, to_id, msg_type, 0, 0, content)
 	case 3: