mysql count语句需要添加别名

2023-03-04 19:17:16 +00:00 · 2013-07-29 11:25:53 +08:00
parent 0d7ff82c7b
commit abbf31d5d7
6 changed files with 272 additions and 270 deletions
@@ -284,7 +284,9 @@ public class PagerUtils {
        SQLSelectItem countItem = createCountItem(dbType);
        countSelectQuery.getSelectList().add(countItem);

-        countSelectQuery.setFrom(new SQLSubqueryTableSource(select));
+        SQLSubqueryTableSource fromSubquery = new SQLSubqueryTableSource(select);
+        fromSubquery.setAlias("ALIAS_COUNT_TABLE");
+        countSelectQuery.setFrom(fromSubquery);

        SQLSelect countSelect = new SQLSelect(countSelectQuery);
        SQLSelectStatement countStmt = new SQLSelectStatement(countSelect);
@@ -1,54 +1,54 @@
-package com.alibaba.druid.bvt.sql;
-
-import junit.framework.TestCase;
-
-import org.junit.Assert;
-
-import com.alibaba.druid.sql.PagerUtils;
-import com.alibaba.druid.util.JdbcConstants;
-
-public class PagerUtilsTest_Count_MySql_0 extends TestCase {
-
-    public void test_mysql_0() throws Exception {
-        String sql = "select * from t";
-        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_mysql_1() throws Exception {
-        String sql = "select id, name from t";
-        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_mysql_2() throws Exception {
-        String sql = "select id, name from t order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_mysql_group_0() throws Exception {
-        String sql = "select type, count(*) from t group by type";
-        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
-        Assert.assertEquals("SELECT COUNT(*)" + //
-                            "\nFROM (SELECT type, COUNT(*)" + //
-                            "\n\tFROM t" + //
-                            "\n\tGROUP BY type" + //
-                            "\n\t)", result);
-    }
-
-    public void test_mysql_union_0() throws Exception {
-        String sql = "select id, name from t1 union select id, name from t2 order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM (SELECT id, name" + //
-                            "\n\tFROM t1" + //
-                            "\n\tUNION" + //
-                            "\n\tSELECT id, name" + //
-                            "\n\tFROM t2" + //
-                            "\n\t)", result);
-    }
-}
+package com.alibaba.druid.bvt.sql;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PagerUtils;
+import com.alibaba.druid.util.JdbcConstants;
+
+public class PagerUtilsTest_Count_MySql_0 extends TestCase {
+
+    public void test_mysql_0() throws Exception {
+        String sql = "select * from t";
+        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_mysql_1() throws Exception {
+        String sql = "select id, name from t";
+        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_mysql_2() throws Exception {
+        String sql = "select id, name from t order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_mysql_group_0() throws Exception {
+        String sql = "select type, count(*) from t group by type";
+        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
+        Assert.assertEquals("SELECT COUNT(*)" + //
+                            "\nFROM (SELECT type, COUNT(*)" + //
+                            "\n\tFROM t" + //
+                            "\n\tGROUP BY type" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+
+    public void test_mysql_union_0() throws Exception {
+        String sql = "select id, name from t1 union select id, name from t2 order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.MYSQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM (SELECT id, name" + //
+                            "\n\tFROM t1" + //
+                            "\n\tUNION" + //
+                            "\n\tSELECT id, name" + //
+                            "\n\tFROM t2" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+}
@@ -1,54 +1,54 @@
-package com.alibaba.druid.bvt.sql;
-
-import junit.framework.TestCase;
-
-import org.junit.Assert;
-
-import com.alibaba.druid.sql.PagerUtils;
-import com.alibaba.druid.util.JdbcConstants;
-
-public class PagerUtilsTest_Count_Oracle_0 extends TestCase {
-
-    public void test_oracle_0() throws Exception {
-        String sql = "select * from t";
-        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_oracle_1() throws Exception {
-        String sql = "select id, name from t";
-        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_oracle_2() throws Exception {
-        String sql = "select id, name from t order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_oracle_group_0() throws Exception {
-        String sql = "select type, count(*) from t group by type";
-        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
-        Assert.assertEquals("SELECT COUNT(*)" + //
-                            "\nFROM (SELECT type, COUNT(*)" + //
-                            "\n\tFROM t" + //
-                            "\n\tGROUP BY type" + //
-                            "\n\t)", result);
-    }
-
-    public void test_oracle_union_0() throws Exception {
-        String sql = "select id, name from t1 union select id, name from t2 order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM (SELECT id, name" + //
-                            "\n\tFROM t1" + //
-                            "\n\tUNION" + //
-                            "\n\tSELECT id, name" + //
-                            "\n\tFROM t2" + //
-                            "\n\t)", result);
-    }
-}
+package com.alibaba.druid.bvt.sql;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PagerUtils;
+import com.alibaba.druid.util.JdbcConstants;
+
+public class PagerUtilsTest_Count_Oracle_0 extends TestCase {
+
+    public void test_oracle_0() throws Exception {
+        String sql = "select * from t";
+        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_oracle_1() throws Exception {
+        String sql = "select id, name from t";
+        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_oracle_2() throws Exception {
+        String sql = "select id, name from t order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_oracle_group_0() throws Exception {
+        String sql = "select type, count(*) from t group by type";
+        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
+        Assert.assertEquals("SELECT COUNT(*)" + //
+                            "\nFROM (SELECT type, COUNT(*)" + //
+                            "\n\tFROM t" + //
+                            "\n\tGROUP BY type" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+
+    public void test_oracle_union_0() throws Exception {
+        String sql = "select id, name from t1 union select id, name from t2 order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.ORACLE);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM (SELECT id, name" + //
+                            "\n\tFROM t1" + //
+                            "\n\tUNION" + //
+                            "\n\tSELECT id, name" + //
+                            "\n\tFROM t2" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+}
@@ -1,54 +1,54 @@
-package com.alibaba.druid.bvt.sql;
-
-import junit.framework.TestCase;
-
-import org.junit.Assert;
-
-import com.alibaba.druid.sql.PagerUtils;
-import com.alibaba.druid.util.JdbcConstants;
-
-public class PagerUtilsTest_Count_PG_0 extends TestCase {
-
-    public void test_pg_0() throws Exception {
-        String sql = "select * from t";
-        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_pg_1() throws Exception {
-        String sql = "select id, name from t";
-        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_pg_2() throws Exception {
-        String sql = "select id, name from t order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_pg_group_0() throws Exception {
-        String sql = "select type, count(*) from t group by type";
-        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
-        Assert.assertEquals("SELECT COUNT(*)" + //
-                            "\nFROM (SELECT type, COUNT(*)" + //
-                            "\n\tFROM t" + //
-                            "\n\tGROUP BY type" + //
-                            "\n\t)", result);
-    }
-
-    public void test_pg_union_0() throws Exception {
-        String sql = "select id, name from t1 union select id, name from t2 order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM (SELECT id, name" + //
-                            "\n\tFROM t1" + //
-                            "\n\tUNION" + //
-                            "\n\tSELECT id, name" + //
-                            "\n\tFROM t2" + //
-                            "\n\t)", result);
-    }
-}
+package com.alibaba.druid.bvt.sql;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PagerUtils;
+import com.alibaba.druid.util.JdbcConstants;
+
+public class PagerUtilsTest_Count_PG_0 extends TestCase {
+
+    public void test_pg_0() throws Exception {
+        String sql = "select * from t";
+        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_pg_1() throws Exception {
+        String sql = "select id, name from t";
+        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_pg_2() throws Exception {
+        String sql = "select id, name from t order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_pg_group_0() throws Exception {
+        String sql = "select type, count(*) from t group by type";
+        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
+        Assert.assertEquals("SELECT COUNT(*)" + //
+                            "\nFROM (SELECT type, COUNT(*)" + //
+                            "\n\tFROM t" + //
+                            "\n\tGROUP BY type" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+
+    public void test_pg_union_0() throws Exception {
+        String sql = "select id, name from t1 union select id, name from t2 order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.POSTGRESQL);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM (SELECT id, name" + //
+                            "\n\tFROM t1" + //
+                            "\n\tUNION" + //
+                            "\n\tSELECT id, name" + //
+                            "\n\tFROM t2" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+}
@@ -1,54 +1,54 @@
-package com.alibaba.druid.bvt.sql;
-
-import junit.framework.TestCase;
-
-import org.junit.Assert;
-
-import com.alibaba.druid.sql.PagerUtils;
-import com.alibaba.druid.util.JdbcConstants;
-
-public class PagerUtilsTest_Count_SQLServer_0 extends TestCase {
-
-    public void test_sqlserver_0() throws Exception {
-        String sql = "select * from t";
-        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_sqlserver_1() throws Exception {
-        String sql = "select id, name from t";
-        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_sqlserver_2() throws Exception {
-        String sql = "select id, name from t order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM t", result);
-    }
-
-    public void test_sqlserver_group_0() throws Exception {
-        String sql = "select type, count(*) from t group by type";
-        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
-        Assert.assertEquals("SELECT COUNT(*)" + //
-                            "\nFROM (SELECT type, COUNT(*)" + //
-                            "\n\tFROM t" + //
-                            "\n\tGROUP BY type" + //
-                            "\n\t)", result);
-    }
-
-    public void test_sqlserver_union_0() throws Exception {
-        String sql = "select id, name from t1 union select id, name from t2 order by id";
-        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
-        Assert.assertEquals("SELECT COUNT(*)\n" + //
-                            "FROM (SELECT id, name" + //
-                            "\n\tFROM t1" + //
-                            "\n\tUNION" + //
-                            "\n\tSELECT id, name" + //
-                            "\n\tFROM t2" + //
-                            "\n\t)", result);
-    }
-}
+package com.alibaba.druid.bvt.sql;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PagerUtils;
+import com.alibaba.druid.util.JdbcConstants;
+
+public class PagerUtilsTest_Count_SQLServer_0 extends TestCase {
+
+    public void test_sqlserver_0() throws Exception {
+        String sql = "select * from t";
+        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_sqlserver_1() throws Exception {
+        String sql = "select id, name from t";
+        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_sqlserver_2() throws Exception {
+        String sql = "select id, name from t order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM t", result);
+    }
+
+    public void test_sqlserver_group_0() throws Exception {
+        String sql = "select type, count(*) from t group by type";
+        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
+        Assert.assertEquals("SELECT COUNT(*)" + //
+                            "\nFROM (SELECT type, COUNT(*)" + //
+                            "\n\tFROM t" + //
+                            "\n\tGROUP BY type" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+
+    public void test_sqlserver_union_0() throws Exception {
+        String sql = "select id, name from t1 union select id, name from t2 order by id";
+        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
+        Assert.assertEquals("SELECT COUNT(*)\n" + //
+                            "FROM (SELECT id, name" + //
+                            "\n\tFROM t1" + //
+                            "\n\tUNION" + //
+                            "\n\tSELECT id, name" + //
+                            "\n\tFROM t2" + //
+                            "\n\t) ALIAS_COUNT_TABLE", result);
+    }
+}
@@ -1,53 +1,53 @@
-package com.alibaba.druid.bvt.sql;
-
-import junit.framework.TestCase;
-
-import org.junit.Assert;
-
-import com.alibaba.druid.sql.PagerUtils;
-import com.alibaba.druid.util.JdbcConstants;
-
-public class PagerUtilsTest_Limit_db2_0 extends TestCase {
-
-    public void test_mysql_0() throws Exception {
-        String sql = "select * from t";
-        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 0, 10);
-        Assert.assertEquals("SELECT *" + //
-                            "\nFROM t" + //
-                            "\nFETCH FIRST 10 ROWS ONLY", result);
-    }
-
-    public void test_mysql_1() throws Exception {
-        String sql = "select * from t";
-        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 10, 10);
-        Assert.assertEquals("SELECT *"//
-                            + "\nFROM (SELECT *, ROW_NUMBER() AS ROWNUM"//
-                            + "\n\tFROM t"//
-                            + "\n\t) XX"//
-                            + "\nWHERE ROWNUM > 10"//
-                            + "\n\tAND ROWNUM <= 20", result);
-    }
-
-    public void test_mysql_2() throws Exception {
-        String sql = "select * from t where age > 100";
-        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 20, 10);
-        Assert.assertEquals("SELECT *"//
-                            + "\nFROM (SELECT *, ROW_NUMBER() AS ROWNUM"//
-                            + "\n\tFROM t"//
-                            + "\n\tWHERE age > 100"//
-                            + "\n\t) XX"//
-                            + "\nWHERE ROWNUM > 20"//
-                            + "\n\tAND ROWNUM <= 30", result);
-    }
-
-    public void test_mysql_3() throws Exception {
-        String sql = "select id, name, salary from t order by id, name";
-        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 20, 10);
-        Assert.assertEquals("SELECT *"//
-                            + "\nFROM (SELECT id, name, salary, ROW_NUMBER() OVER (ORDER BY id, name) AS ROWNUM"//
-                            + "\n\tFROM t"//
-                            + "\n\t) XX"//
-                            + "\nWHERE ROWNUM > 20"//
-                            + "\n\tAND ROWNUM <= 30", result);
-    }
-}
+package com.alibaba.druid.bvt.sql;
+
+import junit.framework.TestCase;
+
+import org.junit.Assert;
+
+import com.alibaba.druid.sql.PagerUtils;
+import com.alibaba.druid.util.JdbcConstants;
+
+public class PagerUtilsTest_Limit_db2_0 extends TestCase {
+
+    public void test_mysql_0() throws Exception {
+        String sql = "select * from t";
+        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 0, 10);
+        Assert.assertEquals("SELECT *" + //
+                            "\nFROM t" + //
+                            "\nFETCH FIRST 10 ROWS ONLY", result);
+    }
+
+    public void test_mysql_1() throws Exception {
+        String sql = "select * from t";
+        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 10, 10);
+        Assert.assertEquals("SELECT *"//
+                            + "\nFROM (SELECT *, ROW_NUMBER() OVER () AS ROWNUM"//
+                            + "\n\tFROM t"//
+                            + "\n\t) XX"//
+                            + "\nWHERE ROWNUM > 10"//
+                            + "\n\tAND ROWNUM <= 20", result);
+    }
+
+    public void test_mysql_2() throws Exception {
+        String sql = "select * from t where age > 100";
+        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 20, 10);
+        Assert.assertEquals("SELECT *"//
+                            + "\nFROM (SELECT *, ROW_NUMBER() OVER () AS ROWNUM"//
+                            + "\n\tFROM t"//
+                            + "\n\tWHERE age > 100"//
+                            + "\n\t) XX"//
+                            + "\nWHERE ROWNUM > 20"//
+                            + "\n\tAND ROWNUM <= 30", result);
+    }
+
+    public void test_mysql_3() throws Exception {
+        String sql = "select id, name, salary from t order by id, name";
+        String result = PagerUtils.limit(sql, JdbcConstants.DB2, 20, 10);
+        Assert.assertEquals("SELECT *"//
+                            + "\nFROM (SELECT id, name, salary, ROW_NUMBER() OVER (ORDER BY id, name) AS ROWNUM"//
+                            + "\n\tFROM t"//
+                            + "\n\t) XX"//
+                            + "\nWHERE ROWNUM > 20"//
+                            + "\n\tAND ROWNUM <= 30", result);
+    }
+}