youtu_grpc/app/auth_service/internal/logic/code2_session_logic.go

package logic

import (
	"context"
	"gitea.youtukeji.com.cn/xiabin/youtu_grpc/app/auth_service/auth_service"
	"gitea.youtukeji.com.cn/xiabin/youtu_grpc/app/auth_service/internal/svc"
	"gitea.youtukeji.com.cn/xiabin/youtu_grpc/app/user_management/user_management"

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

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

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

func (l *Code2SessionLogic) Code2Session(in *auth_service.Code2SessionRequest) (*auth_service.Code2SessionResponse, error) {
	//获取cli，调用抖音或者微信api
	cli, err := l.svcCtx.Cli.Get(in.AppId)
	if err != nil {
		return nil, err
	}
	res, err := cli.Code2Session(l.ctx, in.Code, in.AnonymousCode)
	if err != nil {
		return nil, err
	}

	//获取appId
	ac := l.svcCtx.Query.AppAccount
	acModel, err := ac.WithContext(l.ctx).Where(ac.AppID.Eq(in.AppId)).First()
	if err != nil {
		return nil, err
	}

	//获取用户id
	au := l.svcCtx.Query.AppUser
	auModel, err := au.WithContext(l.ctx).Where(au.AppAccountID.Eq(acModel.ID), au.Openid.Eq(res.OpenID)).FirstOrCreate()
	if err != nil {
		return nil, err
	}

	//如果没有用户id，则创建一个空用户
	if auModel.UserID == nil {
		userId, err := l.svcCtx.UserManagerClient.CreateEmptyUser(l.ctx, &user_management.Empty{})
		if err != nil {
			return nil, err
		}
		auModel.UserID = &userId.UserId
	}

	return &auth_service.Code2SessionResponse{
		OpenId:  res.OpenID,
		UnionId: res.UnionID,
		UserId:  *auModel.UserID,
	}, nil
}