Skip to content

Commit 908ce65

Browse files
committed
all: implement GetStream on wrappers of drpc.Stream
Change-Id: If030b544334ffcc86e1eada7689d5bd98942d714
1 parent da770f9 commit 908ce65

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

experiment/import.go

+1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ type streamWrapper struct {
1515
ctx context.Context
1616
}
1717

18+
func (s *streamWrapper) GetStream() drpc.Stream { return s.Stream }
1819
func (s *streamWrapper) Context() context.Context { return s.ctx }
1920

2021
// Handler implements drpc handler interface to extract experiment feature flag.

rpc/rpctracing/handler.go

+1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ type streamWrapper struct {
1818
ctx context.Context
1919
}
2020

21+
func (s *streamWrapper) GetStream() drpc.Stream { return s.Stream }
2122
func (s *streamWrapper) Context() context.Context { return s.ctx }
2223

2324
// ExtractorFunc extracts from some metadata the trace information.

0 commit comments

Comments
 (0)