lone-services/services/analysis/internal/logic/analysis/internalSetRegisteredLogic.go

package analysislogic

import (
	"context"
	"strconv"

	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	analysis "lone-services/rpc/analysis/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"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

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

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

func (l *InternalSetRegisteredLogic) InternalSetRegistered(in *analysis.InternalSetRegisteredReq) (*analysis.InternalSetRegisteredData, error) {
	if in == nil || in.Id < utils.NumberOne {
		return nil, status.Error(codes.InvalidArgument, "用户ID不能为空")
	}

	row, err := common.LoadAnalysisUser(in.Id)
	if err != nil {
		l.Errorf("internal set registered get: %v", err)
		return nil, status.Error(codes.Internal, utils.Fail.Msg)
	}
	if row == nil {
		return nil, status.Error(codes.NotFound, utils.ErrorNotFund.Msg)
	}
	if row.IsRegistered == uint8(utils.StatusOk) {
		return &analysis.InternalSetRegisteredData{Ok: true}, nil
	}

	_, err = model.UserModel{}.Init().Edit(modelbase.Params{
		Eq: map[string]string{"id": strconv.FormatInt(in.Id, utils.NumberTen)},
	}, map[string]interface{}{
		"is_registered": uint8(utils.StatusOk),
	})
	if err != nil {
		l.Errorf("internal set registered edit: %v", err)
		return nil, status.Error(codes.Internal, utils.Fail.Msg)
	}

	return &analysis.InternalSetRegisteredData{Ok: true}, nil
}