org.noear.wood.generator.entity.XmlParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wood-maven-plugin Show documentation
The newest version!
package org.noear.wood.generator.entity;

import org.noear.wood.DbContext;
import org.noear.wood.generator.entity.block.TableItem;
import org.noear.wood.generator.entity.block.XmlEntityBlock;
import org.noear.wood.generator.entity.block.XmlSourceBlock;
import org.noear.wood.generator.entity.block.XmlTableBlock;
import org.noear.wood.generator.utils.NamingUtils;
import org.noear.wood.generator.utils.StringUtils;
import org.noear.wood.generator.utils.XmlUtils;
import org.noear.wood.wrap.TableWrap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

import java.util.ArrayList;
import java.util.List;

public class XmlParser {

    public static XmlSourceBlock getSource(Node n1) throws Exception {
        if (Names.tag_source.equals(n1.getNodeName()) == false) {
            return null;
        }

        XmlSourceBlock source = new XmlSourceBlock();

        source.schema = XmlUtils.attr(n1, Names.att_schema);
        source.url = XmlUtils.attr(n1, Names.att_url);
        source.username = XmlUtils.attr(n1, Names.att_username);
        source.password = XmlUtils.attr(n1, Names.att_password);
        source.driverClassName = XmlUtils.attr(n1, Names.att_driverClassName);
        source.namingStyle = XmlUtils.attr(n1, Names.att_namingStyle);
        source.typeStyle = XmlUtils.attr(n1, Names.att_typeStyle);

        if (source.typeStyle == null) {
            source.typeStyle = "";
        }

        NodeList n2l = n1.getChildNodes();
        for (int i = 0, len = n2l.getLength(); i < len; i++) {
            Node n2 = n2l.item(i);

            if (n2.getNodeType() != 1) {
                continue;
            }

            if (Names.tag_entity.equals(n2.getNodeName())) {
                XmlEntityBlock bk = new XmlEntityBlock();
                bk.targetPackage = XmlUtils.attr(n2, Names.att_targetPackage);
                bk.entityName = XmlUtils.attr(n2, Names.att_entityName);
                bk.code = getCode(n2);

                source.entityBlocks.add(bk);
            }

            if (Names.tag_table.equals(n2.getNodeName())) {
                XmlTableBlock bk = new XmlTableBlock();
                bk.tableName = XmlUtils.attr(n2, Names.att_tableName);
                bk.domainName = XmlUtils.attr(n2, Names.att_domainName);

                source.tableBlocks.add(bk);
            }
        }

        return source;
    }

    public static String getCode(Node n2){
        StringBuilder sb = new StringBuilder();

        NodeList nl = n2.getChildNodes();

        for(int i=0,len=nl.getLength(); i getTables(XmlSourceBlock source, DbContext db){
        List tableItems = new ArrayList<>();

        for(XmlTableBlock tb : source.tableBlocks){
            if("*".equals(tb.tableName)){
                for(TableWrap tw : db.getMetaData().getTableAll()){
                    TableItem item = new TableItem();
                    item.tableName = tw.getName();
                    item.domainName = NamingUtils.toCamelString(item.tableName,true);
                    item.tableWrap = tw;

                    tableItems.add(item);
                }

                break;
            }else{
                for(TableWrap tw : db.getMetaData().getTableAll()){
                    if(tw.getName().equals(tb.tableName)) {
                        TableItem item = new TableItem();
                        item.tableName = tw.getName();
                        if(StringUtils.isEmpty(tb.domainName)) {
                            item.domainName = NamingUtils.toCamelString(item.tableName, true);
                        }else{
                            item.domainName = tb.domainName;
                        }

                        item.tableWrap = tw;

                        tableItems.add(item);
                    }
                }
            }
        }

        return tableItems;
    }
}