001    /*****************************************************************************
002     * Copyright (C) PicoContainer Organization. All rights reserved.            *
003     * ------------------------------------------------------------------------- *
004     * The software in this package is published under the terms of the BSD      *
005     * style license a copy of which has been included with this distribution in *
006     * the LICENSE.txt file.                                                     *
007     *                                                                           *
008     *****************************************************************************/
009    package org.picocontainer.references;
010    
011    import org.picocontainer.ObjectReference;
012    
013    import java.util.Map;
014    
015    /** @author Paul Hammant */
016    public class ThreadLocalMapObjectReference<T> implements ObjectReference<T> {
017        private final ThreadLocal<Map> threadLocal;
018        private final Object componentKey;
019    
020        public ThreadLocalMapObjectReference(ThreadLocal<Map> threadLocal, Object componentKey) {
021            this.threadLocal = threadLocal;
022            this.componentKey = componentKey;
023        }
024    
025        @SuppressWarnings("unchecked")
026        public T get() {
027            return (T) ((Map)threadLocal.get()).get(componentKey) ;
028        }
029    
030        @SuppressWarnings("unchecked")
031        public void set(T item) {
032            ((Map)threadLocal.get()).put(componentKey, item) ;
033    
034        }
035    }