server和client通过grpc建立连接

刺骨的言语ヽ痛彻心扉 2022-12-18 06:56 213阅读 0赞

###### 一、文件结构 ######

server端和client端分布在两个不同的集群里，可以通过下发mq消息，或者grpc连接传递消息，这里介绍通过grpc建立连接，第一步是创建proto文件，要传递的方法和变量都在这里定义

sendData/  
├── data.proto  
└── log

###### 二、创建data.proto文件 ######

注意，如果server端和client端不在同一集群，两边都要创建proto文件

syntax = "proto3";            //指定protobuf的版本
    
    package sendData;        //声明一个包名，一般与文件目录名相同
    
    // service里定义建立连接的方法sayHello 
    service Connect { 
      rpc sayHello (HelloRequest) returns (HelloReply) { }
    }
    
    // 定义传递结构体
    message HelloRequest { 
    	string name = 1;
    	int age = 2;
    	Data data = 3;     //传递的数据的顺序从上到下排列，类型也可以是结构体
    }
    
    message Data { 
        string phone = 1;
    }
    
    message HelloReply { 
        string message = 1;  // message在data.pb.go里会变成Message
    }

运行：protoc --go\_out=plugins=grpc:. data.proto

运行完毕后目录结构  
sendData/  
├── data.proto  
├── data.pb.go  
└── log

###### 三、创建client.go ######

package main
    
    import (
        "log"
        "os"
    
        pb "amber/sendData"
        "golang.org/x/net/context"
        "google.golang.org/grpc"
    )
    
    const (
        address = "localhost:50001"
    )
    
    func main() { 
        //client端主动发起grpc连接，dial对方
        conn, err := grpc.Dial(address, grpc.WithInsecure())  
        if err != nil { 
            log.Fatalf("dial to server with error : %v", err)
        }
    
        defer conn.Close()        //一定不忘忘记关闭连接
    
        c := pb.NewConnectClient(conn)
    	if c== nil { 
            log.Fatalf("connect to server with error : %v", err)
    	}
    	
    	//生成的pb.go里的结构体可以被外部引用
    	data:= &pb.Data{ Phone:"130-4472-0078"} 
    	
        resp, err := c.SayHello(context.Background(), &pb.HelloRequest{ Name: "amber",Data:data})
        if err != nil { 
            log.Fatalf("could not greet: %v", err)
        }
        
        //打印HelloReply里message的内容
        log.Println(resp.Message)
    }

###### 四、创建server.go ######

package main
    
    import (
        "log"
        "net"
    
        pb "amber/sendData"
        "google.golang.org/grpc"
        "golang.org/x/net/context"
    )
    
    const (
        PORT = ":50001"
    )
    
    type server struct { }   //定义空结构体，关联server服务
    
    // 重写SayHello方法
    func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) { 
        log.Println("request: ", in.Name)
        return &pb.HelloReply{ Message: "Hello " + in.Name}, nil
    }
    
    func main() { 
        lis, err := net.Listen("tcp", PORT) //port与client发起dial的一致
        if err != nil { 
            log.Fatalf("failed to listen: %v", err)
        }
    
        s := grpc.NewServer()
        pb.RegisterConnectServer(s, &server{ }) //注册服务到grpc的sever端，RegisterConnectServer是proto里service生成
    
        log.Println("rpc服务已经开启")
    
        s.Serve(lis) //建立连接，开始服务
    }