add sqlite support

anealo · Aug 22, 2017 · e1db858 · e1db858
1 parent 9dc2d5f
commit e1db858
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/src/main/java/com/zzg/mybatis/generator/model/DbType.java b/src/main/java/com/zzg/mybatis/generator/model/DbType.java
@@ -8,7 +8,8 @@ public enum DbType {
     MySQL("com.mysql.jdbc.Driver", "jdbc:mysql://%s:%s/%s?useUnicode=true&useSSL=false&characterEncoding=%s", "mysql-connector-java-5.1.38.jar"),
     Oracle("oracle.jdbc.driver.OracleDriver", "jdbc:oracle:thin:@%s:%s:%s", "ojdbc14.jar"),
     PostgreSQL("org.postgresql.Driver", "jdbc:postgresql://%s:%s/%s", "postgresql-9.4.1209.jar"),
-	SQL_Server("com.microsoft.sqlserver.jdbc.SQLServerDriver", "jdbc:sqlserver://%s:%s;databaseName=%s", "sqljdbc4-4.0.jar");
+	SQL_Server("com.microsoft.sqlserver.jdbc.SQLServerDriver", "jdbc:sqlserver://%s:%s;databaseName=%s", "sqljdbc4-4.0.jar"),
+	Sqlite("org.sqlite.JDBC", "jdbc:sqlite:%s", "sqlite-jdbc-3.19.3.jar");
 
     private final String driverClass;
     private final String connectionUrlPattern;

diff --git a/src/main/java/com/zzg/mybatis/generator/util/DbUtil.java b/src/main/java/com/zzg/mybatis/generator/util/DbUtil.java
@@ -66,7 +66,14 @@ public static List<String> getTableNames(DatabaseConfig config) throws Exception
 			    }
 		    } else if (DbType.valueOf(config.getDbType()) == DbType.Oracle){
 			    rs = md.getTables(null, config.getUsername().toUpperCase(), null, new String[] {"TABLE", "VIEW"});
-		    } else {
+		    } else if (DbType.valueOf(config.getDbType())==DbType.Sqlite){
+		    	String sql = "Select name from sqlite_master;";
+			    rs = connection.createStatement().executeQuery(sql);
+			    while (rs.next()) {
+				    tables.add(rs.getString("name"));
+			    }
+		    } 
+		    else {
 			    // rs = md.getTables(null, config.getUsername().toUpperCase(), null, null);
 				rs = md.getTables(null, "%", "%", new String[] {"TABLE", "VIEW"});			//针对 postgresql 的左侧数据表显示
 		    }

diff --git a/src/main/resources/fxml/newConnection.fxml b/src/main/resources/fxml/newConnection.fxml
@@ -33,6 +33,7 @@
                         <String fx:value="Oracle" />
                         <String fx:value="PostgreSQL" />
                         <String fx:value="SQL_Server" />
+                        <String fx:value="Sqlite" />
                     </FXCollections>
                 </items>
             </ChoiceBox>

diff --git a/src/main/resources/lib/sqlite-jdbc-3.19.3.jar b/src/main/resources/lib/sqlite-jdbc-3.19.3.jar