60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
|
|
"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"
|
|
)
|
|
|
|
type ApiMessageNumberLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
BaseLogic
|
|
}
|
|
|
|
func NewApiMessageNumberLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ApiMessageNumberLogic {
|
|
return &ApiMessageNumberLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
func (l *ApiMessageNumberLogic) ApiMessageNumber(in *message.ApiMessageNumberRequest) (*message.Response, error) {
|
|
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)
|
|
}
|
|
|
|
var info []dao.MessageInfo
|
|
w := modelbase.Params{
|
|
Eq: map[string]string{
|
|
"status": strconv.Itoa(utils.StatusFail),
|
|
"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 {
|
|
l.Errorf("count unread message failed: %v", err)
|
|
return l.fail(utils.Fail)
|
|
}
|
|
return l.ok(count)
|
|
}
|