net.ymate.platform.configuration.impl.PropertyConfigFileParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ymate-platform-configuration Show documentation
YMP Configuration.
There is a newer version: 2.1.2
Show newest version
/*
 * Copyright 2007-2017 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.ymate.platform.configuration.impl;

import net.ymate.platform.configuration.AbstractConfigFileParser;
import org.apache.commons.lang.StringUtils;
import org.xml.sax.SAXException;

import javax.xml.parsers.ParserConfigurationException;
import java.io.*;
import java.net.URL;
import java.util.Enumeration;
import java.util.Properties;

/**
 * 基于Properties解析工具处理properties配置文件的读写操作
 *
 * @author 刘镇 ([email protected]) on 2010-9-5 下午06:37:36
 * @version 1.0
 */
public class PropertyConfigFileParser extends AbstractConfigFileParser {

    private final Properties __rootProps = new Properties();

    public PropertyConfigFileParser(File file) throws IOException {
        FileReader _reader = new FileReader(file);
        try {
            __rootProps.load(_reader);
        } finally {
            _reader.close();
        }
    }

    public PropertyConfigFileParser(InputStream inputStream) throws IOException {
        __rootProps.load(inputStream);
    }

    public PropertyConfigFileParser(URL url) throws ParserConfigurationException, IOException, SAXException {
        InputStream _in = url.openStream();
        try {
            __rootProps.load(_in);
        } finally {
            _in.close();
        }
    }

    @Override
    @SuppressWarnings("unchecked")
    protected void __doLoad() {
        Enumeration _propNames = (Enumeration) __rootProps.propertyNames();
        while (_propNames.hasMoreElements()) {
            String _propName = _propNames.nextElement();
            if (StringUtils.startsWith(_propName, TAG_NAME_ROOT)) {
                String _newPropName = StringUtils.substringAfter(_propName, TAG_NAME_ROOT.concat("."));
                // _propArr[0] = categoryName
                // _propArr[1] = propertyName
                // _propArr[2] = attributes关键字
                // _propArr[3] = attrName
                String[] _propArr = StringUtils.split(_newPropName, ".");
                if (_propArr.length > 1) {
                    // 若为根属性
                    if (_propArr[0].equalsIgnoreCase(TAG_NAME_ATTRIBUTES)) {
                        __rootAttributes.put(_propArr[1], new Attribute(_propArr[1], __rootProps.getProperty(_propName)));
                        continue;
                    }
                    // 若为正常的category, 若category对象不存在, 则创建它
                    Category _category = __categories.get(_propArr[0]);
                    if (_category == null) {
                        _category = new Category(_propArr[0], null, null, __sorted);
                        __categories.put(_propArr[0], _category);
                    }
                    //
                    switch (_propArr.length) {
                        case 4:
                            if (_propArr[2].equalsIgnoreCase(TAG_NAME_ATTRIBUTES)) {
                                Property _prop = __safeGetProperty(_category, _propName, _propArr[1]);
                                if (_prop != null) {
                                    __fixedSetAttribute(_prop, _propName, _propArr[3]);
                                }
                            } else {
                                _category.getPropertyMap().put(_propArr[3], new Property(_propArr[3], __rootProps.getProperty(_propName), null));
                            }
                            break;
                        case 2:
                            __fixedSetProperty(_category, _propName, _propArr[1]);
                            break;
                        default:
                            if (_propArr[1].equalsIgnoreCase(TAG_NAME_ATTRIBUTES)) {
                                _category.getAttributeMap().put(_propArr[2], new Attribute(_propArr[2], __rootProps.getProperty(_propName)));
                            } else {
                                Property _prop = __safeGetProperty(_category, _propName, _propArr[1]);
                                if (_prop != null) {
                                    __fixedSetAttribute(_prop, _propName, _propArr[2]);
                                }
                            }
                            break;
                    }
                }
            }
        }
        // 必须保证DEFAULT_CATEGORY_NAME配置集合存在
        if (!__categories.containsKey(DEFAULT_CATEGORY_NAME)) {
            __categories.put(DEFAULT_CATEGORY_NAME, new Category(DEFAULT_CATEGORY_NAME, null, null, __sorted));
        }
    }

    private Property __safeGetProperty(Category category, String propName, String newPropName) {
        Property _property = category.getProperty(newPropName);
        if (_property == null) {
            _property = new Property(newPropName, null, null);
            category.getPropertyMap().put(newPropName, _property);
        }
        return _property;
    }

    private void __fixedSetAttribute(Property property, String propName, String newPropName) {
        Attribute _attr = property.getAttribute(newPropName);
        String _attrValue = __rootProps.getProperty(propName);
        if (_attr == null) {
            _attr = new Attribute(newPropName, _attrValue);
            property.getAttributeMap().put(newPropName, _attr);
        } else {
            _attr.setKey(newPropName);
            _attr.setValue(_attrValue);
        }
    }

    private void __fixedSetProperty(Category category, String propName, String newPropName) {
        Property _prop = category.getProperty(newPropName);
        String _propContent = __rootProps.getProperty(propName);
        if (_prop == null) {
            _prop = new Property(newPropName, _propContent, null);
            category.getPropertyMap().put(newPropName, _prop);
        } else {
            _prop.setName(newPropName);
            _prop.setContent(_propContent);
        }
    }

    @Override
    public void writeTo(File targetFile) throws IOException {
        // TODO write file
    }

    @Override
    public void writeTo(OutputStream outputStream) throws IOException {
        // TODO write file
    }
}