开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > MySQL数据库 > 查看源码
ReverseEngineeringMssql.java
资源名称：mysql-gui-tools-noinstall-5.0-r17-win32.zip　[点击查看]
上传用户：cccombo
上传日期：2021-01-31
资源大小：16445k
文件大小：23k
源码类别：

MySQL数据库

开发平台：

SQL

ReverseEngineeringMssql.java：源码内容
  1. package com.mysql.grt.modules;
  3. import java.sql.*;
  5. import com.mysql.grt.*;
  6. import com.mysql.grt.db.IndexColumn;
  7. import com.mysql.grt.db.mssql.*;
  9. /**
  10.  * GRT Reverse Engineering Class for MSSQL 2000
  11.  * 
  12.  * @author MikeZ
  13.  * @version 1.0, 05/23/04
  14.  * 
  15.  */
  16. public class ReverseEngineeringMssql extends ReverseEngineeringGeneric {
  18. /**
  19.  * Static function to return information about this class to the GRT
  20.  * environment
  21.  * 
  22.  * @return returns a GRT XML string containing the infos about this class
  23.  */
  24. public static String getModuleInfo() {
  25. return Grt.getModuleInfoXml(ReverseEngineeringMssql.class,
  26. "ReverseEngineering");
  27. }
  29. /*
  30.  * private static String catalogsSelect = "SELECT TABLE_SCHEMA FROM
  31.  * INFORMATION_SCHEMA.TABLES " + "GROUP BY TABLE_SCHEMA UNION " + "SELECT
  32.  * ROUTINE_SCHEMA AS SCHEMANAME FROM INFORMATION_SCHEMA.ROUTINES " + "GROUP
  33.  * BY ROUTINE_SCHEMA";
  34.  */
  36. /**
  37.  * Returns a list of all catalogs from the given JDBC connection
  38.  * 
  39.  * @param dbConn
  40.  *            the connection to use
  41.  * 
  42.  * @return returns a GRT XML string containing a list of schemata names
  43.  */
  44. public static GrtStringList getCatalogs(
  45. com.mysql.grt.db.mgmt.Connection dbConn) throws Exception {
  46. GrtStringList catalogList = new GrtStringList();
  48. GrtStringHashMap paramValues = dbConn.getParameterValues();
  49. paramValues.add("database", "");
  51. // connect to the database
  52. Connection conn = establishConnection(dbConn);
  53. try {
  55. Grt.getInstance().addMsg("Fetching catalog list.");
  56. Grt.getInstance().addMsgDetail("CALL sp_databases;");
  57. Grt.getInstance().flushMessages();
  59. CallableStatement stmt = conn.prepareCall("sp_databases");
  60. try {
  61. ResultSet rset = stmt.executeQuery();
  62. while (rset.next()) {
  63. catalogList.add(rset.getString(1));
  64. }
  65. } finally {
  66. stmt.close();
  67. }
  68. } finally {
  69. conn.close();
  70. }
  72. return catalogList;
  73. }
  75. private static String schemataSelect = "SELECT TABLE_SCHEMA AS SCHEMANAME, "
  76. + " max(TABLE_CATALOG) AS CATALOGNAME FROM INFORMATION_SCHEMA.TABLES "
  77. + "GROUP BY TABLE_SCHEMA UNION "
  78. + "SELECT ROUTINE_SCHEMA AS SCHEMANAME, max(ROUTINE_CATALOG) AS CATALOGNAME "
  79. + "FROM INFORMATION_SCHEMA.ROUTINES " + "GROUP BY ROUTINE_SCHEMA";
  81. private static String schemataSelect70 = "SELECT TABLE_SCHEMA AS SCHEMANAME, "
  82. + " max(TABLE_CATALOG) AS CATALOGNAME FROM INFORMATION_SCHEMA.TABLES "
  83. + "GROUP BY TABLE_SCHEMA";
  85. /**
  86.  * Returns a list of all schemata from the given JDBC connection
  87.  * 
  88.  * @param dbConn
  89.  *            the connection to use
  90.  * 
  91.  * @return returns a GRT XML string containing a list of schemata names
  92.  */
  93. public static GrtStringList getSchemata(
  94. com.mysql.grt.db.mgmt.Connection dbConn) throws Exception {
  96. GrtStringList schemataList = new GrtStringList();
  98. // connect to the database
  99. Connection conn = establishConnection(dbConn);
  100. try {
  101. String sql;
  103. DatabaseMetaData metaData = conn.getMetaData();
  104. if (metaData.getDatabaseMajorVersion() <= 7)
  105. sql = schemataSelect70;
  106. else
  107. sql = schemataSelect;
  109. Grt.getInstance().addMsg("Fetching schemata list.");
  110. Grt.getInstance().addMsgDetail(sql);
  111. Grt.getInstance().flushMessages();
  113. Statement stmt = conn.createStatement();
  114. try {
  115. ResultSet rset = stmt.executeQuery(sql);
  116. while (rset.next()) {
  117. schemataList.add(rset.getString(2) + "."
  118. + rset.getString(1));
  119. }
  120. } finally {
  121. stmt.close();
  122. }
  123. } finally {
  124. conn.close();
  125. }
  127. Grt.getInstance().addMsg("Return schemata list.");
  128. Grt.getInstance().flushMessages();
  130. return schemataList;
  131. }
  133. /**
  134.  * Does the reverse engineering of the given schematas over the JDBC
  135.  * connection and returns the GRT objects
  136.  * 
  137.  * @param jdbcDriver
  138.  *            the class name of the JDBC driver
  139.  * @param jdbcConnectionString
  140.  *            a JDBC connection string
  141.  * @param schemataList
  142.  *            list of schematas to be reverse engineered
  143.  * @return returns a GRT XML string containing a the reverse engineered
  144.  *         objects
  145.  */
  146. public static com.mysql.grt.db.Catalog reverseEngineer(
  147. com.mysql.grt.db.mgmt.Connection dbConn, GrtStringList schemataList)
  148. throws Exception {
  150. boolean reverseEngineerOnlyTableObjects = (Grt.getInstance()
  151. .getGrtGlobalAsInt(
  152. "/migration/applicationData/"
  153. + "reverseEngineerOnlyTableObjects") == 1);
  155. // connect to the database
  156. Connection conn = establishConnection(dbConn);
  158. // create reveng instance
  159. ReverseEngineeringMssql revEng = new ReverseEngineeringMssql();
  161. Catalog catalog = new Catalog(null);
  162. catalog.setName(dbConn.getParameterValues().get("database"));
  164. catalog.setVersion(getVersion(dbConn));
  166. Grt.getInstance().addMsg("Build simple MSSQL datatypes.");
  167. Grt.getInstance().flushMessages();
  168. revEng.buildSimpleDatatypes(dbConn, catalog);
  170. for (int i = 0; i < schemataList.size(); i++) {
  171. String schemaName = schemataList.get(i);
  172. schemaName = schemaName.substring(catalog.getName().length() + 1);
  174. Schema schema = new Schema(catalog);
  175. schema.setName(schemaName);
  176. catalog.getSchemata().add(schema);
  178. // Get Tables
  179. if (revEng.reverseEngineerTables(conn, catalog, schema) == 0
  180. && !reverseEngineerOnlyTableObjects) {
  182. // Get Views
  183. revEng.reverseEngineerViews(conn, catalog, schema);
  185. // Get SPs
  186. revEng.reverseEngineerProcedures(conn, catalog, schema);
  187. }
  188. }
  190. // make sure the Fks use real references instead of
  191. // text names where possible
  192. revEng.reverseEngineerUpdateFkReferences(catalog);
  194. return catalog;
  195. }
  197. protected void buildSimpleDatatypes(
  198. com.mysql.grt.db.mgmt.Connection dbConn, Catalog catalog) {
  199. com.mysql.grt.db.mgmt.Driver driver = dbConn.getDriver();
  200. com.mysql.grt.db.mgmt.Rdbms rdbms = (com.mysql.grt.db.mgmt.Rdbms) driver
  201. .getOwner();
  202. com.mysql.grt.db.SimpleDatatypeList rdbmsDatatypeList = rdbms
  203. .getSimpleDatatypes();
  204. com.mysql.grt.db.SimpleDatatypeList schemaDatatypeList = new com.mysql.grt.db.SimpleDatatypeList();
  206. for (int i = 0; i < rdbmsDatatypeList.size(); i++) {
  207. schemaDatatypeList.add(rdbmsDatatypeList.get(i));
  208. }
  210. catalog.setSimpleDatatypes(schemaDatatypeList);
  211. }
  213. private static String tableCountSelect = "SELECT COUNT(*) AS TABLECOUNT "
  214. + "FROM INFORMATION_SCHEMA.TABLES "
  215. + "WHERE TABLE_TYPE='BASE TABLE' AND TABLE_SCHEMA=?";
  217. private static String tableSelect = "SELECT * FROM INFORMATION_SCHEMA.TABLES "
  218. + "WHERE TABLE_TYPE='BASE TABLE' AND TABLE_SCHEMA=?";
  220. protected int reverseEngineerTables(Connection conn, Catalog catalog,
  221. Schema schema) throws Exception {
  223. int tableCount = 0;
  224. int currentTableNumber = 0;
  226. Grt.getInstance().addMsg(
  227. "Fetch the number of tables in the schema " + schema.getName()
  228. + ".");
  229. Grt.getInstance().addMsgDetail(tableCountSelect);
  231. PreparedStatement stmt = conn.prepareStatement(tableCountSelect);
  232. stmt.setString(1, schema.getName());
  234. ResultSet tblRset = stmt.executeQuery();
  235. if (tblRset.next()) {
  236. tableCount = tblRset.getInt(1);
  237. }
  238. stmt.close();
  240. Grt.getInstance().addMsg(
  241. "Fetching " + tableCount + " table(s) of the schema "
  242. + schema.getName() + ".");
  243. Grt.getInstance().addMsgDetail(tableSelect);
  244. Grt.getInstance().flushMessages();
  246. stmt = conn.prepareStatement(tableSelect);
  247. stmt.setString(1, schema.getName());
  249. tblRset = stmt.executeQuery();
  251. while (tblRset.next()) {
  252. // Create new table
  253. Table table = new Table(schema);
  254. schema.getTables().add(table);
  256. currentTableNumber++;
  258. table.setName(tblRset.getString("TABLE_NAME"));
  260. Grt.getInstance().addProgress(
  261. "Processing table " + table.getName() + ".",
  262. (currentTableNumber * 100) / tableCount);
  263. if (Grt.getInstance().flushMessages() != 0) {
  264. Grt.getInstance().addMsg("Migration canceled by user.");
  265. return 1;
  266. }
  268. reverseEngineerTableColumns(conn, catalog, schema, table);
  270. reverseEngineerTablePK(conn, catalog, schema, table);
  272. reverseEngineerTableIndices(conn, catalog, schema, table);
  274. reverseEngineerTableFKs(conn, catalog, schema, table);
  275. }
  277. Grt.getInstance().addProgress("", -1);
  278. Grt.getInstance().flushMessages();
  280. stmt.close();
  282. return 0;
  283. }
  285. private static String tableColumnSelect = "SELECT COLUMN_NAME, COLUMN_DEFAULT, "
  286. + " IS_NULLABLE, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH, CHARACTER_OCTET_LENGTH, "
  287. + " NUMERIC_PRECISION, NUMERIC_PRECISION_RADIX, NUMERIC_SCALE, DATETIME_PRECISION, "
  288. + " CHARACTER_SET_CATALOG, CHARACTER_SET_SCHEMA, CHARACTER_SET_NAME, "
  289. + " COLLATION_NAME, DOMAIN_CATALOG, DOMAIN_SCHEMA, DOMAIN_NAME, "
  290. + " (c.status & 128) / 128 AS IS_IDENTITY_COLUMN "
  291. + "FROM INFORMATION_SCHEMA.COLUMNS, sysobjects t, sysusers u, syscolumns c "
  292. + "WHERE TABLE_SCHEMA=? AND TABLE_NAME=? AND "
  293. + "u.name=TABLE_SCHEMA AND t.name=TABLE_NAME AND "
  294. + "u.uid=t.uid AND c.id=t.id AND "
  295. + "c.name=COLUMN_NAME "
  296. + "ORDER BY ORDINAL_POSITION";
  298. private static String tableColumnSelect2005 = "SELECT COLUMN_NAME, COLUMN_DEFAULT, "
  299. + " IS_NULLABLE, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH, CHARACTER_OCTET_LENGTH, "
  300. + " NUMERIC_PRECISION, NUMERIC_PRECISION_RADIX, NUMERIC_SCALE, DATETIME_PRECISION, "
  301. + " CHARACTER_SET_CATALOG, CHARACTER_SET_SCHEMA, CHARACTER_SET_NAME, "
  302. + " COLLATION_NAME, DOMAIN_CATALOG, DOMAIN_SCHEMA, DOMAIN_NAME, "
  303. + " (c.status & 128) / 128 AS IS_IDENTITY_COLUMN "
  304. + "FROM INFORMATION_SCHEMA.COLUMNS, sysobjects t, syscolumns c "
  305. + "WHERE TABLE_SCHEMA=? AND TABLE_NAME=? AND "
  306. + "t.name=TABLE_NAME AND "
  307. + "c.id=t.id AND "
  308. + "c.name=COLUMN_NAME "
  309. + "ORDER BY ORDINAL_POSITION";
  311. protected void reverseEngineerTableColumns(Connection conn,
  312. Catalog catalog, Schema schema, Table table) {
  314. try {
  315. Grt.getInstance().addMsg("Fetching column information.");
  317. int version = conn.getMetaData().getDatabaseMajorVersion();
  318. String columnSelect;
  319. switch (version)
  320. {
  321. case 9:
  322. // Starting with version 9 the assignments between schemata and sysusers seems to have changed. 
  323. columnSelect = tableColumnSelect2005;
  324. break;
  325. default:
  326. columnSelect = tableColumnSelect;
  327. }
  328. Grt.getInstance().addMsgDetail(columnSelect);
  330. PreparedStatement stmt = conn.prepareStatement(columnSelect);
  331. stmt.setString(1, schema.getName());
  332. stmt.setString(2, table.getName());
  334. ResultSet colRset = stmt.executeQuery();
  335. while (colRset.next()) {
  336. // create new column
  337. Column column = new Column(table);
  338. table.getColumns().add(column);
  340. column.setName(colRset.getString("COLUMN_NAME"));
  341. column.setDatatypeName(colRset.getString("DATA_TYPE"));
  343. // Get Simple Type
  344. int datatypeIndex = catalog.getSimpleDatatypes()
  345. .getIndexOfName(column.getDatatypeName().toUpperCase());
  347. if (datatypeIndex > -1) {
  348. column.setSimpleType(catalog.getSimpleDatatypes().get(
  349. datatypeIndex));
  350. } else {
  351. column.setSimpleType(catalog.getSimpleDatatypes().get(
  352. catalog.getSimpleDatatypes().getIndexOfName(
  353. "VARCHAR")));
  354. column.setLength(255);
  356. Grt.getInstance().addMsg(
  357. "WARNING: The datatype " + column.getDatatypeName()
  358. + " was not been defined yet.");
  359. }
  361. column.setLength(colRset.getInt("CHARACTER_MAXIMUM_LENGTH"));
  362. column.setPrecision(colRset.getInt("NUMERIC_PRECISION"));
  363. column.setScale(colRset.getInt("NUMERIC_SCALE"));
  365. // Nullable
  366. if (colRset.getString("IS_NULLABLE").compareToIgnoreCase("YES") == 0) {
  367. column.setIsNullable(1);
  368. } else {
  369. column.setIsNullable(0);
  370. }
  372. // Character set
  373. column.setCharacterSetName(colRset
  374. .getString("CHARACTER_SET_NAME"));
  376. column.setCollationName(colRset.getString("COLLATION_NAME"));
  378. // Default Value
  379. String defaultValue = colRset.getString("COLUMN_DEFAULT");
  380. if (defaultValue != null) {
  381. // remove () and (N) from (N'xxxx') but not from (NULL)
  382. if ((defaultValue.startsWith("(N")) && (defaultValue
  383. .endsWith(")"))
  384. && !(defaultValue.equalsIgnoreCase("(NULL)")))
  385. defaultValue = defaultValue.substring(2, defaultValue
  386. .length() - 1);
  387. else if (!defaultValue.equals(""))
  388. defaultValue = defaultValue.substring(1, defaultValue
  389. .length() - 1);
  391. // remove (()) from numeric types
  392. if ((column.getSimpleType().getGroup().getName()
  393. .equals("numeric"))
  394. && defaultValue.startsWith("((")
  395. && defaultValue.endsWith("))")) {
  396. defaultValue = defaultValue.substring(2, defaultValue
  397. .length() - 3);
  398. }
  400. // remove () from numeric types
  401. if ((column.getSimpleType().getGroup().getName()
  402. .equals("numeric"))
  403. && defaultValue.startsWith("(")
  404. && defaultValue.endsWith(")")) {
  405. defaultValue = defaultValue.substring(1, defaultValue
  406. .length() - 1);
  407. }
  409. column.setDefaultValue(defaultValue);
  410. }
  412. // Identity column
  413. column.setIdentity(colRset.getInt("IS_IDENTITY_COLUMN"));
  414. }
  416. stmt.close();
  417. } catch (Exception e) {
  418. Grt.getInstance().addErr(e.getMessage());
  419. }
  420. }
  422. private static String tablePKSP = "sp_pkeys @table_owner=?, @table_name=?";
  424. protected void reverseEngineerTablePK(Connection conn, Catalog catalog,
  425. Schema schema, Table table) {
  427. // String sql;
  429. try {
  430. Grt.getInstance().addMsg("Fetching primary key information.");
  431. Grt.getInstance().addMsgDetail(tablePKSP);
  433. PreparedStatement stmt = conn.prepareCall(tablePKSP);
  434. stmt.setString(1, schema.getName());
  435. stmt.setString(2, table.getName());
  437. ResultSet colRset = stmt.executeQuery();
  439. Index primaryKey = null;
  441. while (colRset.next()) {
  442. if (primaryKey == null) {
  443. primaryKey = new Index(table);
  444. primaryKey.setName("PRIMARY");
  446. primaryKey.setIsPrimary(1);
  448. table.getIndices().add(primaryKey);
  450. table.setPrimaryKey(primaryKey);
  451. }
  453. IndexColumn indexColumn = new IndexColumn(primaryKey);
  454. indexColumn.setName(colRset.getString("COLUMN_NAME"));
  455. indexColumn.setColumnLength(0);
  456. indexColumn.setDescend(0);
  458. // find reference table column
  459. for (int j = 0; j < table.getColumns().size(); j++) {
  460. Column column = (Column) (table.getColumns().get(j));
  462. if (column.getName().compareToIgnoreCase(
  463. indexColumn.getName()) == 0) {
  464. indexColumn.setReferedColumn(column);
  465. break;
  466. }
  467. }
  469. primaryKey.getColumns().add(indexColumn);
  470. }
  472. stmt.close();
  473. } catch (Exception e) {
  474. Grt.getInstance().addErr(e.getMessage());
  475. }
  476. }
  478. private static String tableIndexSelect = "SELECT u.name as TABLE_SCHEMA, "
  479. + " o.name as TABLE_NAME, i.name AS INDEX_NAME, c.name AS COLUMN_NAME, "
  480. + " (i.status & 1) AS IGNORE_DUPLICATE_KEYS, "
  481. + " (i.status & 2) / 2 AS IS_UNIQUE, "
  482. + " (i.status & 4) / 4 AS IGNORE_DUPLICATE_ROWS, "
  483. + " (i.status & 16) / 16 AS IS_CLUSTERED, "
  484. + " (i.status & 2048) / 2048 AS IS_PRIMARY_KEY, "
  485. + " (i.status & 4096) / 4096 AS IS_UNIQUE_KEY "
  486. + "FROM sysindexes i, sysobjects o, sysusers u, sysindexkeys k, syscolumns c "
  487. + "WHERE u.uid=o.uid AND i.id = o.id AND k.indid=i.indid AND k.id=i.id AND "
  488. + " c.id=i.id AND c.colid=k.colid AND "
  489. + " i.indid > 0 AND i.indid < 255 AND o.type = 'U' AND "
  490. + " (i.status & 64)=0 AND (i.status & 8388608)=0 AND "
  491. + " (i.status & 2048)=0 AND " + " u.name=? AND o.name=? "
  492. + "ORDER BY i.name, k.keyno";
  494. protected void reverseEngineerTableIndices(Connection conn,
  495. Catalog catalog, Schema schema, Table table) {
  497. try {
  498. Grt.getInstance().addMsg("Fetching indices information.");
  499. Grt.getInstance().addMsgDetail(tableIndexSelect);
  501. PreparedStatement stmt = conn.prepareStatement(tableIndexSelect);
  502. stmt.setString(1, schema.getName());
  503. stmt.setString(2, table.getName());
  505. ResultSet rset = stmt.executeQuery();
  507. String indexName = "";
  508. Index index = null;
  510. while (rset.next()) {
  511. String newIndexName = rset.getString("INDEX_NAME");
  513. if ((rset.getInt("IS_PRIMARY_KEY") != 0)
  514. || (newIndexName.equalsIgnoreCase("PRIMARY")))
  515. continue;
  517. if (indexName.compareToIgnoreCase(newIndexName) != 0) {
  518. if (index != null)
  519. table.getIndices().add(index);
  521. index = new Index(table);
  522. index.setName(newIndexName);
  523. indexName = newIndexName;
  525. if (rset.getInt("IS_UNIQUE") != 0)
  526. index.setUnique(1);
  527. else
  528. index.setUnique(0);
  530. index.setDeferability(0);
  532. index.setClustered(rset.getInt("IS_CLUSTERED"));
  533. index.setIgnoreDuplicateRows(rset
  534. .getInt("IGNORE_DUPLICATE_ROWS"));
  535. }
  537. IndexColumn indexColumn = new IndexColumn(index);
  538. indexColumn.setName(rset.getString("COLUMN_NAME"));
  539. indexColumn.setColumnLength(0);
  540. indexColumn.setDescend(0);
  542. // find reference table column
  543. for (int j = 0; j < table.getColumns().size(); j++) {
  544. Column column = (Column) (table.getColumns().get(j));
  546. if (column.getName().compareToIgnoreCase(
  547. indexColumn.getName()) == 0) {
  548. indexColumn.setReferedColumn(column);
  549. break;
  550. }
  551. }
  553. index.getColumns().add(indexColumn);
  554. }
  556. if (index != null)
  557. table.getIndices().add(index);
  559. stmt.close();
  560. } catch (Exception e) {
  561. Grt.getInstance().addErr(e.getMessage());
  562. }
  563. }
  565. private static String tableFKSelect = "SELECT fk.name AS CONSTRAINT_NAME, "
  566. + " c.name AS COLUMN_NAME, ref_u.name AS REF_SCHEMA_NAME, "
  567. + " ref_tbl.name AS REF_TABLE_NAME, ref_c.name AS REF_COLUMN_NAME, "
  568. + " CASE WHEN (ObjectProperty(sfk.constid, 'CnstIsUpdateCascade')=1) THEN "
  569. + "  'CASCADE' ELSE 'NO ACTION' END as UPDATE_RULE, "
  570. + " CASE WHEN (ObjectProperty(sfk.constid, 'CnstIsDeleteCascade')=1) THEN "
  571. + "  'CASCADE' ELSE 'NO ACTION' END as DELETE_RULE "
  572. + "FROM sysusers u, sysobjects t, sysobjects fk, sysforeignkeys sfk, "
  573. + " syscolumns c, sysobjects ref_tbl, sysusers ref_u, syscolumns ref_c "
  574. + "WHERE u.name=? AND t.name=? AND "
  575. + " t.uid=u.uid AND t.xtype='U' AND "
  576. + " sfk.fkeyid=t.id AND fk.id=sfk.constid AND "
  577. + " c.id=t.id AND c.colid=sfk.fkey AND "
  578. + " ref_tbl.id=sfk.rkeyid AND ref_tbl.uid=ref_u.uid AND "
  579. + " ref_c.id=ref_tbl.id AND ref_c.colid=sfk.rkey "
  580. + "ORDER BY sfk.constid, sfk.keyno";
  582. protected void reverseEngineerTableFKs(Connection conn, Catalog catalog,
  583. Schema schema, Table table) {
  585. try {
  586. Grt.getInstance().addMsg("Fetching FK information.");
  587. Grt.getInstance().addMsgDetail(tableFKSelect);
  589. PreparedStatement stmt = conn.prepareStatement(tableFKSelect);
  590. stmt.setString(1, schema.getName());
  591. stmt.setString(2, table.getName());
  593. ResultSet rset = stmt.executeQuery();
  595. String fkName = "";
  596. ForeignKey foreignKey = null;
  598. while (rset.next()) {
  599. String newFkName = rset.getString("CONSTRAINT_NAME");
  601. if (fkName.compareToIgnoreCase(newFkName) != 0) {
  602. if (foreignKey != null)
  603. table.getForeignKeys().add(foreignKey);
  605. fkName = newFkName;
  607. foreignKey = new ForeignKey(table);
  608. foreignKey.setName(newFkName);
  610. foreignKey.setDeferability(0);
  612. foreignKey.setDeleteRule(rset.getString("DELETE_RULE"));
  613. foreignKey.setUpdateRule(rset.getString("UPDATE_RULE"));
  615. foreignKey.setReferedTableSchemaName(rset
  616. .getString("REF_SCHEMA_NAME"));
  617. foreignKey.setReferedTableName(rset
  618. .getString("REF_TABLE_NAME"));
  619. }
  621. foreignKey.getReferedColumnNames().add(
  622. rset.getString("REF_COLUMN_NAME"));
  624. // find reference table column
  625. String colName = rset.getString("COLUMN_NAME");
  626. for (int j = 0; j < table.getColumns().size(); j++) {
  627. Column column = (Column) (table.getColumns().get(j));
  629. if (column.getName().compareToIgnoreCase(colName) == 0)
  630. foreignKey.getColumns().add(column);
  631. }
  632. }
  634. if (foreignKey != null)
  635. table.getForeignKeys().add(foreignKey);
  637. stmt.close();
  638. } catch (Exception e) {
  639. Grt.getInstance().addErr(e.getMessage());
  640. }
  641. }
  643. private static String viewSelect = "SELECT * FROM INFORMATION_SCHEMA.VIEWS "
  644. + "WHERE TABLE_SCHEMA=?";
  646. protected void reverseEngineerViews(Connection conn, Catalog catalog,
  647. Schema schema) throws Exception {
  649. Grt.getInstance().addMsg(
  650. "Fetch all views of the schema " + schema.getName() + ".");
  651. Grt.getInstance().addMsgDetail(viewSelect);
  652. Grt.getInstance().flushMessages();
  654. PreparedStatement stmt = conn.prepareStatement(viewSelect);
  655. stmt.setString(1, schema.getName());
  657. ResultSet rset = stmt.executeQuery();
  659. while (rset.next()) {
  660. // Create new view
  661. View view = new View(schema);
  662. schema.getViews().add(view);
  664. view.setName(rset.getString("TABLE_NAME"));
  666. Grt.getInstance().addMsg("Processing view " + view.getName() + ".");
  668. view.setQueryExpression(rset.getString("VIEW_DEFINITION"));
  670. if (rset.getString("CHECK_OPTION").equalsIgnoreCase("NONE"))
  671. view.setWithCheckCondition(0);
  672. else
  673. view.setWithCheckCondition(1);
  674. }
  676. stmt.close();
  678. Grt.getInstance().addMsg("Views fetched.");
  679. }
  681. private static String procedureCountSelect = "SELECT COUNT(*) AS NUM "
  682. + "FROM INFORMATION_SCHEMA.ROUTINES WHERE ROUTINE_SCHEMA=?";
  684. private static String procedureSelect = "SELECT ROUTINE_NAME, ROUTINE_TYPE, "
  685. + "MODULE_NAME, ROUTINE_DEFINITION "
  686. + "FROM INFORMATION_SCHEMA.ROUTINES WHERE ROUTINE_SCHEMA=?";
  688. protected int reverseEngineerProcedures(Connection conn, Catalog catalog,
  689. Schema schema) throws Exception {
  690. int spCount = 0;
  691. int currentSpNumber = 0;
  693. Grt.getInstance().addMsg(
  694. "Fetch count of stored procedures of the schema "
  695. + schema.getName() + ".");
  696. Grt.getInstance().addMsgDetail(procedureCountSelect);
  698. try {
  699. PreparedStatement stmt = conn
  700. .prepareStatement(procedureCountSelect);
  701. stmt.setString(1, schema.getName());
  703. ResultSet rset = stmt.executeQuery();
  705. if (rset.next()) {
  706. spCount = rset.getInt("NUM");
  707. }
  709. Grt.getInstance().addMsg(
  710. "Fetching " + spCount
  711. + " stored procedure(s) of the schema "
  712. + schema.getName() + ".");
  713. Grt.getInstance().addMsgDetail(procedureSelect);
  714. Grt.getInstance().flushMessages();
  716. stmt = conn.prepareStatement(procedureSelect);
  717. stmt.setString(1, schema.getName());
  719. rset = stmt.executeQuery();
  721. while (rset.next()) {
  722. // Create new view
  723. Routine proc = new Routine(schema);
  724. schema.getRoutines().add(proc);
  726. proc.setName(rset.getString("ROUTINE_NAME"));
  728. currentSpNumber++;
  730. Grt.getInstance().addProgress(
  731. "Processing procedure " + proc.getName() + ".",
  732. (currentSpNumber * 100) / spCount);
  733. if (Grt.getInstance().flushMessages() != 0) {
  734. Grt.getInstance().addMsg("Migration canceled by user.");
  735. return 1;
  736. }
  738. Grt.getInstance().addMsg(
  739. "Processing procedure " + proc.getName() + ".");
  741. proc.setRoutineType(rset.getString("ROUTINE_TYPE"));
  743. proc.setRoutineCode(rset.getString("ROUTINE_DEFINITION"));
  744. }
  746. stmt.close();
  748. Grt.getInstance().addMsg("Routines fetched.");
  750. Grt.getInstance().addProgress("", -1);
  752. } catch (Exception e) {
  753. Grt.getInstance().addMsg("Stored procedures cannot be fetched.");
  754. Grt.getInstance().addMsgDetail(e.getMessage());
  755. }
  757. return 0;
  758. }
  759. }