golang-grpc流模式

pb/food.proto

syntax="proto3";

option go_package="./pb";

service FoodService{
    rpc SayName(FoodStreamRequest) returns (stream FoodStreamResponse); // 服务端流模式
    rpc PostName(stream FoodStreamRequest) returns (FoodStreamResponse); // 客户端流模式
    rpc FullStream(stream FoodStreamRequest) returns (stream FoodStreamResponse); // 双向流模式
}

message FoodStreamRequest{
    string name=1;
}

message FoodStreamResponse{
    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 food.proto

server.go

package main

import (
	"log"
	"net"
	"study/pb"
	"sync"
	"time"

	"google.golang.org/grpc"
)

type FoodInfo struct {
	pb.UnimplementedFoodServiceServer
}

func (f *FoodInfo) SayName(request *pb.FoodStreamRequest, server pb.FoodService_SayNameServer) error {
	log.Printf("SayName已请求")
	server.Send(&pb.FoodStreamResponse{Msg: "我要吃：" + request.Name})
	return nil
}
func (f *FoodInfo) PostName(server pb.FoodService_PostNameServer) error {
	for {
		recv, err := server.Recv()
		if err != nil {
			log.Fatalf("server.Recv err: %v", err)
			break
		}
		log.Printf("PostName recv: %v", recv)
	}
	return nil
}
func (f *FoodInfo) FullStream(server pb.FoodService_FullStreamServer) error {
	var wg sync.WaitGroup
	wg.Add(2)
	c := make(chan string, 5)
	go func() {
		defer wg.Done()
		for {
			item, err := server.Recv()
			if err != nil {
				log.Fatalf("server.FullStream err: %v", err)
			}
			c <- item.Name
			log.Printf("server.Recv: %v", item.Name)
			time.Sleep(1 * time.Second)
		}
	}()

	go func() {
		defer wg.Done()
		for {
			foodName := <-c
			err := server.Send(&pb.FoodStreamResponse{Msg: "server.Recv: " + foodName})
			if err != nil {
				log.Fatalf("server.FullStream err: %v", err)
			}
			time.Sleep(1 * time.Second)
		}
	}()

	wg.Wait()
	return nil
}

func main() {
	server := grpc.NewServer()
	pb.RegisterFoodServiceServer(server, &FoodInfo{})
	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 (
	"context"
	"log"
	"os"
	"strings"
	"study/pb"
	"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.NewFoodServiceClient(conn)

	// name := "鱼香肉丝"
	// if len(os.Args) > 1 {
	// 	name = os.Args[1]
	// }
	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
	defer cancel()

	// 服务端流模式
	// res, err := client.SayName(ctx, &pb.FoodStreamRequest{Name: name})
	// if err != nil {
	// 	log.Fatalf("could not SayName: %v", err)
	// }
	// for {
	// 	res, err := res.Recv()
	// 	if err != nil {
	// 		log.Fatalf("res.Recv err: %v", err)
	// 		break
	// 	}
	// 	log.Printf("Server Response: %s", res.Msg)
	// }

	// 客户端流模式
	postNameClient, err := client.PostName(ctx)
	if err != nil {
		log.Fatalf("could not PostName: %v", err)
	}
	foods := []string{"鱼香肉丝", "宫保鸡丁", "辣子鸡"}
	if len(os.Args) > 1 {
		foods = strings.Split(",", os.Args[1])
	}
	for _, item := range foods {
		log.Printf("我要吃: %v", item)
		err = postNameClient.Send(&pb.FoodStreamRequest{Name: item})
		time.Sleep(1 * time.Second)
		if err != nil {
			log.Printf("err: %v", err)
			break
		}
	}

	// 双向流模式
	// fullClient, err := client.FullStream(ctx)
	// if err != nil {
	// 	log.Fatalf("could not PostName: %v", err)
	// 	return
	// }
	// foods := []string{"鱼香肉丝", "宫保鸡丁", "辣子鸡"}
	// if len(os.Args) > 1 {
	// 	foods = strings.Split(",", os.Args[1])
	// }

	// var wg sync.WaitGroup
	// wg.Add(2)

	// go func() {
	// 	defer wg.Done()
	// 	for {
	// 		item, err := fullClient.Recv()
	// 		if err != nil {
	// 			log.Fatalf("err: %v", err)
	// 			break
	// 		}
	// 		log.Printf("fullClient.Recv: %v", item.Msg)
	// 	}
	// }()

	// go func(s []string) {
	// 	defer wg.Done()
	// 	for _, item := range s {
	// 		log.Printf("fullClient.Send: %v", item)
	// 		err := fullClient.Send(&pb.FoodStreamRequest{Name: item})
	// 		if err != nil {
	// 			log.Fatalf("err: %v", err)
	// 			break
	// 		}
	// 	}
	// }(foods)

	// wg.Wait()
}