增加一个通用的selectAll方法

abel533 · abel533 · commit 948339dd4bb6 · 2015-08-29T00:07:04.000+08:00
diff --git a/src/main/java/tk/mybatis/mapper/common/base/BaseSelectMapper.java b/src/main/java/tk/mybatis/mapper/common/base/BaseSelectMapper.java
@@ -24,10 +24,7 @@
 
 package tk.mybatis.mapper.common.base;
 
-import tk.mybatis.mapper.common.base.select.SelectByPrimaryKeyMapper;
-import tk.mybatis.mapper.common.base.select.SelectCountMapper;
-import tk.mybatis.mapper.common.base.select.SelectMapper;
-import tk.mybatis.mapper.common.base.select.SelectOneMapper;
+import tk.mybatis.mapper.common.base.select.*;
 
 /**
  * 通用Mapper接口,基础查询
@@ -38,6 +35,7 @@
 public interface BaseSelectMapper<T> extends
         SelectOneMapper<T>,
         SelectMapper<T>,
+        SelectAllMapper<T>,
         SelectCountMapper<T>,
         SelectByPrimaryKeyMapper<T> {
 
diff --git a/src/main/java/tk/mybatis/mapper/common/base/select/SelectAllMapper.java b/src/main/java/tk/mybatis/mapper/common/base/select/SelectAllMapper.java
@@ -0,0 +1,20 @@
+package tk.mybatis.mapper.common.base.select;
+
+import org.apache.ibatis.annotations.SelectProvider;
+import tk.mybatis.mapper.provider.base.BaseSelectProvider;
+
+import java.util.List;
+
+/**
+ * @author liuzh
+ */
+public interface SelectAllMapper<T> {
+
+    /**
+     * 查询全部结果
+     *
+     * @return
+     */
+    @SelectProvider(type = BaseSelectProvider.class, method = "dynamicSQL")
+    List<T> selectAll();
+}
diff --git a/src/main/java/tk/mybatis/mapper/provider/base/BaseSelectProvider.java b/src/main/java/tk/mybatis/mapper/provider/base/BaseSelectProvider.java
@@ -147,4 +147,21 @@ public SqlNode selectCount(MappedStatement ms) {
         sqlNodes.add(new WhereSqlNode(ms.getConfiguration(), getAllIfColumnNode(entityClass)));
         return new MixedSqlNode(sqlNodes);
     }
+
+    /**
+     * 查询全部结果
+     *
+     * @param ms
+     * @return
+     */
+    public String selectAll(MappedStatement ms) {
+        final Class<?> entityClass = getSelectReturnType(ms);
+        //修改返回值类型为实体类型
+        setResultType(ms, entityClass);
+        //开始拼sql
+        StringBuilder sql = new StringBuilder();
+        sql.append("select ").append(EntityHelper.getSelectColumns(entityClass)).append(" from ");
+        sql.append(tableName(entityClass));
+        return sql.toString();
+    }
 }
diff --git a/src/test/java/tk/mybatis/mapper/test/country/TestSelectAll.java b/src/test/java/tk/mybatis/mapper/test/country/TestSelectAll.java
@@ -0,0 +1,35 @@
+package tk.mybatis.mapper.test.country;
+
+import org.apache.ibatis.session.SqlSession;
+import org.junit.Assert;
+import org.junit.Test;
+import tk.mybatis.mapper.mapper.CountryMapper;
+import tk.mybatis.mapper.mapper.MybatisHelper;
+import tk.mybatis.mapper.model.Country;
+
+import java.util.List;
+
+/**
+ * 通过实体类属性进行查询
+ *
+ * @author liuzh
+ */
+public class TestSelectAll {
+
+    /**
+     * 查询全部
+     */
+    @Test
+    public void testDynamicSelectPage() {
+        SqlSession sqlSession = MybatisHelper.getSqlSession();
+        try {
+            CountryMapper mapper = sqlSession.getMapper(CountryMapper.class);
+
+            List<Country> countryList = mapper.selectAll();
+            //查询总数
+            Assert.assertEquals(183, countryList.size());
+        } finally {
+            sqlSession.close();
+        }
+    }
+}
