darks.orm.core.config.sqlmap.DDLConfigReader.java Source code

Introduction

Here is the source code for darks.orm.core.config.sqlmap.DDLConfigReader.java

Source

/**
 * 
 * Copyright 2014 The Darks ORM Project (Liu lihua)
 *
 * 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 darks.orm.core.config.sqlmap;

import java.util.Iterator;

import darks.orm.core.data.xml.DDLData;
import darks.orm.core.data.xml.DDLData.DDLType;
import darks.orm.util.UUIDHelper;
import org.dom4j.Attribute;
import org.dom4j.Element;

@SuppressWarnings("unchecked")
public class DDLConfigReader {

    private SqlMapConfiguration sqlMapConfig;

    public DDLConfigReader(SqlMapConfiguration sqlMapConfig) {
        this.sqlMapConfig = sqlMapConfig;
    }

    /**
     * DDL
     * 
     * @param element 
     */
    public void reader(Element element) {
        // DDL
        for (Iterator<Attribute> it = element.attributeIterator(); it.hasNext();) {
            try {
                Attribute at = it.next();
                String name = at.getName().trim();
                String value = at.getValue().trim();
                if ("schema".equalsIgnoreCase(name)) {
                    sqlMapConfig.setSchema(value);
                } else if ("catalog".equalsIgnoreCase(name)) {
                    sqlMapConfig.setCatalog(value);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        // DDL
        for (Iterator<Element> it = element.elementIterator(); it.hasNext();) {
            try {
                Element el = it.next();
                String name = el.getName().trim();
                if ("create".equalsIgnoreCase(name)) {
                    readCreate(el);
                } else if ("alter".equalsIgnoreCase(name)) {
                    readAlter(el);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * <create>
     * 
     * @param element 
     */
    public void readCreate(Element element) {
        readLabel(element, DDLType.Create);
    }

    /**
     * <alter>
     * 
     * @param element 
     */
    public void readAlter(Element element) {
        readLabel(element, DDLType.Alter);
    }

    /**
     * 
     * 
     * @param element 
     * @param type 
     */
    public void readLabel(Element element, DDLType type) {
        DDLData ddlData = new DDLData();
        String sql = element.getTextTrim();
        String strRndId = UUIDHelper.getUUID();
        ddlData.setId(strRndId);
        ddlData.setSql(sql);
        ddlData.setType(type);
        for (Iterator<Attribute> it = element.attributeIterator(); it.hasNext();) {
            try {
                Attribute at = it.next();
                String name = at.getName().trim();
                String value = at.getValue().trim();
                if ("id".equalsIgnoreCase(name)) {
                    ddlData.setId(value);
                } else if ("tableName".equalsIgnoreCase(name)) {
                    ddlData.setTableName(value);
                } else if ("checkTable".equalsIgnoreCase(name)) {
                    if ("true".equalsIgnoreCase(value))
                        ddlData.setCheckTable(true);
                    else if ("false".equalsIgnoreCase(value))
                        ddlData.setCheckTable(false);
                } else if ("autoRunable".equalsIgnoreCase(name)) {
                    if ("true".equalsIgnoreCase(value))
                        ddlData.setAutoRunable(true);
                    else if ("false".equalsIgnoreCase(value))
                        ddlData.setAutoRunable(false);
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        sqlMapConfig.addDDLData(ddlData.getId(), ddlData);
    }

    public SqlMapConfiguration getSqlMapConfig() {
        return sqlMapConfig;
    }

    public void setSqlMapConfig(SqlMapConfiguration sqlMapConfig) {
        this.sqlMapConfig = sqlMapConfig;
    }

}