1 /** 2 * This file Copyright (c) 2017 Magnolia International 3 * Ltd. (http://www.magnolia-cms.com). All rights reserved. 4 * 5 * 6 * This file is dual-licensed under both the Magnolia 7 * Network Agreement and the GNU General Public License. 8 * You may elect to use one or the other of these licenses. 9 * 10 * This file is distributed in the hope that it will be 11 * useful, but AS-IS and WITHOUT ANY WARRANTY; without even the 12 * implied warranty of MERCHANTABILITY or FITNESS FOR A 13 * PARTICULAR PURPOSE, TITLE, or NONINFRINGEMENT. 14 * Redistribution, except as permitted by whichever of the GPL 15 * or MNA you select, is prohibited. 16 * 17 * 1. For the GPL license (GPL), you can redistribute and/or 18 * modify this file under the terms of the GNU General 19 * Public License, Version 3, as published by the Free Software 20 * Foundation. You should have received a copy of the GNU 21 * General Public License, Version 3 along with this program; 22 * if not, write to the Free Software Foundation, Inc., 51 23 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 24 * 25 * 2. For the Magnolia Network Agreement (MNA), this file 26 * and the accompanying materials are made available under the 27 * terms of the MNA which accompanies this distribution, and 28 * is available at http://www.magnolia-cms.com/mna.html 29 * 30 * Any modifications to this file must keep this entire header 31 * intact. 32 * 33 */ 34 package info.magnolia.ui.framework.ioc; 35 36 import java.util.HashMap; 37 import java.util.Map; 38 import java.util.function.BiConsumer; 39 40 import com.google.inject.Key; 41 42 /** 43 * Key-value storage capable of keeping objects of arbitrary type 44 * mapped to a {@link Key}. Such storages are used to emulate the 45 * the scopes of apps and sub-apps. 46 * 47 * @see SessionStore 48 * @see UiScopes 49 */ 50 public final class BeanStore { 51 52 private final Map<Key, Object> storage = new HashMap<>(); 53 54 BeanStore() { 55 } 56 57 public void put(Key<?> key, Object bean) { 58 this.storage.put(key, bean); 59 } 60 61 public void put(Class<?> key, Object bean) { 62 this.put(Key.get(key), bean); 63 } 64 65 public <T> T get(Class<? extends T> key) { 66 //noinspection unchecked 67 return (T) this.storage.get(Key.get(key)); 68 } 69 70 public <T> T get(Key<? extends T> key) { 71 //noinspection unchecked 72 return (T) this.storage.get(key); 73 } 74 75 public void clear() { 76 this.storage.clear(); 77 } 78 79 public void forEach(BiConsumer<? super Key, ? super Object> handler) { 80 storage.forEach(handler); 81 } 82 }