pb/book.proto
syntax="proto3";
option go_package="./pb";
service BookService{
rpc SayName(BookRequest) returns (BookResponse);
}
message BookRequest{
string name=1;
}
message BookResponse{
string msg=1;
}
// go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.28
// go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.2
// export PATH="$PATH:$(go env GOPATH)/bin"
// protoc --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative book.proto
server.go
package main
import (
"bili/pb"
"context"
"log"
"net"
"google.golang.org/grpc"
)
type BookInfo struct {
pb.UnimplementedBookServiceServer
}
func (b *BookInfo) SayName(ctx context.Context, req *pb.BookRequest) (*pb.BookResponse, error) {
return &pb.BookResponse{Msg: "我要学习:" + req.Name}, nil
}
func main() {
server := grpc.NewServer()
pb.RegisterBookServiceServer(server, &BookInfo{})
listen, err := net.Listen("tcp", "0.0.0.0:8080")
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
if err = server.Serve(listen); err != nil {
log.Fatalf("failed to serve: %v", err)
}
}
client.go
package main
import (
"bili/pb"
"context"
"log"
"os"
"time"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
)
func main() {
conn, err := grpc.Dial("127.0.0.1:8080", grpc.WithTransportCredentials(insecure.NewCredentials()))
if err != nil {
log.Fatalf("did not connect: %v", err)
}
defer conn.Close()
client := pb.NewBookServiceClient(conn)
name := "Go语言从入门到精通"
if len(os.Args) > 1 {
name = os.Args[1]
}
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
// res, err := client.SayName(context.Background(), &pb.BookRequest{Name: name})
res, err := client.SayName(ctx, &pb.BookRequest{Name: name})
if err != nil {
log.Fatalf("could not SayName: %v", err)
}
log.Printf("Server Response: %s", res.Msg)
}