git add proxy

Author: edwardhey

.DS_Store

@@ -4,12 +4,14 @@
 package proxy
 
 import (
+	"fmt"
 	"io"
 
 	"golang.org/x/net/context"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/transport"
+	// "google.golang.org/grpc/transport"
+	// "google.golang.org/grpc/transport"
 )
 
 var (
@@ -59,11 +61,12 @@ type handler struct {
 // It is invoked like any gRPC server stream and uses the gRPC server framing to get and receive bytes from the wire,
 // forwarding it to a ClientStream established against the relevant ClientConn.
 func (s *handler) handler(srv interface{}, serverStream grpc.ServerStream) error {
+	fmt.Println(serverStream.Context())
 	// little bit of gRPC internals never hurt anyone
-	lowLevelServerStream, ok := transport.StreamFromContext(serverStream.Context())
-	if !ok {
-		return grpc.Errorf(codes.Internal, "lowLevelServerStream not exists in context")
-	}
+	lowLevelServerStream := grpc.ServerTransportStreamFromContext(serverStream.Context())
+	// if !ok {
+	// 	return grpc.Errorf(codes.Internal, "lowLevelServerStream not exists in context")
+	// }
 	fullMethodName := lowLevelServerStream.Method()
 	// We require that the director's returned context inherits from the serverStream.Context().
 	outgoingCtx, backendConn, err := s.director(serverStream.Context(), fullMethodName)