com.centit.support.database.metadata.PdmReader Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of centit-database Show documentation
Show all versions of centit-database Show documentation
数据库操作通用方法和函数，从以前的util包中分离出来，并且整合了部分sys-module中的函数
package com.centit.support.database.metadata;
import com.centit.support.xml.IgnoreDTDEntityResolver;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.apache.commons.lang3.tuple.Pair;
import org.dom4j.*;
import org.dom4j.io.SAXReader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.sql.Connection;
import java.util.ArrayList;
import java.util.List;
public class PdmReader implements DatabaseMetadata {
protected static final Logger logger = LoggerFactory.getLogger(PdmReader.class);
private Document doc=null;
private String sDBSchema= null;
private List pkColumnIDs;
public boolean loadPdmFile(String sPath)
{
boolean b=false;
try (FileInputStream is = new FileInputStream(new File(sPath))){
//InputStream is = getClass().getResourceAsStream(sPath + sHbmFile);
SAXReader builder=new SAXReader(false);
builder.setValidation(false);
builder.setEntityResolver(new IgnoreDTDEntityResolver());
//Attribute attr;
doc= builder.read(is);
b = true;
} catch (DocumentException e) {
logger.error(e.getMessage(),e);//e.printStackTrace();
} catch (IOException e) {
logger.error(e.getMessage(),e);//e.printStackTrace();
}
return b;
}
private QName getPdmQName(String sPreFix,String sName)
{
String uri="attribute";
// xmlns:a="attribute" xmlns:c="collection" xmlns:o="object">
if("c".equals(sPreFix))
uri="collection";
else if("o".equals(sPreFix))
uri="object";
return new QName(sName,Namespace.get(sPreFix, uri), sPreFix+':'+sName);
}
/* @SuppressWarnings("unused")
private QName getPdmQName(String sQName)
{
int nPos = sQName.indexOf(':');
String sPreFix = sQName.substring(0,nPos);
String sName = sQName.substring(nPos+1);
return getPdmQName( sPreFix, sName);
}*/
private String getElementText(Element e,String sPreFix,String sName)
{
if(e==null)
return null;
Element f = e.element(getPdmQName(sPreFix,sName) );
if(f==null)
return null;
//System.out.println(f.asXML());
return f.getText();
}
private String getAttributeValue(Element e,String xPath)
{
if(e==null)
return null;
Attribute at = (Attribute)e.selectSingleNode(xPath);
if(at==null)
return null;
return at.getValue();
}
/**
*
* @return List Pair tableCode,tableName
*/
public List> getAllTableCode()
{
List> tabNames = new ArrayList<>();
List tabNodes = (List)doc.selectNodes("//c:Tables/o:Table");
for(Node tNode : tabNodes){
tabNames.add(new ImmutablePair(
getElementText((Element)tNode,"a","Code"),
getElementText((Element)tNode,"a","Name")));
}
return tabNames;
}
public SimpleTableInfo getTableMetadata(String tabName) {
pkColumnIDs = new ArrayList();
if (doc==null)
return null;
SimpleTableInfo tab = new SimpleTableInfo(tabName.toUpperCase());
if(sDBSchema!=null)
tab.setSchema(sDBSchema);
Node nTab = doc.selectSingleNode("//c:Tables/o:Table[a:Code='"+tabName+"']");
if(nTab==null)
return null;
//System.out.println(nTab.asXML());
Element eTab = (Element)nTab;
tab.setTableLabelName(getElementText(eTab,"a","Name"));
tab.setTableComment(getElementText(eTab,"a","Comment"));
//System.out.println(getElementText(eTab.element("a:Name")));
Element elColumns = eTab.element(getPdmQName("c","Columns"));///o:Column
if(elColumns==null)
return tab;
//获取 表的字段列表
List columns = (List) elColumns.elements(getPdmQName("o","Column"));///o:Column
for(Element col : columns){
SimpleTableField field = new SimpleTableField();
field.setColumnName(getElementText(col,"a","Code"));
//System.out.println(col.attributeValue("a:Code"));
field.setColumnType(getElementText(col,"a","DataType"));
String stemp = getElementText(col,"a","Length");
if(stemp !=null){
field.setMaxLength(Integer.valueOf(stemp));
field.setPrecision(Integer.valueOf(stemp));
}
//PDM 中的这个定义和数据库中的好像不一致
stemp = getElementText(col,"a","Precision");
if(stemp !=null)
field.setScale(Integer.valueOf(stemp));
stemp = getElementText(col,"a","Mandatory");
if(stemp !=null)
field.setMandatory(stemp);
field.setFieldLabelName(getElementText(col,"a","Name"));
field.setColumnComment(getElementText(col,"a","Comment"));
field.mapToMetadata();
tab.getColumns().add(field);
}
//获取 表主键
Attribute pkID = (Attribute)eTab.selectSingleNode("c:PrimaryKey/o:Key/@Ref");
if(pkID==null)
return tab;
String sPkID = pkID.getValue();
Element elPK = (Element) eTab.selectSingleNode("c:Keys/o:Key[@Id='"+sPkID+"']");
if(elPK==null)
return tab;
tab.setPkName(getElementText(elPK,"a","Code"));
//tab.setPkName(pKCode);
List pkColAttr = elPK.selectNodes("c:Key.Columns/o:Column/@Ref");
for( Node pkCA: pkColAttr){
pkColumnIDs.add(((Attribute) pkCA).getValue());
Element elPKCol = (Element) eTab.selectSingleNode(
"c:Columns/o:Column[@Id='"+((Attribute) pkCA).getValue()+"']/a:Code");
if(elPKCol!=null){
//System.out.println(elPKCol.asXML());
tab.getPkColumns().add(elPKCol.getText());
}
}
//获取所有的外键
List elReferences = (List) doc.selectNodes("//c:References/o:Reference[c:ParentKey/o:Key/@Ref='"+sPkID+"']");
for(Node elRef:elReferences){
SimpleTableReference ref = new SimpleTableReference();
ref.setParentTableName(tabName);
ref.setReferenceCode(((Element)elRef).attributeValue("Id")); //getElementText(elRef,"a","Code"));
ref.setReferenceName(getElementText(((Element)elRef),"a","Name"));
String sChildTabID = getAttributeValue(((Element)elRef),"c:ChildTable/o:Table/@Ref"); //="o501" />
if (sChildTabID==null)
sChildTabID = getAttributeValue(((Element)elRef),"c:Object2/o:Table/@Ref"); //="o501" />
Element eChildTab = null;
if(sChildTabID !=null){
eChildTab = (Element)doc.selectSingleNode("//c:Tables/o:Table[@Id='"+sChildTabID+"']");
}else{
String fpk = pkColumnIDs.get(0);
String ffk = getAttributeValue(((Element)elRef),
"c:Joins/o:ReferenceJoin[c:Object1/o:Column/@Ref='"+fpk+"']/c:Object2/o:Column/@Ref");
if(ffk!=null)
eChildTab = (Element)doc.selectSingleNode("//c:Tables/o:Table[c:Columns/o:Column/@Id='"+ffk+"']");
}
if(eChildTab==null)
continue;
ref.setTableName(getElementText(eChildTab,"a","Code"));
for(String pkColID : pkColumnIDs) {
SimpleTableField field = new SimpleTableField();
String sChildColId = getAttributeValue(((Element)elRef),
"c:Joins/o:ReferenceJoin[c:Object1/o:Column/@Ref='"+pkColID+"']/c:Object2/o:Column/@Ref");
//System.out.println(sChildColId);
Element col =(Element) eChildTab.selectSingleNode("c:Columns/o:Column[@Id='"+sChildColId+"']");
if(col==null)
continue;
//System.out.println(col.asXML());
field.setColumnName(getElementText(col,"a","Code"));
//System.out.println(col.attributeValue("a:Code"));
field.setColumnType(getElementText(col,"a","DataType"));
String stemp = getElementText(col,"a","Length");
if(stemp !=null){
field.setMaxLength(Integer.valueOf(stemp));
field.setPrecision(Integer.valueOf(stemp));
}
stemp = getElementText(col,"a","Precision");
if(stemp !=null)
field.setScale(Integer.valueOf(stemp));
stemp = getElementText(col,"a","Mandatory");
if(stemp !=null)
field.setMandatory(stemp);
field.setFieldLabelName(getElementText(col,"a","Name"));
field.setColumnComment(getElementText(col,"a","Comment"));
field.mapToMetadata();
ref.getFkColumns().add(field);
ref.getReferenceColumns().put(pkColID,
field.getColumnName());
}
tab.getReferences().add(ref );
}
return tab;
}
public HibernateMapInfo toHibernateMetadata(SimpleTableInfo tableMeta){
HibernateMapInfo hibernateMeta = new HibernateMapInfo();
hibernateMeta.setClassName(tableMeta.getPackageName()+'.'+tableMeta.getClassName());
hibernateMeta.setTableName(tableMeta.getTableName().toUpperCase());
hibernateMeta.setTableLabelName(tableMeta.getTableLabelName());
hibernateMeta.setTableComment(tableMeta.getTableComment());
hibernateMeta.setMainTable(true);
hibernateMeta.setHasID( tableMeta.getPkColumns().size()>1 );
if(hibernateMeta.isHasID()){
hibernateMeta.setIdType( tableMeta.getPackageName()+'.'+tableMeta.getClassName()+"Id" );
hibernateMeta.setIdName( "cid");
}else if(tableMeta.getPkColumns().size()==1){
SimpleTableField field = tableMeta.findField(tableMeta.getPkColumns().get(0));
hibernateMeta.setIdType(field.getHibernateType() );
hibernateMeta.setIdName( field.getPropertyName() );
}
for(SimpleTableField col : tableMeta.getColumns()){
if(tableMeta.isParmaryKey(col.getColumnName())){
hibernateMeta.getKeyProperties().add(col);
}else{
hibernateMeta.getProperties().add(col);
}
}
hibernateMeta.setReferences(tableMeta.getReferences());
return hibernateMeta;
}
public HibernateMapInfo getHibernateMetadata(String tabName,String sPackageName) {
SimpleTableInfo tabMeta = this.getTableMetadata(tabName);
if(tabMeta==null)
return null;
tabMeta.setPackageName(sPackageName);
HibernateMapInfo tab = toHibernateMetadata(tabMeta );
for(SimpleTableReference ref :tab.getReferences()){
SimpleTableInfo subTabMeta = this.getTableMetadata(ref.getTableName());
subTabMeta.setPackageName(sPackageName);
HibernateMapInfo subTab =
toHibernateMetadata(subTabMeta);
subTab.setMainTable(false);
tab.getOne2manys().add(subTab);
}
return tab;
}
public String getDBSchema() {
return sDBSchema;
}
public void setDBSchema(String schema) {
if(schema !=null)
sDBSchema = schema.toUpperCase();
}
/*
public static void main(String[] args) {
PdmReader reader = new PdmReader();
reader.loadPdmFile("E:\\temp\\BS开发框架.xml");
reader.getTableMetadata("TEST_REF");
}
*/
@Override
public void setDBConfig(Connection dbc) {
// not needed
}
}