mirror of https://github.com/apache/jclouds.git
Moved HttpMessage.getLowercaseHeaders() to Multimaps2.transformKeys() and
org.jclouds.functions.ToLowerCase for better reuse. Added org.jclouds.util.Multimaps2Test for unit test.
This commit is contained in:
parent
8d9499b3c1
commit
27dee0bfd9
|
@ -0,0 +1,39 @@
|
||||||
|
/**
|
||||||
|
* Licensed to jclouds, Inc. (jclouds) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. jclouds licenses this file
|
||||||
|
* to you under the Apache License, Version 2.0 (the
|
||||||
|
* "License"); you may not use this file except in compliance
|
||||||
|
* with the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing,
|
||||||
|
* software distributed under the License is distributed on an
|
||||||
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||||
|
* KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations
|
||||||
|
* under the License.
|
||||||
|
*/
|
||||||
|
package org.jclouds.functions;
|
||||||
|
|
||||||
|
import static com.google.common.base.Preconditions.checkNotNull;
|
||||||
|
|
||||||
|
import javax.inject.Singleton;
|
||||||
|
|
||||||
|
import com.google.common.base.Function;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Everett Toews
|
||||||
|
*/
|
||||||
|
@Singleton
|
||||||
|
public class ToLowerCase implements Function<String, String> {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String apply(String input) {
|
||||||
|
checkNotNull(input, "input cannot be null");
|
||||||
|
return input.toLowerCase();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -24,6 +24,7 @@ import java.io.File;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
|
||||||
|
import org.jclouds.functions.ToLowerCase;
|
||||||
import org.jclouds.http.internal.PayloadEnclosingImpl;
|
import org.jclouds.http.internal.PayloadEnclosingImpl;
|
||||||
import org.jclouds.io.Payload;
|
import org.jclouds.io.Payload;
|
||||||
import org.jclouds.io.Payloads;
|
import org.jclouds.io.Payloads;
|
||||||
|
@ -32,7 +33,6 @@ import org.jclouds.util.Multimaps2;
|
||||||
|
|
||||||
import com.google.common.base.Objects;
|
import com.google.common.base.Objects;
|
||||||
import com.google.common.base.Objects.ToStringHelper;
|
import com.google.common.base.Objects.ToStringHelper;
|
||||||
import com.google.common.collect.HashMultimap;
|
|
||||||
import com.google.common.collect.ImmutableMultimap;
|
import com.google.common.collect.ImmutableMultimap;
|
||||||
import com.google.common.collect.Multimap;
|
import com.google.common.collect.Multimap;
|
||||||
|
|
||||||
|
@ -186,25 +186,15 @@ public class HttpMessage extends PayloadEnclosingImpl {
|
||||||
return headers;
|
return headers;
|
||||||
}
|
}
|
||||||
|
|
||||||
private Multimap<String, String> getLowercaseHeaders() {
|
|
||||||
Multimap<String, String> lowercaseHeaders = HashMultimap.create(getHeaders().keys().size(), 1);
|
|
||||||
|
|
||||||
for (String header: getHeaders().keys()) {
|
|
||||||
for (String value: getHeaders().get(header)) {
|
|
||||||
lowercaseHeaders.put(header.toLowerCase(), value);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return lowercaseHeaders;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* try to get the value, then try as lowercase.
|
* try to get the value, then try as lowercase.
|
||||||
*/
|
*/
|
||||||
public String getFirstHeaderOrNull(String string) {
|
public String getFirstHeaderOrNull(String string) {
|
||||||
Collection<String> values = headers.get(string);
|
Collection<String> values = headers.get(string);
|
||||||
if (values.size() == 0)
|
if (values.size() == 0) {
|
||||||
values = getLowercaseHeaders().get(string.toLowerCase());
|
Multimap<String, String> lowerCaseHeaders = Multimaps2.transformKeys(getHeaders(), new ToLowerCase());
|
||||||
|
values = lowerCaseHeaders.get(string.toLowerCase());
|
||||||
|
}
|
||||||
return (values.size() >= 1) ? values.iterator().next() : null;
|
return (values.size() >= 1) ? values.iterator().next() : null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -22,7 +22,9 @@ import static com.google.common.base.Preconditions.checkNotNull;
|
||||||
|
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Set;
|
import java.util.Set;
|
||||||
|
import java.util.Map.Entry;
|
||||||
|
|
||||||
|
import com.google.common.base.Function;
|
||||||
import com.google.common.base.Predicates;
|
import com.google.common.base.Predicates;
|
||||||
import com.google.common.collect.ImmutableMap;
|
import com.google.common.collect.ImmutableMap;
|
||||||
import com.google.common.collect.ImmutableMultimap;
|
import com.google.common.collect.ImmutableMultimap;
|
||||||
|
@ -89,4 +91,28 @@ public class Multimaps2 {
|
||||||
public static <K, V> Multimap<K, V> withoutKeys(Multimap<K, V> fromMultimap, Set<K> keys) {
|
public static <K, V> Multimap<K, V> withoutKeys(Multimap<K, V> fromMultimap, Set<K> keys) {
|
||||||
return Multimaps.<K, V> filterKeys(fromMultimap, Predicates.not(Predicates.in(keys)));
|
return Multimaps.<K, V> filterKeys(fromMultimap, Predicates.not(Predicates.in(keys)));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* change the keys but keep the values in-tact.
|
||||||
|
*
|
||||||
|
* @param <K1>
|
||||||
|
* input key type
|
||||||
|
* @param <K2>
|
||||||
|
* output key type
|
||||||
|
* @param <V>
|
||||||
|
* value type
|
||||||
|
* @param in
|
||||||
|
* input map to transform
|
||||||
|
* @param fn
|
||||||
|
* how to transform the values
|
||||||
|
* @return immutableMap with the new keys.
|
||||||
|
*/
|
||||||
|
public static <K1, K2, V> Multimap<K2, V> transformKeys(Multimap<K1, V> in, Function<K1, K2> fn) {
|
||||||
|
checkNotNull(in, "input map");
|
||||||
|
checkNotNull(fn, "function");
|
||||||
|
Builder<K2, V> returnVal = ImmutableMultimap.builder();
|
||||||
|
for (Entry<K1, V> entry : in.entries())
|
||||||
|
returnVal.put(fn.apply(entry.getKey()), entry.getValue());
|
||||||
|
return returnVal.build();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,41 @@
|
||||||
|
/**
|
||||||
|
* Licensed to jclouds, Inc. (jclouds) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. jclouds licenses this file
|
||||||
|
* to you under the Apache License, Version 2.0 (the
|
||||||
|
* "License"); you may not use this file except in compliance
|
||||||
|
* with the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing,
|
||||||
|
* software distributed under the License is distributed on an
|
||||||
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||||
|
* KIND, either express or implied. See the License for the
|
||||||
|
* specific language governing permissions and limitations
|
||||||
|
* under the License.
|
||||||
|
*/
|
||||||
|
package org.jclouds.util;
|
||||||
|
|
||||||
|
import static org.testng.Assert.assertEquals;
|
||||||
|
|
||||||
|
import org.jclouds.functions.ToLowerCase;
|
||||||
|
import org.testng.annotations.Test;
|
||||||
|
|
||||||
|
import com.google.common.collect.ImmutableMultimap;
|
||||||
|
import com.google.common.collect.Multimap;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Everett Toews
|
||||||
|
*/
|
||||||
|
@Test(groups = "unit")
|
||||||
|
public class Multimaps2Test {
|
||||||
|
public void testTransformKeysToLowerCase() {
|
||||||
|
Multimap<String, String> map = ImmutableMultimap.of("oNe", "1", "TWO", "2", "three", "3", "Three", "3.0");
|
||||||
|
Multimap<String, String> expected = ImmutableMultimap.of("one", "1", "two", "2", "three", "3", "three", "3.0");
|
||||||
|
Multimap<String, String> transformed = Multimaps2.transformKeys(map, new ToLowerCase());
|
||||||
|
|
||||||
|
assertEquals(transformed, expected);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue