75 lines
1.9 KiB
Go
75 lines
1.9 KiB
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, 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, serviceKey string) (int64, error) {
|
|
switch serviceKey {
|
|
case "user_skin":
|
|
return createAnalysisBizUser(ctx, svcCtx.AnalysisSvcName)
|
|
default:
|
|
return utils.NumberZero, nil
|
|
}
|
|
}
|
|
|
|
func createAnalysisBizUser(ctx context.Context, svcName string) (int64, error) {
|
|
cli, err := svc.GetRpcClient(svcName)
|
|
if err != nil {
|
|
return utils.NumberZero, err
|
|
}
|
|
res, err := analysispb.NewAnalysisClient(cli.Conn()).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")
|
|
}
|
|
return res.Id, nil
|
|
}
|