Merge pull request mybatis#104 from jeffgbutler/master

jeffgbutler · web-flow · commit 319396c2d1bd · 2019-07-02T06:35:53.000-04:00
Refactor QueryExpressionDSL
diff --git a/src/main/java/org/mybatis/dynamic/sql/select/QueryExpressionDSL.java b/src/main/java/org/mybatis/dynamic/sql/select/QueryExpressionDSL.java
@@ -51,12 +51,31 @@ public class QueryExpressionDSL<R> implements Buildable<R> {
     private List<JoinSpecification> joinSpecifications = new ArrayList<>();
     
     private QueryExpressionDSL(FromGatherer<R> fromGatherer) {
-        connector = fromGatherer.builder.connector;
-        selectList = Arrays.asList(fromGatherer.builder.selectList);
-        isDistinct = fromGatherer.builder.isDistinct;
-        selectDSL = Objects.requireNonNull(fromGatherer.builder.selectDSL);
+        connector = fromGatherer.connector;
+        selectList = Arrays.asList(fromGatherer.selectList);
+        isDistinct = fromGatherer.isDistinct;
+        selectDSL = Objects.requireNonNull(fromGatherer.selectDSL);
         table = Objects.requireNonNull(fromGatherer.table);
-        tableAliases.putAll(fromGatherer.tableAliasMap);
+    }
+    
+    private QueryExpressionDSL(FromGatherer<R> fromGatherer, String tableAlias) {
+        this(fromGatherer);
+        tableAliases.put(table, tableAlias);
+    }
+    
+    public static <R> FromGatherer<R> select(SelectDSL<R> selectDSL, BasicColumn...selectList) {
+        return new FromGatherer.Builder<R>()
+                .withSelectList(selectList)
+                .withSelectDSL(selectDSL)
+                .build();
+    }
+    
+    public static <R> FromGatherer<R> selectDistinct(SelectDSL<R> selectDSL, BasicColumn...selectList) {
+        return new FromGatherer.Builder<R>()
+                .withSelectList(selectList)
+                .withSelectDSL(selectDSL)
+                .isDistinct()
+                .build();
     }
     
     public <T> QueryExpressionWhereBuilder where(BindableColumn<T> column, VisitableCondition<T> condition) {
@@ -160,55 +179,58 @@ public SelectDSL<R>.FetchFirstFinisher fetchFirst(long fetchFirstRows) {
     }
     
     public static class FromGatherer<R> {
-        private FromGathererBuilder<R> builder;
-        private Map<SqlTable, String> tableAliasMap = new HashMap<>();
+        private String connector;
+        private BasicColumn[] selectList;
+        private SelectDSL<R> selectDSL;
+        private boolean isDistinct;
         private SqlTable table;
         
-        public FromGatherer(FromGathererBuilder<R> builder) {
-            this.builder = builder;
+        public FromGatherer(Builder<R> builder) {
+            this.connector = builder.connector;
+            this.selectList = Objects.requireNonNull(builder.selectList);
+            this.selectDSL = Objects.requireNonNull(builder.selectDSL);
+            this.isDistinct = builder.isDistinct;
         }
         
         public QueryExpressionDSL<R> from(SqlTable table) {
             this.table = table;
-            
             return new QueryExpressionDSL<>(this);
         }
 
         public QueryExpressionDSL<R> from(SqlTable table, String tableAlias) {
             this.table = table;
-            tableAliasMap.put(table, tableAlias);
-            return new QueryExpressionDSL<>(this);
-        }
-    }
-    
-    public static class FromGathererBuilder<R> {
-        private String connector;
-        private BasicColumn[] selectList;
-        private SelectDSL<R> selectDSL;
-        private boolean isDistinct;
-        
-        public FromGathererBuilder<R> withConnector(String connector) {
-            this.connector = connector;
-            return this;
-        }
-
-        public FromGathererBuilder<R> withSelectList(BasicColumn[] selectList) {
-            this.selectList = selectList;
-            return this;
-        }
-
-        public FromGathererBuilder<R> withSelectDSL(SelectDSL<R> selectDSL) {
-            this.selectDSL = selectDSL;
-            return this;
+            return new QueryExpressionDSL<>(this, tableAlias);
         }
         
-        public FromGathererBuilder<R> isDistinct() {
-            this.isDistinct = true;
-            return this;
-        }
-        
-        public FromGatherer<R> build() {
-            return new FromGatherer<>(this);
+        public static class Builder<R> {
+            private String connector;
+            private BasicColumn[] selectList;
+            private SelectDSL<R> selectDSL;
+            private boolean isDistinct;
+            
+            public Builder<R> withConnector(String connector) {
+                this.connector = connector;
+                return this;
+            }
+
+            public Builder<R> withSelectList(BasicColumn[] selectList) {
+                this.selectList = selectList;
+                return this;
+            }
+
+            public Builder<R> withSelectDSL(SelectDSL<R> selectDSL) {
+                this.selectDSL = selectDSL;
+                return this;
+            }
+            
+            public Builder<R> isDistinct() {
+                this.isDistinct = true;
+                return this;
+            }
+            
+            public FromGatherer<R> build() {
+                return new FromGatherer<>(this);
+            }
         }
     }
     
@@ -469,19 +491,19 @@ public UnionBuilder(String connector) {
         }
         
         public FromGatherer<R> select(BasicColumn...selectList) {
-            return new FromGathererBuilder<R>()
+            return new FromGatherer.Builder<R>()
                     .withConnector(connector)
                     .withSelectList(selectList)
                     .withSelectDSL(selectDSL)
                     .build();
         }
 
         public FromGatherer<R> selectDistinct(BasicColumn...selectList) {
-            return new FromGathererBuilder<R>()
+            return new FromGatherer.Builder<R>()
                     .withConnector(connector)
-                    .isDistinct()
                     .withSelectList(selectList)
                     .withSelectDSL(selectDSL)
+                    .isDistinct()
                     .build();
         }
     }
diff --git a/src/main/java/org/mybatis/dynamic/sql/select/SelectDSL.java b/src/main/java/org/mybatis/dynamic/sql/select/SelectDSL.java
@@ -21,8 +21,6 @@
 import java.util.function.Function;
 
 import org.mybatis.dynamic.sql.BasicColumn;
-import org.mybatis.dynamic.sql.select.QueryExpressionDSL.FromGatherer;
-import org.mybatis.dynamic.sql.select.QueryExpressionDSL.FromGathererBuilder;
 import org.mybatis.dynamic.sql.select.render.SelectStatementProvider;
 import org.mybatis.dynamic.sql.util.Buildable;
 
@@ -44,47 +42,32 @@ private SelectDSL(Function<SelectModel, R> adapterFunction) {
         this.adapterFunction = Objects.requireNonNull(adapterFunction);
     }
 
-    private FromGatherer<R> queryExpressionBuilder(BasicColumn...selectList) {
-        return new FromGathererBuilder<R>()
-                .withSelectDSL(this)
-                .withSelectList(selectList)
-                .build();
-    }
-    
-    private FromGatherer<R> distinctQueryExpressionBuilder(BasicColumn...selectList) {
-        return new FromGathererBuilder<R>()
-                .withSelectDSL(this)
-                .withSelectList(selectList)
-                .isDistinct()
-                .build();
-    }
-    
-    public static FromGatherer<SelectModel> select(BasicColumn...selectList) {
+    public static QueryExpressionDSL.FromGatherer<SelectModel> select(BasicColumn...selectList) {
         return select(Function.identity(), selectList);
     }
     
-    public static <R> FromGatherer<R> select(Function<SelectModel, R> adapterFunction,
+    public static <R> QueryExpressionDSL.FromGatherer<R> select(Function<SelectModel, R> adapterFunction,
             BasicColumn...selectList) {
-        SelectDSL<R> selectModelBuilder = new SelectDSL<>(adapterFunction);
-        return selectModelBuilder.queryExpressionBuilder(selectList);
+        SelectDSL<R> selectDSL = new SelectDSL<>(adapterFunction);
+        return QueryExpressionDSL.select(selectDSL, selectList);
     }
     
-    public static FromGatherer<SelectModel> selectDistinct(BasicColumn...selectList) {
+    public static QueryExpressionDSL.FromGatherer<SelectModel> selectDistinct(BasicColumn...selectList) {
         return selectDistinct(Function.identity(), selectList);
     }
     
-    public static <R> FromGatherer<R> selectDistinct(Function<SelectModel, R> adapterFunction,
+    public static <R> QueryExpressionDSL.FromGatherer<R> selectDistinct(Function<SelectModel, R> adapterFunction,
             BasicColumn...selectList) {
-        SelectDSL<R> selectModelBuilder = new SelectDSL<>(adapterFunction);
-        return selectModelBuilder.distinctQueryExpressionBuilder(selectList);
+        SelectDSL<R> selectDSL = new SelectDSL<>(adapterFunction);
+        return QueryExpressionDSL.selectDistinct(selectDSL, selectList);
     }
     
-    public static <T> FromGatherer<MyBatis3SelectModelAdapter<T>> selectWithMapper(
+    public static <T> QueryExpressionDSL.FromGatherer<MyBatis3SelectModelAdapter<T>> selectWithMapper(
             Function<SelectStatementProvider, T> mapperMethod, BasicColumn...selectList) {
         return select(selectModel -> MyBatis3SelectModelAdapter.of(selectModel, mapperMethod), selectList);
     }
     
-    public static <T> FromGatherer<MyBatis3SelectModelAdapter<T>> selectDistinctWithMapper(
+    public static <T> QueryExpressionDSL.FromGatherer<MyBatis3SelectModelAdapter<T>> selectDistinctWithMapper(
             Function<SelectStatementProvider, T> mapperMethod, BasicColumn...selectList) {
         return selectDistinct(selectModel -> MyBatis3SelectModelAdapter.of(selectModel, mapperMethod),
                 selectList);
