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

package logic

import (
	"context"
	"strconv"

	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	analysispb "lone-services/rpc/analysis/pb"
	"lone-services/services/user/internal/dao"
	"lone-services/services/user/internal/model"
	"lone-services/services/user/internal/svc"

	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

func ensureClientBizUser(ctx context.Context, svcCtx *svc.ServiceContext, userId int64, clientCode string) error {
	var row dao.UserClient
	err := model.UserClientModel{}.Init().GetOne(modelbase.Params{
		Eq: map[string]string{
			"user_id":     strconv.FormatInt(userId, utils.NumberTen),
			"client_code": clientCode,
		},
	}, &row)
	if err != nil {
		return err
	}
	if row.Id < utils.NumberOne {
		return status.Error(codes.FailedPrecondition, "用户端关联不存在")
	}
	if row.BizId > utils.NumberZero {
		return nil
	}

	bizId, err := createBizUser(ctx, svcCtx, userId, clientCode)
	if err != nil {
		return err
	}
	if bizId < utils.NumberOne {
		return nil
	}

	_, err = model.UserClientModel{}.Init().Edit(modelbase.Params{
		Eq: map[string]string{"id": strconv.FormatInt(row.Id, utils.NumberTen)},
	}, map[string]interface{}{
		"biz_id": bizId,
	})
	return err
}

func createBizUser(ctx context.Context, svcCtx *svc.ServiceContext, userId int64, serviceKey string) (int64, error) {
	switch serviceKey {
	case "user_skin":
		return createAnalysisBizUser(ctx, svcCtx.AnalysisSvcName, userId)
	default:
		return utils.NumberZero, nil
	}
}

func createAnalysisBizUser(ctx context.Context, svcName string, userId int64) (int64, error) {
	cli, err := svc.GetRpcClient(svcName)
	if err != nil {
		return utils.NumberZero, err
	}
	client := analysispb.NewAnalysisClient(cli.Conn())
	res, err := client.InternalCreateUser(ctx, &analysispb.InternalCreateUserReq{})
	if err != nil {
		return utils.NumberZero, err
	}
	if res == nil || res.Id < utils.NumberOne {
		return utils.NumberZero, status.Error(codes.Internal, "analysis create user empty id")
	}
	bizId := res.Id
	var userRow dao.UserRow
	err = model.UserModel{}.Init().GetOne(modelbase.Params{
		Eq: map[string]string{"id": strconv.FormatInt(userId, utils.NumberTen)},
	}, &userRow)
	if err != nil {
		return utils.NumberZero, err
	}
	if userRow.Mobile != utils.StringEmpty {
		if _, err := client.InternalSetRegistered(ctx, &analysispb.InternalSetRegisteredReq{
			Id: bizId,
		}); err != nil {
			return utils.NumberZero, err
		}
	}
	return bizId, nil
}