添加生成insert sql参数列表字符串的代码

wxdtkkd · Aug 23, 2016 · 7c0bd39 · 7c0bd39
1 parent 8260a86
commit 7c0bd39
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 14 deletions.
diff --git a/src/main/java/cn/binarywang/tools/generator/InsertSQLGenerator.java b/src/main/java/cn/binarywang/tools/generator/InsertSQLGenerator.java
@@ -1,5 +1,7 @@
 package cn.binarywang.tools.generator;
 
+import static com.google.common.collect.Collections2.transform;
+
 import java.sql.Connection;
 import java.sql.DriverManager;
 import java.sql.PreparedStatement;
@@ -9,6 +11,8 @@
 import java.util.Collections;
 import java.util.List;
 
+import com.google.common.base.CaseFormat;
+import com.google.common.base.Function;
 import com.google.common.base.Joiner;
 import com.google.common.collect.Lists;
 
@@ -32,33 +36,45 @@ public InsertSQLGenerator(String url, String username, String password,
     }
 
     public String generateSQL() {
+        List<String> columns = getColumns();
+
+        return String.format("insert into %s(%s) values(%s)", this.tableName,
+            COMMA_JOINER.join(columns),
+            COMMA_JOINER.join(Collections.nCopies(columns.size(), "?")));
+    }
+
+    public String generateParams() {
+        return COMMA_JOINER
+            .join(transform(getColumns(), new Function<String, String>() {
+
+                @Override
+                public String apply(String input) {
+                    return "abc.get" + CaseFormat.LOWER_UNDERSCORE
+                        .to(CaseFormat.UPPER_CAMEL, input) + "()";
+                }
+            }));
+    }
+
+    private List<String> getColumns() {
+        List<String> columns = Lists.newArrayList();
         try (PreparedStatement ps = this.con
             .prepareStatement("select * from " + this.tableName);
                 ResultSet rs = ps.executeQuery();) {
 
             ResultSetMetaData rsm = rs.getMetaData();
-            List<String> columns = Lists.newArrayList();
             for (int i = 1; i <= rsm.getColumnCount(); i++) {
                 String columnName = rsm.getColumnName(i);
-                String dbType = rsm.getColumnTypeName(i);
-
                 System.out.print("Name: " + columnName);
-                System.out.println(", DBType : " + dbType);
+                System.out.println(", Type : " + rsm.getColumnClassName(i));
                 columns.add(columnName);
             }
 
-            return String.format("insert into %s(%s) values(%s)",
-                this.tableName, COMMA_JOINER.join(columns),
-                COMMA_JOINER.join(Collections.nCopies(columns.size(), "?")));
         } catch (SQLException e) {
             e.printStackTrace();
+            return null;
         }
 
-        return null;
-    }
-
-    public String generateParams() {
-        return null;
+        return columns;
     }
 
     public void close() {

diff --git a/src/test/java/cn/binarywang/tools/generator/InsertSQLGeneratorTest.java b/src/test/java/cn/binarywang/tools/generator/InsertSQLGeneratorTest.java
@@ -1,6 +1,5 @@
 package cn.binarywang.tools.generator;
 
-import org.testng.Assert;
 import org.testng.annotations.AfterTest;
 import org.testng.annotations.BeforeTest;
 import org.testng.annotations.Guice;
@@ -64,7 +63,7 @@ public void testGenerateSQL() {
      * .
      */
     public void testGenerateParams() {
-        Assert.fail("Not yet implemented");
+        System.err.println(this.generator.generateParams());
     }
 
 }