org.jboss.resteasy.client.jaxrs.cache.MapCache Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of resteasy-client Show documentation
There is a newer version: 7.0.0.Alpha4
package org.jboss.resteasy.client.jaxrs.cache;

import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class MapCache implements BrowserCache
{
   protected Map> cache = null;

   public MapCache()
   {
      this(new ConcurrentHashMap>());
   }

   public MapCache(final Map> cache)
   {
      this.cache = cache;
   }

   protected Map> createCache()
   {
      return new ConcurrentHashMap>();
   }

   public Entry get(String key, MediaType accept)
   {
      Map parent = cache.get(key);
      if (parent == null || parent.isEmpty()) {
         return null;
      }
      if (accept.isWildcardType()) {
         // if the client accepts */*, return just the first entry for requested URL
         return parent.entrySet().iterator().next().getValue();
      } else if (accept.isWildcardSubtype()) {
         // if the client accepts /*, return the first entry which media type starts with /
         for (Map.Entry parentEntry : parent.entrySet()) {
            if (parentEntry.getKey().startsWith(accept.getType() + "/")) {
               return parentEntry.getValue();
            }
         }
      }
      return parent.get(accept.toString());
   }

   public Entry getAny(String key)
   {
      Map parent = cache.get(key);
      if (parent == null) return null;
      Iterator iterator = parent.values().iterator();
      if (iterator.hasNext()) return iterator.next();
      return null;
   }

   public Entry getEntry(String key, MediaType accept)
   {
      Map parent = cache.get(key);
      if (parent == null) return null;
      return parent.get(accept.toString());
   }

   public Entry remove(String key, MediaType type)
   {
      Map data = cache.get(key);
      if (data == null) return null;
      Entry removed = data.remove(type.toString());
      if (data.isEmpty())
      {
         cache.remove(key);
      }
      else
      {
         cache.put(key, data);
      }
      return removed;
   }

   public void clear()
   {
      cache.clear();
   }

   public Entry put(CacheEntry cacheEntry)
   {
      Map map = cache.get(cacheEntry.getKey());
      if (map == null)
      {
         map = new ConcurrentHashMap();
      }
      map.put(cacheEntry.getMediaType().toString(), cacheEntry);
      cache.put(cacheEntry.getKey(), map);
      return cacheEntry;
   }

   public Entry put(String key, MediaType mediaType,
               MultivaluedMap headers, byte[] cached, int expires,
               String etag, String lastModified)
   {
      return put(new CacheEntry(key, headers, cached, expires, etag, lastModified, mediaType));
   }
}