Java ClientInvocation.writeRequestBody方法代码示例

import org.jboss.resteasy.client.jaxrs.internal.ClientInvocation; //导入方法依赖的package包/类
private MockHttpRequest createRequest(ClientInvocation request) {
  MockHttpRequest mockRequest =
      MockHttpRequest.create(request.getMethod(), request.getUri(), baseUri);

  if (request.getEntity() != null) {
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    request.getDelegatingOutputStream().setDelegate(baos);
    try {
      request.writeRequestBody(request.getEntityStream());
      baos.close();
    } catch (IOException ioe) {
      throw new RuntimeException(ioe);
    }
    mockRequest.setInputStream(new ByteArrayInputStream(baos.toByteArray()));
  }

  MultivaluedMap<String, String> requestHeaders = request.getHeaders().asMap();
  mockRequest.getMutableHeaders().putAll(requestHeaders);
  copyCookies(mockRequest, requestHeaders);

  return mockRequest;
}
 

import org.jboss.resteasy.client.jaxrs.internal.ClientInvocation; //导入方法依赖的package包/类
/**
 * Creates the request OutputStream, to be sent to the end Service invoked, as a
 * <a href="http://commons.apache.org/io/api-release/org/apache/commons/io/output/DeferredFileOutputStream.html"
 * >DeferredFileOutputStream</a>.
 *
 *
 * @param request -
 * @return - DeferredFileOutputStream with the ClientRequest written out per HTTP specification.
 * @throws IOException -
 */
private DeferredFileOutputStream writeRequestBodyToOutputStream(final ClientInvocation request)
    throws IOException {
  DeferredFileOutputStream memoryManagedOutStream =
      new DeferredFileOutputStream(
          this.fileUploadInMemoryThresholdLimit * getMemoryUnitMultiplier(),
          getTempfilePrefix(), ".tmp", this.fileUploadTempFileDir);
  request.getDelegatingOutputStream().setDelegate(memoryManagedOutStream);
  request.writeRequestBody(request.getEntityStream());
  memoryManagedOutStream.close();
  return memoryManagedOutStream;
}
 

import org.jboss.resteasy.client.jaxrs.internal.ClientInvocation; //导入方法依赖的package包/类
private RequestBody createRequestBody(final ClientInvocation request) {
  if (request.getEntity() == null) {
    return null;
  }

  // NOTE: this will invoke WriterInterceptors which can possibly change the request,
  // so it must be done first, before reading any header.
  final Buffer buffer = new Buffer();
  try {
    request.writeRequestBody(buffer.outputStream());
  } catch (IOException e) {
    throw new RuntimeException(e);
  }

  javax.ws.rs.core.MediaType mediaType = request.getHeaders().getMediaType();
  final MediaType contentType =
      (mediaType == null) ? null : MediaType.parse(mediaType.toString());

  return new RequestBody() {
    @Override
    public long contentLength() throws IOException {
      return buffer.size();
    }

    @Override
    public MediaType contentType() {
      return contentType;
    }

    @Override
    public void writeTo(BufferedSink sink) throws IOException {
      buffer.copyTo(sink.buffer(), 0, buffer.size());
    }
  };
}