33
33
import javax .ws .rs .ProcessingException ;
34
34
import javax .ws .rs .client .Client ;
35
35
import javax .ws .rs .core .Configuration ;
36
- import javax .ws .rs .core .MultivaluedMap ;
37
36
import javax .ws .rs .core .Response ;
38
37
39
38
import org .glassfish .jersey .client .ClientProperties ;
@@ -178,7 +177,7 @@ public AsyncHttpClient getGrizzlyClient() {
178
177
@ Override
179
178
public ClientResponse apply (final ClientRequest request ) {
180
179
final Request connectorRequest = translate (request );
181
- final Map <String , String > clientHeadersSnapshot = writeOutBoundHeaders (request . getHeaders () , connectorRequest );
180
+ final Map <String , String > clientHeadersSnapshot = writeOutBoundHeaders (request , connectorRequest );
182
181
183
182
final CompletableFuture <ClientResponse > responseFuture = new CompletableFuture <>();
184
183
final ByteBufferInputStream entityStream = new ByteBufferInputStream ();
@@ -201,7 +200,8 @@ public STATE onHeadersReceived(HttpResponseHeaders headers) throws Exception {
201
200
}
202
201
203
202
HeaderUtils .checkHeaderChanges (clientHeadersSnapshot , request .getHeaders (),
204
- GrizzlyConnector .this .getClass ().getName ());
203
+ GrizzlyConnector .this .getClass ().getName (),
204
+ request .getConfiguration ());
205
205
206
206
responseFuture .complete (translate (request , this .status , headers , entityStream ));
207
207
return STATE .CONTINUE ;
@@ -242,7 +242,7 @@ public void onThrowable(Throwable t) {
242
242
@ Override
243
243
public Future <?> apply (final ClientRequest request , final AsyncConnectorCallback callback ) {
244
244
final Request connectorRequest = translate (request );
245
- final Map <String , String > clientHeadersSnapshot = writeOutBoundHeaders (request . getHeaders () , connectorRequest );
245
+ final Map <String , String > clientHeadersSnapshot = writeOutBoundHeaders (request , connectorRequest );
246
246
final ByteBufferInputStream entityStream = new ByteBufferInputStream ();
247
247
final AtomicBoolean callbackInvoked = new AtomicBoolean (false );
248
248
@@ -264,7 +264,7 @@ public STATE onHeadersReceived(HttpResponseHeaders headers) throws Exception {
264
264
}
265
265
266
266
HeaderUtils .checkHeaderChanges (clientHeadersSnapshot , request .getHeaders (),
267
- GrizzlyConnector .this .getClass ().getName ());
267
+ GrizzlyConnector .this .getClass ().getName (), request . getConfiguration () );
268
268
// hand-off to grizzly's application thread pool for response processing
269
269
processResponse (new Runnable () {
270
270
@ Override
@@ -462,9 +462,10 @@ public OutputStream getOutputStream(int contentLength) throws IOException {
462
462
return baos .toByteArray ();
463
463
}
464
464
465
- private static Map <String , String > writeOutBoundHeaders (final MultivaluedMap < String , Object > headers ,
465
+ private static Map <String , String > writeOutBoundHeaders (final ClientRequest clientRequest ,
466
466
final com .ning .http .client .Request request ) {
467
- Map <String , String > stringHeaders = HeaderUtils .asStringHeadersSingleValue (headers );
467
+ Map <String , String > stringHeaders =
468
+ HeaderUtils .asStringHeadersSingleValue (clientRequest .getHeaders (), clientRequest .getConfiguration ());
468
469
469
470
for (Map .Entry <String , String > e : stringHeaders .entrySet ()) {
470
471
request .getHeaders ().add (e .getKey (), e .getValue ());
0 commit comments