傅行帆 5 år sedan
förälder
incheckning
db44aea921

+ 14
- 9
generator_service/src/main/java/com/example/demo/controller/DbController.java Visa fil

19
 import javax.servlet.http.HttpServletResponse;
19
 import javax.servlet.http.HttpServletResponse;
20
 import java.io.*;
20
 import java.io.*;
21
 import java.net.URLEncoder;
21
 import java.net.URLEncoder;
22
-import java.sql.Connection;
23
-import java.sql.DatabaseMetaData;
24
-import java.sql.DriverManager;
25
-import java.sql.ResultSet;
22
+import java.sql.*;
26
 import java.util.ArrayList;
23
 import java.util.ArrayList;
27
 import java.util.List;
24
 import java.util.List;
28
 import java.util.UUID;
25
 import java.util.UUID;
37
         String mysqlUrl = "jdbc:mysql://"+dbConfig.getMysqlIp()+":"+dbConfig.getMysqlPort()+"/"+dbConfig.getMysqlName()+"?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true";
34
         String mysqlUrl = "jdbc:mysql://"+dbConfig.getMysqlIp()+":"+dbConfig.getMysqlPort()+"/"+dbConfig.getMysqlName()+"?useUnicode=true&characterEncoding=UTF-8&allowMultiQueries=true";
38
         try {
35
         try {
39
             Connection con = DriverManager.getConnection(mysqlUrl,dbConfig.getMysqlUsername(),dbConfig.getMysqlUserpass());
36
             Connection con = DriverManager.getConnection(mysqlUrl,dbConfig.getMysqlUsername(),dbConfig.getMysqlUserpass());
40
-            DatabaseMetaData meta = con.getMetaData();
41
-            ResultSet rs = meta.getTables(null, null, null,
42
-                    new String[] { "TABLE" });
43
-            while (rs.next()) {
44
-                tables.add(rs.getString(3));
37
+//            DatabaseMetaData meta = con.getMetaData();
38
+//            ResultSet rs = meta.getTables(null, dbConfig.getMysqlUsername(), null,
39
+//                    new String[] { "TABLE" });
40
+//            while (rs.next()) {
41
+//                tables.add(rs.getString(3));
42
+//            }
43
+            String sql = "select table_name from information_schema.tables where table_schema='" + dbConfig.getMysqlName() + "'";
44
+            Statement statement = con.createStatement();
45
+            ResultSet result = statement.executeQuery(sql);
46
+            while(result.next())
47
+            {
48
+                tables.add(result.getString("TABLE_NAME"));
45
             }
49
             }
46
             con.close();
50
             con.close();
47
         } catch (Exception e) {
51
         } catch (Exception e) {
52
 
56
 
53
     @PostMapping("/create")
57
     @PostMapping("/create")
54
     public void create(@RequestBody DbConfig dbConfig, HttpServletRequest request, HttpServletResponse response) throws UnsupportedEncodingException {
58
     public void create(@RequestBody DbConfig dbConfig, HttpServletRequest request, HttpServletResponse response) throws UnsupportedEncodingException {
59
+//        String newFileName = "D:/" + UUID.randomUUID().toString().replaceAll("-", "") + "/" + "cj-demo";
55
         String newFileName = "/home/yansen/generator/temp/" + UUID.randomUUID().toString().replaceAll("-", "") + "/" + "cj-demo";
60
         String newFileName = "/home/yansen/generator/temp/" + UUID.randomUUID().toString().replaceAll("-", "") + "/" + "cj-demo";
56
         dbConfig.getMysqlLib().forEach(e -> {
61
         dbConfig.getMysqlLib().forEach(e -> {
57
             AutoGeneratorFile(dbConfig,newFileName,e);
62
             AutoGeneratorFile(dbConfig,newFileName,e);