MyScope.java example

Explorer
hasor-master
/*
 * Copyright 2008-2009 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.test.hasor.core._04_scope;
import net.hasor.core.Provider;
import net.hasor.core.Scope;
import net.hasor.core.provider.SingleProvider;

import java.util.HashMap;
import java.util.Map;
/**
 * 一个自定义 Scope ，实现了线程间单例化
 * @version : 2015年11月9日
 * @author 赵永春(zyc@hasor.net)
 */
public class MyScope implements Scope {
    private ThreadLocal<Map<Object, Provider<?>>> threadSington = new ThreadLocal<Map<Object, Provider<?>>>();
    //
    public MyScope() {
        this.threadSington.set(new HashMap<Object, Provider<?>>());
    }
    public <T> Provider<T> scope(Object key, final Provider<T> provider) {
        Map<Object, Provider<?>> scopeMap = threadSington.get();
        Provider<?> returnData = scopeMap.get(key);
        if (returnData == null) {
            Provider<T> newSingleProvider = new SingleProvider<T>(provider) {
            };
            returnData = scopeMap.put(key, newSingleProvider);
            if (returnData == null) {
                returnData = newSingleProvider;
            }
        }
        return (Provider<T>) returnData;
    }
}