| 68 | 1 /* | 
|  | 2 Copyright (c) 2008  Franklin Schmidt <fschmidt@gmail.com> | 
|  | 3 | 
|  | 4 Permission is hereby granted, free of charge, to any person obtaining a copy | 
|  | 5 of this software and associated documentation files (the "Software"), to deal | 
|  | 6 in the Software without restriction, including without limitation the rights | 
|  | 7 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | 
|  | 8 copies of the Software, and to permit persons to whom the Software is | 
|  | 9 furnished to do so, subject to the following conditions: | 
|  | 10 | 
|  | 11 The above copyright notice and this permission notice shall be included in | 
|  | 12 all copies or substantial portions of the Software. | 
|  | 13 | 
|  | 14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | 
|  | 15 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | 
|  | 16 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | 
|  | 17 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | 
|  | 18 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | 
|  | 19 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | 
|  | 20 THE SOFTWARE. | 
|  | 21 */ | 
|  | 22 | 
|  | 23 package fschmidt.util.java; | 
|  | 24 | 
|  | 25 import java.util.Map; | 
|  | 26 import java.util.LinkedHashMap; | 
|  | 27 import java.util.Iterator; | 
|  | 28 import fschmidt.db.util.WeakCacheMap; | 
|  | 29 | 
|  | 30 | 
|  | 31 public class TimedCacheMap<K,V> extends WeakCacheMap<K,V> { | 
|  | 32 | 
|  | 33 	private static class IdentityRef { | 
|  | 34 		private final Object obj; | 
|  | 35 | 
|  | 36 		IdentityRef(Object obj) { | 
|  | 37 			this.obj = obj; | 
|  | 38 		} | 
|  | 39 | 
|  | 40 		public int hashCode() { | 
|  | 41 			return System.identityHashCode(obj); | 
|  | 42 		} | 
|  | 43 | 
|  | 44 		public boolean equals(Object obj) { | 
|  | 45 			return obj instanceof IdentityRef | 
|  | 46 				&& obj==((IdentityRef)obj).obj; | 
|  | 47 		} | 
|  | 48 	} | 
|  | 49 | 
|  | 50 	private final long timeLimit; | 
|  | 51 	private final Map<IdentityRef,Long> vals = new LinkedHashMap<IdentityRef,Long>(); | 
|  | 52 | 
|  | 53 	public TimedCacheMap(long timeLimit) { | 
|  | 54 		this.timeLimit = timeLimit; | 
|  | 55 	} | 
|  | 56 | 
|  | 57 	private void add(V value,long now) { | 
|  | 58 		if( value==null ) | 
|  | 59 			return; | 
|  | 60 		IdentityRef ref = new IdentityRef(value); | 
|  | 61 		vals.remove(ref); | 
|  | 62 		vals.put( ref, now + timeLimit ); | 
|  | 63 	} | 
|  | 64 | 
|  | 65 	private void sweep(long now) { | 
|  | 66 		for( Iterator<Long> i = vals.values().iterator(); i.hasNext(); ) { | 
|  | 67 			long time = i.next(); | 
|  | 68 			if( time > now ) | 
|  | 69 				return; | 
|  | 70 			i.remove(); | 
|  | 71 		} | 
|  | 72 	} | 
|  | 73 | 
|  | 74 	public V put(K key, V value) { | 
|  | 75 		long now = System.currentTimeMillis(); | 
|  | 76 		sweep(now); | 
|  | 77 		add(value,now); | 
|  | 78 		return super.put(key,value); | 
|  | 79 	} | 
|  | 80 | 
|  | 81 	public V get(Object key) { | 
|  | 82 		long now = System.currentTimeMillis(); | 
|  | 83 		V value = super.get(key); | 
|  | 84 		add(value,now); | 
|  | 85 		return value; | 
|  | 86 	} | 
|  | 87 } |