remove StreamMiddleware from middleware and replace options for clien…

…t stream interface with streamMiddleware
go-kratos · Sep 6, 2024 · 93484a9 · 93484a9
1 parent b65b7e2
commit 93484a9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/middleware/middleware.go b/middleware/middleware.go
@@ -19,5 +19,3 @@ func Chain(m ...Middleware) Middleware {
 		return next
 	}
 }
-
-type StreamMiddleware func(Handler) Handler
diff --git a/transport/grpc/client.go b/transport/grpc/client.go
@@ -133,6 +133,7 @@ type clientOptions struct {
 	timeout                time.Duration
 	discovery              registry.Discovery
 	middleware             []middleware.Middleware
+	streamMiddleware       []middleware.Middleware
 	ints                   []grpc.UnaryClientInterceptor
 	streamInts             []grpc.StreamClientInterceptor
 	grpcOpts               []grpc.DialOption
@@ -167,7 +168,7 @@ func dial(ctx context.Context, insecure bool, opts ...ClientOption) (*grpc.Clien
 		unaryClientInterceptor(options.middleware, options.timeout, options.filters),
 	}
 	sints := []grpc.StreamClientInterceptor{
-		streamClientInterceptor(options.middleware, options.filters),
+		streamClientInterceptor(options.streamMiddleware, options.filters),
 	}
 
 	if len(options.ints) > 0 {