具体文件目录看上一篇的grpc,这个demo演示的是服务端以流的形式推送给客户端
pb/hello.proto:
syntax = "proto3";
option go_package = "./;pb";
package pb;
service Greeter{
rpc GetStream (StreamReqData) returns (stream StreamResData) {}
}
message StreamReqData{
string data = 1;
}
message StreamResData{
string data = 1;
}
client/main.go:
package main
import (
"context"
pb "demo01/pb"
"log"
"google.golang.org/grpc"
_ "google.golang.org/grpc/balancer/grpclb"
)
const (
ADDRESS = "localhost:50051"
)
func main(){
conn, err := grpc.Dial(ADDRESS, grpc.WithInsecure())
if err != nil {
return
}
defer conn.Close()
c := pb.NewGreeterClient(conn)
reqstreamData := &pb.StreamReqData{Data:"aaa"}
res, _ := c.GetStream(context.Background(),reqstreamData)
for {
aa, err := res.Recv()
if err != nil {
log.Println(err)
return
}
log.Println(aa)
}
}
server/main.go:
package main
import (
pb "demo01/pb"
"fmt"
"net"
"time"
"google.golang.org/grpc"
)
const (
port = ":50051"
)
type server struct {
pb.UnimplementedGreeterServer
}
func (s *server) GetStream(req *pb.StreamReqData,res pb.Greeter_GetStreamServer) error{
i := 0
for {
i++
res.Send(&pb.StreamResData{Data: fmt.Sprintf("%v",time.Now().Unix())})
time.Sleep(time.Second)
if i > 10 {
break
}
}
return nil
}
func main(){
lis, err := net.Listen("tcp", port)
if err != nil {
return
}
s := grpc.NewServer()
pb.RegisterGreeterServer(s, &server{})
s.Serve(lis)
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)