From 63f81760ae9958d67337232011fe7408b4a4fe64 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Fri, 28 Mar 2025 15:21:57 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=A8=A1=E5=9E=8B=E3=80=81?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD=E6=9F=A5=E6=89=BE=E5=88=A4=E6=96=AD=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handler/im.go           | 6 +++---
 service/modelService.go | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/handler/im.go b/handler/im.go
index 7278029..114c904 100644
--- a/handler/im.go
+++ b/handler/im.go
@@ -199,11 +199,11 @@ func doReceiveGenChatMessage(userId int, sessionID *uint, data *proto.WSMessageR
 	var model dao.Model
 	//查看请求功能类型
 	models, funcs := service.FindFuncModelByFunction(data.Function, userId)
-	log.Println("find function model by function:", models, funcs)
-	if funcs == nil || len(funcs) == 0 {
+	//log.Println("find function model by function:", models, funcs)
+	if len(funcs) == 0 {
 		return errors.New("function not exist")
 	}
-	if models == nil || len(models) == 0 {
+	if len(models) == 0 {
 		return errors.New("model not exist")
 	}
 	model = models[0]
diff --git a/service/modelService.go b/service/modelService.go
index 8d546b6..2ef3a1c 100644
--- a/service/modelService.go
+++ b/service/modelService.go
@@ -112,7 +112,7 @@ func FindFuncModelByFunction(function string, userID int) ([]dao.Model, []dao.Fu
 	//先查找对应功能
 	funcModels := dao.FindFunctionModelByFunction(function)
 	if len(funcModels) == 0 {
-		return nil, nil
+		return []dao.Model{}, funcModels
 	}
 	modelID := funcModels[0].ModelID
 	//再查找对应模型