EntityMapper增加selectOne方法

Author: abel533

src/main/java/com/github/abel533/entity/CommonProvider.java

@@ -37,6 +37,37 @@
  */
 public class CommonProvider extends BaseProvider {
 
+    /**
+     * 查询，入参可以是Entity.class或new Entity()
+     *
+     * @param params
+     * @return
+     */
+    public String selectOne(final Map<String, Object> params) {
+        return new SQL() {{
+            Object entity = getEntity(params);
+            Class<?> entityClass = getEntityClass(params);
+            EntityHelper.EntityTable entityTable = EntityHelper.getEntityTable(entityClass);
+            SELECT(EntityHelper.getAllColumns(entityClass));
+            FROM(entityTable.getName());
+            if (entity != null) {
+                final MetaObject metaObject = MapperTemplate.forObject(entity);
+                for (EntityHelper.EntityColumn column : entityTable.getEntityClassColumns()) {
+                    Object value = metaObject.getValue(column.getProperty());
+                    if (value == null) {
+                        continue;
+                    } else if (column.getJavaType().equals(String.class)) {
+                        if (isNotEmpty((String) value)) {
+                            WHERE(column.getColumn() + "=#{record." + column.getProperty() + "}");
+                        }
+                    } else {
+                        WHERE(column.getColumn() + "=#{record." + column.getProperty() + "}");
+                    }
+                }
+            }
+        }}.toString();
+    }
+
     /**
      * 查询，入参可以是Entity.class或new Entity()
      *

src/main/java/com/github/abel533/entity/EntityMapper.java

@@ -45,6 +45,22 @@ public EntityMapper(CommonMapper commonMapper) {
         this.commonMapper = commonMapper;
     }
 
+    /**
+     * 根据参数进行查询，查询结果最多只能有一个
+     * <br>查询条件为属性String类型不为空，其他类型!=null时
+     * <br>where property = ? and property2 = ? 条件
+     *
+     * @param record
+     * @param <T>
+     * @return
+     */
+    public <T> T selectOne(T record) {
+        if (record == null) {
+            throw new NullPointerException("实体或者主键参数不能为空!");
+        }
+        return (T) EntityHelper.map2Bean(commonMapper.selectOne(record), record.getClass());
+    }
+
     /**
      * 根据参数进行查询,record可以是Class<?>类型
      * <br>查询条件为属性String类型不为空，其他类型!=null时

src/main/java/com/github/abel533/entity/mapper/CommonMapper.java

@@ -36,6 +36,18 @@
  * @author liuzh
  */
 public interface CommonMapper {
+    /**
+     * 根据参数进行查询，查询结果最多只能有一个
+     * <br>查询条件为属性String类型不为空，其他类型!=null时
+     * <br>where property = ? and property2 = ? 条件
+     *
+     * @param record
+     * @param <T>
+     * @return
+     */
+    @SelectProvider(type = CommonProvider.class, method = "selectOne")
+    <T> Map<String, Object> selectOne(@Param("record") T record);
+
     /**
      * 根据参数进行查询,record可以是Class<?>类型
      * <br>查询条件为属性String类型不为空，其他类型!=null时
@@ -45,9 +57,8 @@ public interface CommonMapper {
      * @param <T>
      * @return
      */
-    @Options(flushCache = true)
     @SelectProvider(type = CommonProvider.class, method = "select")
-    <T> List<Map<String,Object>> select(@Param("record") T record);
+    <T> List<Map<String, Object>> select(@Param("record") T record);
 
     /**
      * 根据参数进行查询总数,record可以是Class<?>类型
@@ -58,7 +69,6 @@ public interface CommonMapper {
      * @param <T>
      * @return
      */
-    @Options(flushCache = true)
     @SelectProvider(type = CommonProvider.class, method = "count")
     <T> int count(@Param("record") T record);
 
@@ -70,9 +80,8 @@ public interface CommonMapper {
      * @param <T>
      * @return
      */
-    @Options(flushCache = true)
     @SelectProvider(type = CommonProvider.class, method = "selectByPrimaryKey")
-    <T> Map<String,Object> selectByPrimaryKey(@Param("entityClass") Class<T> entityClass, @Param("key") Object key);
+    <T> Map<String, Object> selectByPrimaryKey(@Param("entityClass") Class<T> entityClass, @Param("key") Object key);
 
     /**
      * 插入数据库，主键字段没有值的时候不会出现在sql中
@@ -146,7 +155,6 @@ public interface CommonMapper {
      * @param <T>
      * @return
      */
-    @Options(flushCache = true)
     @SelectProvider(type = CommonProvider.class, method = "countByExample")
     <T> int countByExample(@Param("entityClass") Class<T> entityClass, @Param("example") Object example);
 
@@ -169,9 +177,8 @@ public interface CommonMapper {
      * @param <T>
      * @return
      */
-    @Options(flushCache = true)
     @SelectProvider(type = CommonProvider.class, method = "selectByExample")
-    <T> List<Map<String,Object>> selectByExample(@Param("entityClass") Class<T> entityClass, @Param("example") Object example);
+    <T> List<Map<String, Object>> selectByExample(@Param("entityClass") Class<T> entityClass, @Param("example") Object example);
 
     /**
      * 通过Example进行更新非空字段

src/test/java/com/github/abel533/entity/test/TestSelectOne.java

@@ -0,0 +1,84 @@
+/*
+ * The MIT License (MIT)
+ *
+ * Copyright (c) 2014 [email protected]
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+
+package com.github.abel533.entity.test;
+
+import com.github.abel533.entity.EntityMapper;
+import com.github.abel533.entity.mapper.CommonMapper;
+import com.github.abel533.entity.model.Country;
+import com.github.abel533.mapper.MybatisHelper;
+import org.apache.ibatis.exceptions.TooManyResultsException;
+import org.apache.ibatis.session.SqlSession;
+import org.junit.Assert;
+import org.junit.Test;
+
+/**
+ * 测试查询
+ *
+ * @author liuzh
+ */
+public class TestSelectOne {
+
+    @Test(expected = TooManyResultsException.class)
+    public void testSelectAllByNew() {
+        SqlSession sqlSession = MybatisHelper.getSqlSession();
+        try {
+            CommonMapper commonMapper = sqlSession.getMapper(CommonMapper.class);
+            EntityMapper entityMapper = new EntityMapper(commonMapper);
+            entityMapper.selectOne(new Country());
+        } finally {
+            sqlSession.close();
+        }
+    }
+
+    @Test
+    public void testSelectOne() {
+        SqlSession sqlSession = MybatisHelper.getSqlSession();
+        try {
+            CommonMapper commonMapper = sqlSession.getMapper(CommonMapper.class);
+            EntityMapper entityMapper = new EntityMapper(commonMapper);
+
+            Country country = new Country();
+            country.setCountrycode("CN");
+            Country result = entityMapper.selectOne(country);
+
+            Assert.assertEquals("China", result.getCountryname());
+        } finally {
+            sqlSession.close();
+        }
+    }
+
+    @Test(expected = Exception.class)
+    public void testSelectAllByNull() {
+        SqlSession sqlSession = MybatisHelper.getSqlSession();
+        try {
+            CommonMapper commonMapper = sqlSession.getMapper(CommonMapper.class);
+            EntityMapper entityMapper = new EntityMapper(commonMapper);
+
+            entityMapper.selectOne(null);
+        } finally {
+            sqlSession.close();
+        }
+    }
+}