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 //再查找对应模型