lone-services/services/user/internal/logic/internalCheckUpdateLogic.go

package logic

import (
	"context"
	"strconv"

	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	"lone-services/pkg/validate"
	user "lone-services/rpc/user/pb"
	"lone-services/services/user/internal/dao"
	"lone-services/services/user/internal/model"
	"lone-services/services/user/internal/svc"
	"lone-services/services/user/validator"

	"github.com/zeromicro/go-zero/core/logx"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

type InternalCheckUpdateLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewInternalCheckUpdateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InternalCheckUpdateLogic {
	return &InternalCheckUpdateLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *InternalCheckUpdateLogic) InternalCheckUpdate(in *user.InternalCheckUpdateReq) (*user.InternalCheckUpdateData, error) {
	var req validator.InternalCheckUpdateValidator
	if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
		return nil, status.Error(codes.InvalidArgument, msg)
	}

	userId, findErr := findUserIdByBiz(l.svcCtx.DB, req.BizId, req.SubjectType)
	if findErr != nil {
		l.Errorf("internal check update find by biz: %v", findErr)
		return nil, status.Error(codes.Internal, utils.Fail.Msg)
	}
	if userId < utils.NumberOne {
		return nil, status.Error(codes.NotFound, utils.ErrorNotFund.Msg)
	}

	userModel := model.UserModel{}.Init()
	var row dao.UserRow
	if err := userModel.GetOne(modelbase.Params{
		Eq: map[string]string{"id": strconv.FormatInt(userId, utils.NumberTen)},
	}, &row); err != nil {
		l.Errorf("internal check update get user: %v", err)
		return nil, status.Error(codes.Internal, utils.Fail.Msg)
	}
	if row.Id < utils.NumberOne {
		return nil, status.Error(codes.NotFound, utils.ErrorNotFund.Msg)
	}

	encryptMobile, encErr := utils.EncryptPhone(req.Mobile)
	if encErr != nil {
		l.Errorf("internal check update encrypt: %v", encErr)
		return nil, status.Error(codes.Internal, utils.ErrorEncryptAesError.Msg)
	}

	if encryptMobile == row.Mobile {
		return &user.InternalCheckUpdateData{Result: true}, nil
	}

	var occupied dao.UserRow
	if err := userModel.GetOne(modelbase.Params{
		Eq: map[string]string{"mobile": encryptMobile},
	}, &occupied); err != nil {
		l.Errorf("internal check update: %v", err)
		return nil, status.Error(codes.Internal, utils.Fail.Msg)
	}

	if occupied.Id > utils.NumberZero && occupied.Id != row.Id {
		return &user.InternalCheckUpdateData{Result: false}, nil
	}
	return &user.InternalCheckUpdateData{Result: true}, nil
}