Wed, 31 Dec 2014 07:22:50 +0100
Correct previous dual key logic pending first delivery installment.
michael@0 | 1 | /* |
michael@0 | 2 | * ==================================================================== |
michael@0 | 3 | * Licensed to the Apache Software Foundation (ASF) under one |
michael@0 | 4 | * or more contributor license agreements. See the NOTICE file |
michael@0 | 5 | * distributed with this work for additional information |
michael@0 | 6 | * regarding copyright ownership. The ASF licenses this file |
michael@0 | 7 | * to you under the Apache License, Version 2.0 (the |
michael@0 | 8 | * "License"); you may not use this file except in compliance |
michael@0 | 9 | * with the License. You may obtain a copy of the License at |
michael@0 | 10 | * |
michael@0 | 11 | * http://www.apache.org/licenses/LICENSE-2.0 |
michael@0 | 12 | * |
michael@0 | 13 | * Unless required by applicable law or agreed to in writing, |
michael@0 | 14 | * software distributed under the License is distributed on an |
michael@0 | 15 | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
michael@0 | 16 | * KIND, either express or implied. See the License for the |
michael@0 | 17 | * specific language governing permissions and limitations |
michael@0 | 18 | * under the License. |
michael@0 | 19 | * ==================================================================== |
michael@0 | 20 | * |
michael@0 | 21 | * This software consists of voluntary contributions made by many |
michael@0 | 22 | * individuals on behalf of the Apache Software Foundation. For more |
michael@0 | 23 | * information on the Apache Software Foundation, please see |
michael@0 | 24 | * <http://www.apache.org/>. |
michael@0 | 25 | * |
michael@0 | 26 | */ |
michael@0 | 27 | |
michael@0 | 28 | package ch.boye.httpclientandroidlib.protocol; |
michael@0 | 29 | |
michael@0 | 30 | import java.util.HashMap; |
michael@0 | 31 | import java.util.Iterator; |
michael@0 | 32 | import java.util.Map; |
michael@0 | 33 | |
michael@0 | 34 | /** |
michael@0 | 35 | * Maintains a map of objects keyed by a request URI pattern. |
michael@0 | 36 | * <br> |
michael@0 | 37 | * Patterns may have three formats: |
michael@0 | 38 | * <ul> |
michael@0 | 39 | * <li><code>*</code></li> |
michael@0 | 40 | * <li><code>*<uri></code></li> |
michael@0 | 41 | * <li><code><uri>*</code></li> |
michael@0 | 42 | * </ul> |
michael@0 | 43 | * <br> |
michael@0 | 44 | * This class can be used to resolve an object matching a particular request |
michael@0 | 45 | * URI. |
michael@0 | 46 | * |
michael@0 | 47 | * @since 4.0 |
michael@0 | 48 | */ |
michael@0 | 49 | public class UriPatternMatcher { |
michael@0 | 50 | |
michael@0 | 51 | /** |
michael@0 | 52 | * TODO: Replace with ConcurrentHashMap |
michael@0 | 53 | */ |
michael@0 | 54 | private final Map map; |
michael@0 | 55 | |
michael@0 | 56 | public UriPatternMatcher() { |
michael@0 | 57 | super(); |
michael@0 | 58 | this.map = new HashMap(); |
michael@0 | 59 | } |
michael@0 | 60 | |
michael@0 | 61 | /** |
michael@0 | 62 | * Registers the given object for URIs matching the given pattern. |
michael@0 | 63 | * |
michael@0 | 64 | * @param pattern the pattern to register the handler for. |
michael@0 | 65 | * @param obj the object. |
michael@0 | 66 | */ |
michael@0 | 67 | public synchronized void register(final String pattern, final Object obj) { |
michael@0 | 68 | if (pattern == null) { |
michael@0 | 69 | throw new IllegalArgumentException("URI request pattern may not be null"); |
michael@0 | 70 | } |
michael@0 | 71 | this.map.put(pattern, obj); |
michael@0 | 72 | } |
michael@0 | 73 | |
michael@0 | 74 | /** |
michael@0 | 75 | * Removes registered object, if exists, for the given pattern. |
michael@0 | 76 | * |
michael@0 | 77 | * @param pattern the pattern to unregister. |
michael@0 | 78 | */ |
michael@0 | 79 | public synchronized void unregister(final String pattern) { |
michael@0 | 80 | if (pattern == null) { |
michael@0 | 81 | return; |
michael@0 | 82 | } |
michael@0 | 83 | this.map.remove(pattern); |
michael@0 | 84 | } |
michael@0 | 85 | |
michael@0 | 86 | /** |
michael@0 | 87 | * @deprecated use {@link #setObjects(Map)} |
michael@0 | 88 | */ |
michael@0 | 89 | public synchronized void setHandlers(final Map map) { |
michael@0 | 90 | if (map == null) { |
michael@0 | 91 | throw new IllegalArgumentException("Map of handlers may not be null"); |
michael@0 | 92 | } |
michael@0 | 93 | this.map.clear(); |
michael@0 | 94 | this.map.putAll(map); |
michael@0 | 95 | } |
michael@0 | 96 | |
michael@0 | 97 | /** |
michael@0 | 98 | * Sets objects from the given map. |
michael@0 | 99 | * @param map the map containing objects keyed by their URI patterns. |
michael@0 | 100 | */ |
michael@0 | 101 | public synchronized void setObjects(final Map map) { |
michael@0 | 102 | if (map == null) { |
michael@0 | 103 | throw new IllegalArgumentException("Map of handlers may not be null"); |
michael@0 | 104 | } |
michael@0 | 105 | this.map.clear(); |
michael@0 | 106 | this.map.putAll(map); |
michael@0 | 107 | } |
michael@0 | 108 | |
michael@0 | 109 | /** |
michael@0 | 110 | * Looks up an object matching the given request URI. |
michael@0 | 111 | * |
michael@0 | 112 | * @param requestURI the request URI |
michael@0 | 113 | * @return object or <code>null</code> if no match is found. |
michael@0 | 114 | */ |
michael@0 | 115 | public synchronized Object lookup(String requestURI) { |
michael@0 | 116 | if (requestURI == null) { |
michael@0 | 117 | throw new IllegalArgumentException("Request URI may not be null"); |
michael@0 | 118 | } |
michael@0 | 119 | //Strip away the query part part if found |
michael@0 | 120 | int index = requestURI.indexOf("?"); |
michael@0 | 121 | if (index != -1) { |
michael@0 | 122 | requestURI = requestURI.substring(0, index); |
michael@0 | 123 | } |
michael@0 | 124 | |
michael@0 | 125 | // direct match? |
michael@0 | 126 | Object obj = this.map.get(requestURI); |
michael@0 | 127 | if (obj == null) { |
michael@0 | 128 | // pattern match? |
michael@0 | 129 | String bestMatch = null; |
michael@0 | 130 | for (Iterator it = this.map.keySet().iterator(); it.hasNext();) { |
michael@0 | 131 | String pattern = (String) it.next(); |
michael@0 | 132 | if (matchUriRequestPattern(pattern, requestURI)) { |
michael@0 | 133 | // we have a match. is it any better? |
michael@0 | 134 | if (bestMatch == null |
michael@0 | 135 | || (bestMatch.length() < pattern.length()) |
michael@0 | 136 | || (bestMatch.length() == pattern.length() && pattern.endsWith("*"))) { |
michael@0 | 137 | obj = this.map.get(pattern); |
michael@0 | 138 | bestMatch = pattern; |
michael@0 | 139 | } |
michael@0 | 140 | } |
michael@0 | 141 | } |
michael@0 | 142 | } |
michael@0 | 143 | return obj; |
michael@0 | 144 | } |
michael@0 | 145 | |
michael@0 | 146 | /** |
michael@0 | 147 | * Tests if the given request URI matches the given pattern. |
michael@0 | 148 | * |
michael@0 | 149 | * @param pattern the pattern |
michael@0 | 150 | * @param requestUri the request URI |
michael@0 | 151 | * @return <code>true</code> if the request URI matches the pattern, |
michael@0 | 152 | * <code>false</code> otherwise. |
michael@0 | 153 | */ |
michael@0 | 154 | protected boolean matchUriRequestPattern(final String pattern, final String requestUri) { |
michael@0 | 155 | if (pattern.equals("*")) { |
michael@0 | 156 | return true; |
michael@0 | 157 | } else { |
michael@0 | 158 | return |
michael@0 | 159 | (pattern.endsWith("*") && requestUri.startsWith(pattern.substring(0, pattern.length() - 1))) || |
michael@0 | 160 | (pattern.startsWith("*") && requestUri.endsWith(pattern.substring(1, pattern.length()))); |
michael@0 | 161 | } |
michael@0 | 162 | } |
michael@0 | 163 | |
michael@0 | 164 | } |