public static MockHttpRequest deepCopy(HttpRequest request) throws IOException {
   MockHttpRequest mock = new MockHttpRequest();
   mock.uri = request.getUri();
   mock.httpHeaders = (ResteasyHttpHeaders) request.getHttpHeaders();
   mock.httpMethod = request.getHttpMethod();
   byte[] bytes = ReadFromStream.readFromStream(1024, request.getInputStream());
   mock.inputStream = new ByteArrayInputStream(bytes);
   return mock;
 }
示例#2
0
  public ClientResponse cacheIfPossible(ClientRequest request, BaseClientResponse response)
      throws Exception {
    String cc = (String) response.getResponseHeaders().getFirst(HttpHeaders.CACHE_CONTROL);
    String exp = (String) response.getResponseHeaders().getFirst(HttpHeaders.EXPIRES);
    int expires = -1;

    if (cc != null) {
      CacheControl cacheControl = CacheControl.valueOf(cc);
      if (cacheControl.isNoCache()) return response;
      expires = cacheControl.getMaxAge();
    } else if (exp != null) {
      Date date = DateUtil.parseDate(exp);
      expires = (int) ((date.getTime() - System.currentTimeMillis()) / 1000);
    }

    String lastModified =
        (String) response.getResponseHeaders().getFirst(HttpHeaders.LAST_MODIFIED);
    String etag = (String) response.getResponseHeaders().getFirst(HttpHeaders.ETAG);

    String contentType = (String) response.getResponseHeaders().getFirst(HttpHeaders.CONTENT_TYPE);

    byte[] cached =
        ReadFromStream.readFromStream(1024, response.getStreamFactory().getInputStream());
    response.getStreamFactory().performReleaseConnection();

    MediaType mediaType = MediaType.valueOf(contentType);
    final BrowserCache.Entry entry =
        cache.put(
            request.getUri(),
            mediaType,
            response.getResponseHeaders(),
            cached,
            expires,
            etag,
            lastModified);

    response.setStreamFactory(new CachedStreamFactory(entry));

    return response;
  }
  /**
   * Store entity within a byte array input stream because we want to release the connection if a
   * ClientResponseFailure is thrown. Copy status and headers, but ignore all type information
   * stored in the ClientResponse.
   *
   * @param copy
   * @return
   */
  public static ClientResponse copyFromError(ClientResponse copy) {
    if (copy instanceof BaseClientResponse) {
      BaseClientResponse base = (BaseClientResponse) copy;
      InputStream is = null;
      if (copy.getResponseHeaders().containsKey("Content-Type")) {
        try {
          is = base.streamFactory.getInputStream();
          byte[] bytes = ReadFromStream.readFromStream(1024, is);
          is = new ByteArrayInputStream(bytes);
        } catch (IOException e) {
          // ignored
        }
      }
      final InputStream theIs = is;
      BaseClientResponse tmp =
          new BaseClientResponse(
              new BaseClientResponse.BaseClientResponseStreamFactory() {
                InputStream stream;

                public InputStream getInputStream() throws IOException {
                  return theIs;
                }

                public void performReleaseConnection() {}
              });
      tmp.executor = base.executor;
      tmp.status = base.status;
      tmp.providerFactory = base.providerFactory;
      tmp.headers = new CaseInsensitiveMap<String>();
      tmp.headers.putAll(base.headers);
      tmp.readerInterceptors = base.readerInterceptors;
      return tmp;
    } else {
      // Not sure how this codepath could ever be reached.
      InputStream is = null;
      if (copy.getResponseHeaders().containsKey("Content-Type")) {
        GenericType<byte[]> gt = new GenericType<byte[]>() {};
        try {
          byte[] bytes = (byte[]) copy.getEntity(gt);
          is = new ByteArrayInputStream(bytes);
        } catch (Exception ignore) {
        }
      }
      final InputStream theIs = is;
      BaseClientResponse tmp =
          new BaseClientResponse(
              new BaseClientResponse.BaseClientResponseStreamFactory() {
                InputStream stream;

                public InputStream getInputStream() throws IOException {
                  return theIs;
                }

                public void performReleaseConnection() {}
              });
      tmp.status = copy.getStatus();
      tmp.providerFactory = ResteasyProviderFactory.getInstance();
      tmp.headers = new CaseInsensitiveMap<String>();
      tmp.headers.putAll(copy.getResponseHeaders());
      tmp.headers.remove(
          "Content-Encoding"); // remove encoding because we will have already extracted byte array
      return tmp;
    }
  }