使用Java批量修改PowerDesigner的pdm文件

相信很多同学都使用PowerDesigner画数据库ER图，但是大家是否知道PowerDesigner的pdm文件是并不是一个二进制文件，而是一个xml格式文件，可以直接使用文件编辑器或XmlSpy这样的工具直接打开编辑。

前段时间做项目正好做到数据库设计和评审，客户要求数据库中每个字段都添加中文注释，最简单的方法就是把PDM中数据列的中文名为作注释，在生成数据库脚本时自动生成创建注释的语句。

既然PDM文件本质上是xml格式，那批量修改PDM文件列注释的工作就可以交给Java程序来完成：

package com.xxx.util.pdm;

import java.io.File;
import java.io.FileWriter;
import java.util.List;

import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;

/**
* To change this template use File | Settings | File Templates.
*/
public class ModifyPdm {

    String pdmFile = "E:/原PDM文件名.pdm";
    String outputFile = "E:/新PDM文件名.pdm";

    public static void main(String[] args) {
        new ModifyPdm().AddComment();
    }
 
    /**
     * 为数据表和列添加Comment
     * 使用该方法时需要注意，保存文件后，Report中定义的格式会受影响
     */
    public void AddComment() {
        try{
            SAXReader reader = new SAXReader();
            Document document = reader.read(new File(pdmFile));
            Element root = document.getRootElement();
          
            
            //找到每一个Column定义，添加或修改注释
            List<Node> nodes = document.selectNodes("/Model/o:RootObject/c:Children/o:Model/c:Packages/o:Package/c:Tables/o:Table/c:Columns/o:Column");
            
            Node nameNode, commentNode;
            Element element;
            String columnName, columnComment;
            for (Node node : nodes) {
                nameNode = node.selectSingleNode("a:Name");
                commentNode = node.selectSingleNode("a:Comment");
                
                columnName = nameNode.getText();
                //设置Column的注释。如果不存在注释，则取字段中文名+引号作为注释
                if (commentNode == null) {
                        element = nameNode.getParent();
                        element.addElement("a:Comment").setText(columnName + ":");
                } else {
                    columnComment = commentNode.getText();
                    if (!columnComment.startsWith(columnName + ":")) {
                            commentNode.setText(columnName + ":" + columnComment);
                    }
                }
            }
            
            //找到每一个Table定义，添加或修改注释
            nodes = document.selectNodes("/Model/o:RootObject/c:Children/o:Model/c:Packages/o:Package/c:Tables/o:Table");
            
            String tableName, tableComment;
            for (Node node : nodes) {
                nameNode = node.selectSingleNode("a:Name");
                commentNode = node.selectSingleNode("a:Comment");
                
                tableName = nameNode.getText();
                //设置Table的注释。如果不存在注释，则取字段中文名+引号作为注释
                if (commentNode == null) {
                    element = nameNode.getParent();
                    element.addElement("a:Comment").setText(tableName + ":");
                } else {
                    tableComment = commentNode.getText();
                    if (!columnComment.startsWith(tableName + ":")) {
                            commentNode.setText(tableName + ":" + tableComment);
                    }
                }
            }           
            
            
            //保存为PDM文件
            OutputFormat format = OutputFormat.createPrettyPrint();   
            format.setEncoding("UTF-8");    // 指定XML编码           
            XMLWriter writer = new XMLWriter(new FileWriter(outputFile),format);                
            writer.write(document);   
            writer.close();   
            
            System.out.println("完成! 请用PowerDesigner打开文件：" + outputFile);
            System.out.println("Report中定义的格式会受影响，请重新定义输出的rft格式。");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}