lone-services/services/analysis/internal/logic/analysisuser/userStatusLogic.go

package analysisuserlogic

import (
	"context"
	"strconv"

	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	analysis "lone-services/rpc/analysis/pb"
	userpb "lone-services/rpc/user/pb"
	"lone-services/services/analysis/internal/logic/common"
	"lone-services/services/analysis/internal/model"
	"lone-services/services/analysis/internal/svc"

	"github.com/zeromicro/go-zero/core/logx"
)

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

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

func (l *UserStatusLogic) UserStatus(in *analysis.UserStatusReq) (*analysis.Response, error) {
	bizId, loginUser, fail := common.RequireAnalysisUserId(l.ctx)
	if fail != nil {
		return fail, nil
	}

	row, err := common.LoadAnalysisUser(bizId)
	if err != nil {
		l.Errorf("user status load: %v", err)
		return common.FailResponse(utils.Fail), nil
	}
	if row == nil {
		return common.FailResponse(utils.ErrorNotFund), nil
	}

	isRegistered := l.syncRegisteredFromUser(bizId, loginUser.ID, row.IsRegistered)

	status := utils.NumberOne
	if isRegistered == uint8(utils.StatusOk) {
		status = utils.NumberTwo
	}
	if row.IsFaceVerified == uint8(utils.StatusOk) {
		status = utils.NumberThree
	}
	if row.IsProfileCompleted == uint8(utils.StatusOk) {
		status = utils.NumberFour
	}

	return common.OkResponse(map[string]int{"status": status}), nil
}

func (l *UserStatusLogic) syncRegisteredFromUser(bizId, userId int64, isRegistered uint8) uint8 {
	if isRegistered == uint8(utils.StatusOk) {
		return isRegistered
	}
	if isRegistered != uint8(utils.StatusFail) {
		return isRegistered
	}

	cli, err := svc.GetRpcClient(l.svcCtx.UserSvcName)
	if err != nil {
		l.Errorf("user status get user rpc: %v", err)
		return isRegistered
	}
	info, err := userpb.NewUserClient(cli.Conn()).InternalGetUserInfo(l.ctx, &userpb.InternalGetUserInfoReq{
		UserId: userId,
	})
	if err != nil {
		l.Errorf("user status get user info: %v", err)
		return isRegistered
	}
	if info == nil || info.Mobile == utils.StringEmpty {
		return isRegistered
	}

	_, err = model.UserModel{}.Init().Edit(modelbase.Params{
		Eq: map[string]string{"id": strconv.FormatInt(bizId, utils.NumberTen)},
	}, map[string]interface{}{
		"is_registered": uint8(utils.NumberOne),
	})
	if err != nil {
		l.Errorf("user status set registered: %v", err)
		return isRegistered
	}
	return uint8(utils.StatusOk)
}