一个基于 gin+ grpc + etcd 等框架开发的小栗子

技术分享 3年前 (2023-02-27) 0 999+

关注

一、标准的项目结构

首先我们看一个标准的项目结构是什么样子的，github 上给出的一个示例：golang-standards/project-layout

二、服务注册与发现流程

三、示例代码

项目地址：gRPC-GoWeb

1、服务注册

服务注册和发现都可参考 etcd官网的注册和发现流程，以下是项目的几个步骤的代码，详细的可下载项目查看。

1、连接etcd注册中心

func New(etcdAddr []string) (*EtcdRegister, error) {     log.Printf("开始连接etcd注册中心... n")     cli, err := clientv3.New(clientv3.Config{         Endpoints:   etcdAddr,         DialTimeout: 5 * time.Second,     })     if err != nil {         log.Printf("连接etcd注册中心失败,失败原因是: %v n", err)         return nil, err     }     log.Printf("连接etcd注册中心成功! n")     return &EtcdRegister{         cli: cli,     }, nil }

2、服务注册

func (etcdRegister *EtcdRegister) Register(srvName string, srvAddr string, ttl int64) error {     etcdRegister.srvName = srvName     etcdRegister.srvAddr = srvAddr     log.Printf("开始创建etcd端点管理器... n")     em, err := endpoints.NewManager(etcdRegister.cli, srvName)     if err != nil {         log.Printf("创建etcd端点管理器失败,失败原因是: %v n", err)         return err     }     etcdRegister.em = em     log.Printf("创建etcd端点管理器成功! n")     log.Printf("开始创建服务租期... n")     lease, err := etcdRegister.cli.Grant(context.TODO(), ttl)     if err != nil {         log.Printf("创建服务租期失败,失败原因是: %v", err)     }     etcdRegister.ttl = ttl     etcdRegister.leaseID = lease.ID     log.Printf("创建服务租期成功! n")     log.Printf("开始注册服务,服务名: %s,服务地址: %s n", srvName, srvAddr)     em.AddEndpoint(context.TODO(), fmt.Sprintf("%v/%v", srvName, srvAddr), endpoints.Endpoint{Addr: srvAddr}, clientv3.WithLease(lease.ID))     if err != nil {         log.Printf("注册服务失败! n")         return err     }     log.Printf("注册服务成功! n")     log.Printf("开始服务定期续租! n")     leaseChan, err := etcdRegister.cli.KeepAlive(context.TODO(), etcdRegister.leaseID)     if err != nil {         log.Printf("服务定期续租失败! n")         return err     }     etcdRegister.leaseChan = leaseChan     log.Printf("服务定期续租成功! n")     return nil }

3、提供服务(需要用到Protobuf)

func init() {     pb.RegisterUserServiceServer(server.GrpcServer, &UserService{}) }

2、服务发现

1、服务发现

func (etcdDiscovery *etcdDiscovery) Discovery(srvName string) (*grpc.ClientConn, error) {     target := fmt.Sprintf("etcd:///%s", srvName)     conn, err := grpc.Dial(target, grpc.WithResolvers(etcdDiscovery.resolver), grpc.WithInsecure())     if err != nil {         log.Printf("%s服务发现失败,原因是: %v n", srvName, err)         return nil, err     }     return conn, nil }

2、调用服务方法

func UserRegister(ginCtx *gin.Context) {     etcdDiscovery := serviceDiscovery.EtcdCenter     //获取用户服务     conn, _ := etcdDiscovery.Discovery("study-user-service")     userService := pb.NewUserServiceClient(conn)     data, err := userService.SayHello(ginCtx, &pb.Request{})     if err != nil {         log.Printf("调用用户服务出错了,原因是: %v n", err)     }     ginCtx.JSON(200, gin.H{         "message": data,     }) }