register.go

package controller

import (
	"context"
	"log"
	"net/http"
	"time"

	"github.com/chaossat/tiktak/service/register/pb"
	"github.com/gin-gonic/gin"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

//注册并生成token
func Register(ctx *gin.Context) {
	//开启debug，观察性能瓶颈
	debugid, ok := <-DebugChan
	if ok {
		now := time.Now()
		log.Println("开始注册请求,操作ID:", debugid)
		defer log.Println("结束注册请求,操作ID:", debugid, "操作耗时：", time.Since(now))
	}

	username := ctx.Query("username")
	password := ctx.Query("password")
	//连接grpc服务
	grpcConn, err := grpc.Dial(":12345", grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		log.Println("连接grpc服务失败")
		ctx.JSON(200, gin.H{
			"status_code": 1,
			"status_msg":  "连接grpc服务失败",
		})
		return
	}
	defer grpcConn.Close()

	//初始化grpc客户端
	grpcClient := pb.NewRegisterClient(grpcConn)

	//创建并初始化registerrequest对象
	var register pb.DouyinUserRegisterRequest
	register.Username = &username
	register.Password = &password
	resp, err := grpcClient.Register(context.TODO(), &register)
	if err != nil {
		log.Println(err.Error())
		log.Println("调用远程服务错误")
		ctx.JSON(200, gin.H{
			"status_code": 1,
			"status_msg":  "调用远程服务错误",
		})
		return
	}
	ctx.JSON(http.StatusOK, resp)
}