@Override
  protected HttpResponse invoke(Request nativeRequest) throws IOException, InterruptedException {
    OkHttpClient requestScopedClient = globalClient.clone();
    requestScopedClient.setProxy(proxyForURI.apply(nativeRequest.uri()));

    Response response = requestScopedClient.newCall(nativeRequest).execute();

    HttpResponse.Builder<?> builder = HttpResponse.builder();
    builder.statusCode(response.code());
    builder.message(response.message());

    Builder<String, String> headerBuilder = ImmutableMultimap.builder();
    Headers responseHeaders = response.headers();
    for (String header : responseHeaders.names()) {
      headerBuilder.putAll(header, responseHeaders.values(header));
    }

    ImmutableMultimap<String, String> headers = headerBuilder.build();

    if (response.code() == 204 && response.body() != null) {
      response.body().close();
    } else {
      Payload payload = newInputStreamPayload(response.body().byteStream());
      contentMetadataCodec.fromHeaders(payload.getContentMetadata(), headers);
      builder.payload(payload);
    }

    builder.headers(filterOutContentHeaders(headers));

    return builder.build();
  }
 private Multimap<Container, Episode> childrenOf(Iterable<Container> equivalentContainers) {
   Builder<Container, Episode> builder = ImmutableMultimap.builder();
   for (Container container : equivalentContainers) {
     builder.putAll(container, childrenOf(container));
   }
   return builder.build();
 }
Exemplo n.º 3
0
 public static <R extends HttpRequest> R indexMapOfIterableToFormValuesWithPrefix(
     R request, String prefix, String keySuffix, String valueSuffix, Object input) {
   checkArgument(
       checkNotNull(input, "input") instanceof Map<?, ?>,
       "this binder is only valid for Map<?,Iterable<?>>: " + input.getClass());
   Map<Object, Iterable<Object>> map = (Map<Object, Iterable<Object>>) input;
   Builder<String, String> builder = ImmutableMultimap.builder();
   int i = 1;
   for (Map.Entry<Object, Iterable<Object>> entry : map.entrySet()) {
     builder.put(
         prefix + "." + i + "." + keySuffix,
         checkNotNull(entry.getKey().toString(), keySuffix.toLowerCase() + "s[" + i + "]"));
     Iterable<Object> iterable = entry.getValue();
     if (!Iterables.isEmpty(iterable)) {
       int j = 1;
       for (Object v : iterable) {
         builder.put(prefix + "." + i + "." + valueSuffix + "." + j, v.toString());
         j++;
       }
     }
     i++;
   }
   ImmutableMultimap<String, String> forms = builder.build();
   return forms.size() == 0 ? request : (R) request.toBuilder().replaceFormParams(forms).build();
 }
 @SuppressWarnings("unchecked")
 @Override
 public <R extends HttpRequest> R bindToRequest(R request, Object input) {
   checkArgument(
       checkNotNull(input, "input") instanceof UserIdGroupPair,
       "this binder is only valid for UserIdGroupPair!");
   UserIdGroupPair pair = (UserIdGroupPair) input;
   Builder<String, String> builder = ImmutableMultimap.builder();
   builder.put("SourceSecurityGroupOwnerId", pair.getUserId());
   builder.put("SourceSecurityGroupName", pair.getGroupName());
   return (R) request.toBuilder().replaceFormParams(builder.build()).build();
 }
Exemplo n.º 5
0
 public static <R extends HttpRequest> R indexStringArrayToFormValuesWithPrefix(
     R request, String prefix, Object input) {
   checkArgument(
       checkNotNull(input, "input") instanceof String[],
       "this binder is only valid for String[] : " + input.getClass());
   String[] values = (String[]) input;
   Builder<String, String> builder = ImmutableMultimap.builder();
   for (int i = 0; i < values.length; i++) {
     builder.put(
         prefix + "." + (i + 1), checkNotNull(values[i], prefix.toLowerCase() + "s[" + i + "]"));
   }
   ImmutableMultimap<String, String> forms = builder.build();
   return forms.size() == 0 ? request : (R) request.toBuilder().replaceFormParams(forms).build();
 }
Exemplo n.º 6
0
 public static <R extends HttpRequest> R indexIterableToFormValuesWithPrefix(
     R request, String prefix, Object input) {
   checkArgument(
       checkNotNull(input, "input") instanceof Iterable<?>,
       "this binder is only valid for Iterable<?>: " + input.getClass());
   Iterable<?> values = (Iterable<?>) input;
   Builder<String, String> builder = ImmutableMultimap.builder();
   int i = 0;
   for (Object o : values) {
     builder.put(
         prefix + "." + (i++ + 1),
         checkNotNull(o.toString(), prefix.toLowerCase() + "s[" + i + "]"));
   }
   ImmutableMultimap<String, String> forms = builder.build();
   return forms.size() == 0 ? request : (R) request.toBuilder().replaceFormParams(forms).build();
 }
