feat: internal product items

services/message/internal/logic/apiMessageNumberLogic.go

@@ -2,14 +2,15 @@ package logic
 
 import (
 	"context"
-	"lone-services/pkg/modelbase"
-	"lone-services/pkg/utils"
-	"lone-services/services/message/internal/dao"
-	"lone-services/services/message/internal/model"
 	"strconv"
 
-	"lone-services/rpc/message/pb"
+	"lone-services/pkg/modelbase"
+	"lone-services/pkg/utils"
+	message "lone-services/rpc/message/pb"
+	"lone-services/services/message/internal/dao"
+	"lone-services/services/message/internal/model"
 	"lone-services/services/message/internal/svc"
+	"lone-services/services/message/validator"
 
 	"github.com/zeromicro/go-zero/core/logx"
 )
@@ -30,8 +31,13 @@ func NewApiMessageNumberLogic(ctx context.Context, svcCtx *svc.ServiceContext) *
 }
 
 func (l *ApiMessageNumberLogic) ApiMessageNumber(in *message.ApiMessageNumberRequest) (*message.Response, error) {
-	adminInfo := utils.GetUserFromCtx(l.ctx)
-	if adminInfo.ID < utils.NumberOne {
+	var v validator.MessageNumberValidator
+	if fail := l.checkParams(in, &v); fail != nil {
+		return fail, nil
+	}
+
+	loginUser := utils.GetUserFromCtx(l.ctx)
+	if loginUser.ID < utils.NumberOne {
 		return l.fail(utils.ErrorNoLoginInfo)
 	}
 
@@ -39,14 +45,14 @@ func (l *ApiMessageNumberLogic) ApiMessageNumber(in *message.ApiMessageNumberReq
 	w := modelbase.Params{
 		Eq: map[string]string{
 			"status":    strconv.Itoa(utils.StatusFail),
-			"target_id": strconv.FormatInt(adminInfo.ID, utils.NumberTen),
+			"target_id": strconv.FormatInt(loginUser.ID, utils.NumberTen),
 			"type":      strconv.Itoa(int(in.Type)),
 		},
 	}
 
 	count, err := model.MessageModel{}.Init().Count(w, &info)
 	if err != nil {
-		utils.Logger.Error(err)
+		l.Errorf("count unread message failed: %v", err)
 		return l.fail(utils.Fail)
 	}
 	return l.ok(count)