| 419 | } |
| 420 | |
| 421 | func (ui *unifiedInterceptor) WrapStreamingClient(next connect.StreamingClientFunc) connect.StreamingClientFunc { |
| 422 | return connect.StreamingClientFunc(func(ctx context.Context, spec connect.Spec) connect.StreamingClientConn { |
| 423 | conn := next(ctx, spec) |
| 424 | if token := ui.tokenMgr.getToken(); token != "" { |
| 425 | conn.RequestHeader().Set("Authorization", fmt.Sprintf("Bearer %s", token)) |
| 426 | } |
| 427 | return conn |
| 428 | }) |
| 429 | } |
| 430 | |
| 431 | func (*unifiedInterceptor) WrapStreamingHandler(next connect.StreamingHandlerFunc) connect.StreamingHandlerFunc { |
| 432 | return connect.StreamingHandlerFunc(func(ctx context.Context, conn connect.StreamingHandlerConn) error { |