ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

获取session添加类型获取

This commit is contained in:
junleea 2025-04-03 19:16:14 +08:00
parent e478c973ea
commit 2d7c14087d
3 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dao/im.go
View File

@ -49,12 +49,12 @@ func FindSessionByID(id int) Session {
} }
// 获取用户最新的创建时间会话列表 50个 // 获取用户最新的创建时间会话列表 50个
func FindSessionByUserID(userID int) []Session { func FindSessionByUserID(userID, sessionType int) []Session {
var sessions []Session var sessions []Session
if proto.Config.SERVER_SQL_LOG { if proto.Config.SERVER_SQL_LOG {
DB.Debug().Where("user_id = ?", userID).Order("created_at DESC").Limit(50).Find(&sessions) DB.Debug().Where("user_id = ? and type = ?", userID, sessionType).Order("created_at DESC").Limit(50).Find(&sessions)
} else { } else {
DB.Where("user_id = ?", userID).Order("created_at DESC").Limit(50).Find(&sessions) DB.Where("user_id = ? and type = ?", userID, sessionType).Order("created_at DESC").Limit(50).Find(&sessions)
} }
return sessions return sessions
} }

13
handler/session.go
View File

@ -16,10 +16,11 @@ func SetUpSessionGroup(router *gin.Engine) {
} }
type Session struct { type Session struct {
ID int `json:"id" form:"id"` ID int `json:"id" form:"id"`
UserID int `json:"user_id" form:"user_id"` UserID int `json:"user_id" form:"user_id"`
Name string `json:"name" form:"name"` Name string `json:"name" form:"name"`
Type string `json:"type" form:"type"` Type string `json:"type" form:"type"`
SessionType int `json:"session_type" form:"session_type"`
} }
func CreateSession(c *gin.Context) { func CreateSession(c *gin.Context) {
@ -46,14 +47,14 @@ func FindSession(c *gin.Context) {
if err := c.ShouldBind(&req); err == nil { if err := c.ShouldBind(&req); err == nil {
// 查找会话 // 查找会话
if req.Type == "ID" { if req.Type == "ID" {
err2, session := service.FindSessionByID(req.ID, userID) err2, session := service.FindSessionByID(req.ID, userID, req.SessionType)
if err2 != nil { if err2 != nil {
c.JSON(http.StatusOK, gin.H{"error": err2.Error(), "code": proto.SessionSearchFailed, "message": "failed"}) c.JSON(http.StatusOK, gin.H{"error": err2.Error(), "code": proto.SessionSearchFailed, "message": "failed"})
return return
} }
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": session}) c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": session})
} else if req.Type == "UserID" { } else if req.Type == "UserID" {
sessions := service.FindSessionByUserID(userID) sessions := service.FindSessionByUserID(userID, req.SessionType)
c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": sessions}) c.JSON(http.StatusOK, gin.H{"code": proto.SuccessCode, "message": "success", "data": sessions})
} else { } else {
c.JSON(http.StatusOK, gin.H{"error": "type error", "code": proto.SessionSearchFailed, "message": "failed"}) c.JSON(http.StatusOK, gin.H{"error": "type error", "code": proto.SessionSearchFailed, "message": "failed"})

6
service/imService.go
View File

@ -19,8 +19,8 @@ func CreateSession(userID, Type int, name string) (error, uint) {
} }
// 查找用户的会话列表 // 查找用户的会话列表
func FindSessionByUserID(userID int) []proto.SessionResponse { func FindSessionByUserID(userID, sessionType int) []proto.SessionResponse {
sessions := dao.FindSessionByUserID(userID) sessions := dao.FindSessionByUserID(userID, sessionType)
sessionsMsgCounts := dao.FindSessionMessageCountByUserID(userID) sessionsMsgCounts := dao.FindSessionMessageCountByUserID(userID)
sessionsTokens := dao.FindModelTotalTokensBySessionID(userID) sessionsTokens := dao.FindModelTotalTokensBySessionID(userID)
smap := make(map[int]int) smap := make(map[int]int)
@ -45,7 +45,7 @@ func FindSessionByUserID(userID int) []proto.SessionResponse {
} }
return res return res
} }
func FindSessionByID(id, userID int) (error, []dao.Session) { func FindSessionByID(id, userID, sessionType int) (error, []dao.Session) {
session := dao.FindSessionByID(id) session := dao.FindSessionByID(id)
var res []dao.Session var res []dao.Session
if session.UserID != userID { if session.UserID != userID {