net.hasor.dataql.binder.AppContextFinder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hasor-dataql Show documentation
the hasor DataQL project.
There is a newer version: 4.2.5
/*
 * 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.hasor.dataql.binder;
import net.hasor.core.AppContext;
import net.hasor.core.BindInfo;
import net.hasor.dataql.Finder;
import net.hasor.dataql.FragmentProcess;
import net.hasor.utils.ExceptionUtils;
import net.hasor.utils.ResourcesUtils;
import net.hasor.utils.ref.LinkedCaseInsensitiveMap;

import java.io.InputStream;
import java.util.List;
import java.util.Map;
import java.util.function.Supplier;

/**
 * 通过 AppContext 进行资源加载。
 * @author 赵永春 ([email protected])
 * @version : 2017-03-23
 */
public class AppContextFinder implements Finder {
    private AppContext                                       appContext;
    private Map> fragmentProcessMap;

    public AppContextFinder(AppContext appContext) {
        this.appContext = appContext;
        this.fragmentProcessMap = new LinkedCaseInsensitiveMap<>();
        List> bindInfos = this.appContext.findBindingRegister(FragmentProcess.class);
        if (bindInfos != null) {
            bindInfos.forEach(fragmentInfo -> {
                Supplier fragmentProcess = appContext.getProvider(fragmentInfo);
                this.fragmentProcessMap.put(fragmentInfo.getBindName().toLowerCase(), fragmentProcess);
            });
        }
    }

    public InputStream findResource(String resourceName) {
        InputStream inputStream = null;
        try {
            ClassLoader classLoader = this.appContext.getEnvironment().getClassLoader();
            if (classLoader != null) {
                resourceName = ResourcesUtils.formatResource(resourceName);
                inputStream = classLoader.getResourceAsStream(resourceName);
            } else {
                inputStream = ResourcesUtils.getResourceAsStream(resourceName);
            }
        } catch (Exception e) {
            String finalResourceName = resourceName;
            throw ExceptionUtils.toRuntimeException(e, throwable -> new RuntimeException("import compiler failed -> '" + finalResourceName + "' not found.", throwable));
        }
        return inputStream;
    }
 
    public Object findBean(Class beanType) {
        return this.appContext.getInstance(beanType);
    }

    @Override
    public FragmentProcess findFragmentProcess(String fragmentType) {
        Supplier fragmentProcessSupplier = this.fragmentProcessMap.get(fragmentType.toLowerCase());
        if (fragmentProcessSupplier == null) {
            return this.appContext.findBindingBean(fragmentType, FragmentProcess.class);
        }
        return fragmentProcessSupplier.get();
    }

    public void addFragmentProcess(String name, Supplier provider) {
        this.fragmentProcessMap.put(name, provider);
    }
}