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:
Everett Toews 2012-08-20 19:02:33 -05:00
parent 8d9499b3c1
commit 27dee0bfd9
4 changed files with 111 additions and 15 deletions

View File

@ -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();
}
}

View File

@ -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;
} }

View File

@ -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();
}
} }

View File

@ -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);
}
}