grpc streaming

gRPC_client/cmd/main.go

@@ -53,4 +53,30 @@ func main() {
 		log.Fatalf("error: %s", err)
 	}
 	fmt.Println(resp)
+
+	ctx, cancel = context.WithTimeout(context.Background(), time.Second)
+	defer cancel()
+	stream, err := c.Location(ctx)
+	if err != nil {
+		log.Fatalf("error: %s", err)
+	}
+
+	lreq := pb.LocationRequest{
+		DriverId: "007",
+		Location: &pb.Location{
+			Lat: 51.4871871,
+			Lng: -0.1266743,
+		},
+	}
+	for i := 0.000; i < 0.003; i += 0.001 {
+		lreq.Location.Lat += i
+		if err = stream.Send(&lreq); err != nil {
+			log.Fatalf("error: %s", err)
+		}
+	}
+	lresp, err := stream.CloseAndRecv()
+	if err != nil {
+		log.Fatalf("error: %s", err)
+	}
+	fmt.Println(lresp)
 }

gRPC_server/cmd/main.go

@@ -2,10 +2,12 @@ package main
 
 import (
 	"context"
+	"errors"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/metadata"
 	"google.golang.org/grpc/reflection"
 	"google.golang.org/grpc/status"
+	"io"
 	"log"
 	"net"
 
@@ -48,6 +50,31 @@ func (r *Rides) Start(ctx context.Context, req *pb.StartRequest) (*pb.StartRespo
 	return &resp, nil
 }
 
+func (r *Rides) Location(stream pb.Rides_LocationServer) error {
+	count := int64(0)
+	driverID := ""
+	for {
+		req, err := stream.Recv()
+		if errors.Is(err, io.EOF) {
+			break
+		}
+		if err != nil {
+			err = status.Errorf(codes.Internal, "can't read")
+			if err != nil {
+				return err
+			}
+		}
+		// TODO: update db
+		driverID = req.DriverId
+		count++
+	}
+	resp := pb.LocationResponse{
+		DriverId: driverID,
+		Count:    count,
+	}
+	return stream.SendAndClose(&resp)
+}
+
 type Rides struct {
 	pb.UnimplementedRidesServer
 }