尝试增加查询列

abel533 · abel533 · commit 72e1dc0a08d8 · 2015-07-17T14:59:57.000+08:00
diff --git a/src/main/java/tk/mybatis/mapper/entity/Example.java b/src/main/java/tk/mybatis/mapper/entity/Example.java
@@ -28,10 +28,7 @@
 import org.apache.ibatis.reflection.SystemMetaObject;
 import tk.mybatis.mapper.mapperhelper.EntityHelper;
 
-import java.util.HashMap;
-import java.util.LinkedList;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * 通用的Example查询对象
@@ -45,6 +42,8 @@ public class Example {
 
     protected boolean exists;
 
+    protected Set<String> selectColumns;
+
     protected List<Criteria> oredCriteria;
 
     protected Class<?> entityClass;
@@ -91,6 +90,24 @@ public void setOrderByClause(String orderByClause) {
         this.orderByClause = orderByClause;
     }
 
+    public Set<String> getSelectColumns() {
+        return selectColumns;
+    }
+
+    public Example selectProperties(String... properties) {
+        if (properties != null && properties.length > 0) {
+            if (this.selectColumns == null) {
+                this.selectColumns = new LinkedHashSet<String>();
+            }
+            for (String property : properties) {
+                if (propertyMap.containsKey(property)) {
+                    this.selectColumns.add(property);
+                }
+            }
+        }
+        return this;
+    }
+
     public boolean isDistinct() {
         return distinct;
     }
diff --git a/src/main/java/tk/mybatis/mapper/provider/MapperProvider.java b/src/main/java/tk/mybatis/mapper/provider/MapperProvider.java
@@ -442,6 +442,7 @@ public SqlNode selectByExample(MappedStatement ms) {
         sqlNodes.add(new StaticTextSqlNode("SELECT"));
         IfSqlNode distinctSqlNode = new IfSqlNode(new StaticTextSqlNode("DISTINCT"), "distinct");
         sqlNodes.add(distinctSqlNode);
+        //TODO 增加Example中指定查询列的功能
         sqlNodes.add(new StaticTextSqlNode(EntityHelper.getSelectColumns(entityClass) + " FROM " + tableName(entityClass)));
         IfSqlNode ifNullSqlNode = new IfSqlNode(exampleWhereClause(ms.getConfiguration()), "_parameter != null");
         sqlNodes.add(ifNullSqlNode);
