gRPC 封装client

valisweet

已于 2022-11-14 23:27:05 修改

阅读量806

点赞数

分类专栏： Golang 文章标签： golang

于 2021-07-10 00:05:53 首次发布

本文链接：https://blog.youkuaiyun.com/javahuazaili/article/details/118616798

版权

Golang 专栏收录该内容

70 篇文章

订阅专栏

本文介绍了如何在Go语言中实现RPC客户端，通过`workload.WorkloadServiceClient`与服务器进行交互，并展示了初始化连接和管理器模块的实践。重点在于描述了`Dial`函数的使用和`InitConnections`的设置，以及`manager`模块中如何整合这些连接以提供API服务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

rpc-client/client.go



package rpc_clients

import (
	"context"
	"google.golang.org/grpc"
	"lch-tests/protobufs/workload"
	"log"
	"time"
)

var (
	WorkloadClient workload.WorkloadServiceClient
)

type Conn struct {
	clientConn  *grpc.ClientConn
	callTimeOut time.Duration
}

func InitConnections(addr string) {
	cc := GetConn(addr)
	WorkloadClient = workload.NewWorkloadServiceClient(cc)
}

func GetConn(addr string) *grpc.ClientConn {
	c := &Conn{
		callTimeOut: 3 * time.Second,
	}
	c.Dial(addr)
	return c.clientConn
}

func (c *Conn) Dial(addr string) error {
	ctx, cancel := context.WithTimeout(context.Background(), c.callTimeOut)
	defer cancel()

	var err error
	c.clientConn, err = grpc.DialContext(ctx, addr, grpc.WithInsecure())
	if err != nil {
		log.Printf("dial rpc error:%s", err.Error())
	}
	return nil
}

manager/manager.go



package manager

import (
	"context"

	"github.com/gin-gonic/gin"
	"lch-tests/handler"
	"lch-tests/kube"
	rpcClient "lch-tests/rpc-clients"
)

const (
	addr = "192.168.2.331:31878"
)

func Main(ctx context.Context) error {
	// 初始化kube配置信息
	kube.InitKube()

	// 初始化rpc连接
	rpcClient.InitConnections(addr)

	engine := gin.New()
	handler := &handler.Handler{}
	engine.GET("/home", handler.Home)
	return engine.Run()
}