ConfigurationFileReader.java example

Explorer
citrus-master
/*
 * Copyright (c) 2002-2012 Alibaba Group Holding Limited.
 * All rights reserved.
 *
 * 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 com.alibaba.citrus.dev.handler.util;

import static com.alibaba.citrus.util.ArrayUtil.*;
import static com.alibaba.citrus.util.Assert.*;
import static com.alibaba.citrus.util.CollectionUtil.*;
import static com.alibaba.citrus.util.ExceptionUtil.*;
import static com.alibaba.citrus.util.StringUtil.*;

import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.List;
import java.util.Set;

import com.alibaba.citrus.dev.handler.util.DomUtil.ElementFilter;
import org.dom4j.Namespace;
import org.dom4j.QName;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;
import org.springframework.core.io.support.ResourcePatternResolver;

public class ConfigurationFileReader {
    private final ConfigurationFile[] configurationFiles;
    private final String              baseURL;
    private final ResourceLoader      loader;

    public ConfigurationFileReader(ResourceLoader loader, String[] configLocations) throws IOException {
        this.loader = assertNotNull(loader);

        // 取得所有顶层resources
        List<NamedResource> resources = createLinkedList();

        for (String configLocation : configLocations) {
            if (loader instanceof ResourcePatternResolver) {
                Resource[] results = ((ResourcePatternResolver) loader).getResources(configLocation);

                if (!isEmptyArray(results)) {
                    for (Resource res : results) {
                        resources.add(new NamedResource(configLocation, res));
                    }
                }
            } else {
                Resource res = loader.getResource(configLocation);

                if (res != null) {
                    resources.add(new NamedResource(configLocation, res));
                }
            }
        }

        // 计算baseurl
        URL base = null;

        try {
            base = URI.create(loader.getResource("WEB-INF/web.xml").getURL().toURI() + "/../..").normalize().toURL();
        } catch (URISyntaxException e) {
            unexpectedException(e);
        }

        this.baseURL = trimEnd(base.toExternalForm(), "/");

        // 读取configuration files
        List<ConfigurationFile> configurationFiles = createLinkedList();
        Set<String> parsedNames = createHashSet();

        for (NamedResource namedResource : resources) {
            ConfigurationFile configurationFile = parseConfigurationFile(namedResource, parsedNames);

            if (configurationFile != null) {
                configurationFiles.add(configurationFile);
            }
        }

        this.configurationFiles = configurationFiles.toArray(new ConfigurationFile[configurationFiles.size()]);
    }

    private ConfigurationFile parseConfigurationFile(final NamedResource namedResource, final Set<String> parsedNames) {
        URL url;

        try {
            url = namedResource.resource.getURL();
        } catch (IOException e) {
            unexpectedException(e);
            return null;
        }

        String name = url.toExternalForm();

        if (name.startsWith(baseURL)) {
            name = name.substring(baseURL.length());
        }

        if (parsedNames.contains(name)) {
            return null;
        }

        parsedNames.add(name);

        final List<ConfigurationFile> importedConfigurationFiles = createLinkedList();
        Element rootElement;

        try {
            rootElement = DomUtil.readDocument(name, url, new ElementFilter() {
                public org.dom4j.Element filter(org.dom4j.Element e) throws Exception {
                    // 删除schemaLocation
                    org.dom4j.Attribute attr = e.attribute(new QName("schemaLocation", new Namespace("xsi",
                                                                                                     "http://www.w3.org/2001/XMLSchema-instance")));

                    if (attr != null) {
                        e.remove(attr);
                    }

                    // 导入beans:import，并删除element
                    if ("http://www.springframework.org/schema/beans".equals(e.getNamespaceURI())
                        && "import".equals(e.getName())) {
                        String importedResourceName = trimToNull(e.attributeValue("resource"));

                        if (importedResourceName != null) {
                            Resource importedResource;

                            if (importedResourceName.contains(":")) {
                                importedResource = loader.getResource(importedResourceName);
                            } else {
                                importedResource = namedResource.resource.createRelative(importedResourceName);
                            }

                            ConfigurationFile importedConfigurationFile = parseConfigurationFile(new NamedResource(
                                    importedResourceName, importedResource), parsedNames);

                            if (importedConfigurationFile != null) {
                                importedConfigurationFiles.add(importedConfigurationFile);
                            }
                        }

                        return null;
                    }

                    return e;
                }
            });
        } catch (Exception e) {
            rootElement = new Element("read-error").setText(getStackTrace(getRootCause(e)));
        }

        return new ConfigurationFile(namedResource.name, url,
                                     importedConfigurationFiles.toArray(new ConfigurationFile[importedConfigurationFiles.size()]),
                                     rootElement);
    }

    public ConfigurationFile[] toConfigurationFiles() {
        return configurationFiles;
    }

    private static class NamedResource {
        private final String   name;
        private final Resource resource;

        public NamedResource(String name, Resource resource) {
            this.name = name;
            this.resource = resource;
        }
    }
}