Exemplo n.º 7
0
 public static <R extends HttpRequest> R indexMapToFormValuesWithPrefix(
     R request, String prefix, String keySuffix, String valueSuffix, Object input) {
   checkArgument(
       checkNotNull(input, "input") instanceof Map<?, ?>,
       "this binder is only valid for Map<?,?>: " + input.getClass());
   Map<?, ?> map = (Map<?, ?>) input;
   Builder<String, String> builder = ImmutableMultimap.builder();
   int i = 1;
   for (Map.Entry<?, ?> e : map.entrySet()) {
     builder.put(
         prefix + "." + i + "." + keySuffix,
         checkNotNull(e.getKey().toString(), keySuffix.toLowerCase() + "s[" + i + "]"));
     if (e.getValue() != null) {
       builder.put(prefix + "." + i + "." + valueSuffix, e.getValue().toString());
     }
     i++;
   }
   ImmutableMultimap<String, String> forms = builder.build();
   return forms.size() == 0 ? request : (R) request.toBuilder().replaceFormParams(forms).build();
 }
  @SuppressWarnings("unchecked")
  @Override
  public <R extends HttpRequest> R bindToRequest(R request, Object input) {
    checkArgument(
        checkNotNull(input, "input") instanceof ObjectMetadata,
        "this binder is only valid for ObjectMetadata!");
    checkNotNull(request, "request");

    ObjectMetadata md = ObjectMetadata.class.cast(input);
    checkArgument(md.getKey() != null, "objectMetadata.getKey() must be set!");

    request = metadataPrefixer.bindToRequest(request, md.getUserMetadata());

    Builder<String, String> headers = ImmutableMultimap.builder();
    if (md.getCacheControl() != null) {
      headers.put(HttpHeaders.CACHE_CONTROL, md.getCacheControl());
    }

    if (md.getContentMetadata().getContentDisposition() != null) {
      headers.put("Content-Disposition", md.getContentMetadata().getContentDisposition());
    }

    if (md.getContentMetadata().getContentEncoding() != null) {
      headers.put("Content-Encoding", md.getContentMetadata().getContentEncoding());
    }

    String contentLanguage = md.getContentMetadata().getContentLanguage();
    if (contentLanguage != null) {
      headers.put(HttpHeaders.CONTENT_LANGUAGE, contentLanguage);
    }

    if (md.getContentMetadata().getContentType() != null) {
      headers.put(HttpHeaders.CONTENT_TYPE, md.getContentMetadata().getContentType());
    } else {
      headers.put(HttpHeaders.CONTENT_TYPE, "binary/octet-stream");
    }

    if (md.getContentMetadata().getContentMD5() != null) {
      headers.put("Content-MD5", base64().encode(md.getContentMetadata().getContentMD5()));
    }

    return (R) request.toBuilder().replaceHeaders(headers.build()).build();
  }
  @Override
  protected HttpResponse invoke(HttpURLConnection connection)
      throws IOException, InterruptedException {
    HttpResponse.Builder<?> builder = HttpResponse.builder();
    InputStream in = null;
    try {
      in = consumeOnClose(connection.getInputStream());
    } catch (IOException e) {
      in = bufferAndCloseStream(connection.getErrorStream());
    } catch (RuntimeException e) {
      closeQuietly(in);
      throw propagate(e);
    }

    int responseCode = connection.getResponseCode();
    if (responseCode == 204) {
      closeQuietly(in);
      in = null;
    }
    builder.statusCode(responseCode);
    builder.message(connection.getResponseMessage());

    Builder<String, String> headerBuilder = ImmutableMultimap.builder();
    for (Map.Entry<String, List<String>> entry : connection.getHeaderFields().entrySet()) {
      String header = entry.getKey();
      // HTTP message comes back as a header without a key
      if (header != null) headerBuilder.putAll(header, entry.getValue());
    }
    ImmutableMultimap<String, String> headers = headerBuilder.build();
    if (in != null) {
      Payload payload = newInputStreamPayload(in);
      contentMetadataCodec.fromHeaders(payload.getContentMetadata(), headers);
      builder.payload(payload);
    }
    builder.headers(filterOutContentHeaders(headers));
    return builder.build();
  }
Exemplo n.º 10
0
 /** @see #toOldSchool */
 @Deprecated
 public static <K, V> ImmutableMultimap<K, V> fromOldSchool(Map<K, Set<V>> in) {
   Builder<K, V> out = ImmutableMultimap.<K, V>builder();
   for (K type : in.keySet()) out.putAll(type, ImmutableSet.copyOf(in.get(type)));
   return out.build();
 }