zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: support WithStreamClientInterceptor for zrpc clients (#1907)

Browse Source 
* feat: support WithStreamClientInterceptor for zrpc clients

* fix: data race
This commit is contained in:
Kevin Wan
2022-05-14 19:58:17 +08:00
committed by GitHub
parent 95282edb78
commit e80a64fa67
6 changed files with 24 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
zrpc/internal/client.go
View File

@@ -131,6 +131,13 @@ func WithNonBlock() ClientOption {
}
}
// WithStreamClientInterceptor returns a func to customize a ClientOptions with given interceptor.
func WithStreamClientInterceptor(interceptor grpc.StreamClientInterceptor) ClientOption {
return func(options *ClientOptions) {
options.DialOptions = append(options.DialOptions, WithStreamClientInterceptors(interceptor))
}
}
// WithTimeout returns a func to customize a ClientOptions with given timeout.
func WithTimeout(timeout time.Duration) ClientOption {
return func(options *ClientOptions) {