go server和client通过grpc建立连接

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
}

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)
}

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) //建立连接，开始服务
}