增加sql server支持

anealo · May 2, 2017 · 0ade97a · 0ade97a
1 parent 496df5b
commit 0ade97a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/pom.xml b/pom.xml
@@ -56,6 +56,11 @@
             <artifactId>ojdbc14</artifactId>
             <version>10.2.0.3.0</version>
         </dependency>
+        <dependency>
+            <groupId>com.microsoft.sqlserver</groupId>
+            <artifactId>sqljdbc4</artifactId>
+            <version>4.0</version>
+        </dependency>
         <dependency>
             <groupId>org.postgresql</groupId>
             <artifactId>postgresql</artifactId>

diff --git a/src/main/java/com/zzg/mybatis/generator/util/DbUtil.java b/src/main/java/com/zzg/mybatis/generator/util/DbUtil.java
@@ -32,7 +32,12 @@ public static List<String> getTableNames(DatabaseConfig config) throws Exception
         _LOG.info("getTableNames, connection url: {}", url);
         Connection conn = DriverManager.getConnection(url, config.getUsername(), config.getPassword());
         DatabaseMetaData md = conn.getMetaData();
-        ResultSet rs = md.getTables(null, config.getUsername().toUpperCase(), null, null);
+	    ResultSet rs;
+        if (DbType.valueOf(config.getDbType()) == DbType.SQL_Server) {
+	        rs = md.getTables(config.getSchema(), null, null, null);
+        } else {
+            rs = md.getTables(null, config.getUsername().toUpperCase(), null, new String[] {"TABLE"});
+        }
         List<String> tables = new ArrayList<>();
         while (rs.next()) {
             tables.add(rs.getString(3));

diff --git a/src/main/resources/lib/sqljdbc4-4.0.jar b/src/main/resources/lib/sqljdbc4-4.0.jar