Revision 45155 trunk/org.gvsig.desktop/org.gvsig.desktop.compat.cdc/org.gvsig.fmap.dal/org.gvsig.fmap.dal.db/org.gvsig.fmap.dal.db.jdbc/src/main/java/org/gvsig/fmap/dal/store/jdbc2/spi/operations/CountOperation.java

View differences:

CountOperation.java
29 29
import java.sql.Statement;
30 30
import java.util.List;
31 31
import org.apache.commons.lang3.StringUtils;
32
import org.gvsig.expressionevaluator.Code;
32 33
import org.gvsig.expressionevaluator.ExpressionBuilder;
34
import static org.gvsig.expressionevaluator.ExpressionBuilder.FUNCTION_LET;
35
import org.gvsig.expressionevaluator.ExpressionUtils;
33 36
import org.gvsig.fmap.dal.SQLBuilder.SelectBuilder;
34 37
import org.gvsig.fmap.dal.exception.DataException;
35 38
import org.gvsig.fmap.dal.feature.FeatureQuery;
......
80 83
        ExpressionBuilder expbuilder = sqlbuilder.expression();
81 84

  
82 85
        SelectBuilder select = sqlbuilder.select();
83
        
84
        if( this.query!=null && this.query.hasGroupByColumns() ) {
85
          JDBCSQLBuilderBase subsqlbuilder = this.createSQLBuilder();
86
          SelectBuilder subselect = subsqlbuilder.select();
87
          subselect.column().value(subsqlbuilder.count().all());
88
          subselect.from().table()
89
                  .database(this.table.getDatabase())
90
                  .schema(this.table.getSchema())
91
                  .name(this.table.getTable());
92
          subselect.from().subquery(this.table.getSubquery());
93
          Evaluator filter = query==null? null:query.getFilter();
94
          if( filter != null ) {
95
              String sqlfilter = filter.getSQL();
96
              if( ! StringUtils.isEmpty(sqlfilter) ) {
97
                  if( this.helper.supportFilter(this.featureType, filter) ) {
98
                      subselect.where().set(expbuilder.toValue(sqlfilter));
99
                  }
100
              }
101
          }
102
          if( ! StringUtils.isEmpty(baseFilter) ) {
103
              subselect.where().and(expbuilder.toValue(baseFilter));
104
          }
105 86

  
106
          FeatureQueryOrder order = query==null? null:query.getOrder();
107
          if( order != null ) {
108
              for( FeatureQueryOrder.FeatureQueryOrderMember member : order.members() ) {
109
                  if( member.hasEvaluator() ) {
110
                      String sqlorder = member.getEvaluator().getSQL();
111
                      if( ! StringUtils.isEmpty(sqlorder) ) {
112
                          subselect.order_by().custom(sqlorder);
113
                      }
114
                  } else {
115
                      subselect.order_by()
116
                              .column(member.getAttributeName())
117
                              .ascending(member.getAscending());
118
                  }
119
              }
120
          }
121
          
122
          List<String> groupbyColumns = query==null? null:query.getGroupByColumns();
123
          if( groupbyColumns!=null && !groupbyColumns.isEmpty() ) {
124
            for(String columnName : groupbyColumns ) {
125
                subselect.group_by(expbuilder.column(columnName));
87
        if (this.query != null && this.query.hasGroupByColumns()) {
88
            JDBCSQLBuilderBase subsqlbuilder = this.createSQLBuilder();
89
            SelectBuilder subselect = subsqlbuilder.select();
90
            subselect.column().value(subsqlbuilder.count().all());
91
            subselect.from().table()
92
                    .database(this.table.getDatabase())
93
                    .schema(this.table.getSchema())
94
                    .name(this.table.getTable());
95
            subselect.from().subquery(this.table.getSubquery());
96
            Evaluator filter = query == null ? null : query.getFilter();
97
            if (filter != null) {
98
                String sqlfilter = filter.getSQL();
99
                if (!StringUtils.isEmpty(sqlfilter)) {
100
                    if (this.helper.supportFilter(this.featureType, filter)) {
101
                        subselect.where().set(expbuilder.toValue(sqlfilter));
102
                    }
103
                }
126 104
            }
127
          }
128
          subsqlbuilder.setProperties(
129
                  ExpressionBuilder.Variable.class, 
130
                  PROP_TABLE, table
131
          );
132
          String subsql = subselect.toString();
133
          select.from().table()
134
                  .database(this.table.getDatabase())
135
                  .schema(this.table.getSchema())
136
                  .name(this.table.getTable());
137
          select.from().subquery(subsql);
138
          
139
          
105
            if (!StringUtils.isEmpty(baseFilter)) {
106
                subselect.where().and(expbuilder.toValue(baseFilter));
107
            }
108

  
109
            FeatureQueryOrder order = query == null ? null : query.getOrder();
110
            if (order != null) {
111
                for (FeatureQueryOrder.FeatureQueryOrderMember member : order.members()) {
112
                    if (member.hasEvaluator()) {
113
                        String sqlorder = member.getEvaluator().getSQL();
114
                        if (!StringUtils.isEmpty(sqlorder)) {
115
                            subselect.order_by().custom(sqlorder);
116
                        }
117
                    } else {
118
                        subselect.order_by()
119
                                .column(member.getAttributeName())
120
                                .ascending(member.getAscending());
121
                    }
122
                }
123
            }
124

  
125
            List<String> groupbyColumns = query == null ? null : query.getGroupByColumns();
126
            if (groupbyColumns != null && !groupbyColumns.isEmpty()) {
127
                for (String columnName : groupbyColumns) {
128
                    if (this.featureType.getAttributeDescriptor(columnName) != null) {
129
                        subselect.group_by(expbuilder.column(columnName));
130
                    } else {
131
                        try {
132
                            try {
133
                                Code groupByColumnCode = ExpressionUtils.compile(columnName);
134
                                if (groupByColumnCode.code() == Code.CALLABLE) {
135
                                    Code.Callable callable = (Code.Callable) groupByColumnCode;
136
                                    if (callable.name().equalsIgnoreCase(FUNCTION_LET)) {
137
                                        Code exp = callable.parameters().get(1);
138
                                        Code name = callable.parameters().get(0);
139
                                        subselect.column().value(exp.toValue())
140
                                                .as((String) ((Code.Constant) name).value());
141
                                        // nombre que se pone en la parte del groupby debe de ser el nombre de la var del set
142
                                        groupByColumnCode = exp; 
143
                                    }
144
                                }
145
                                subselect.group_by(groupByColumnCode.toValue());
146
                            } catch (Exception ex) {
147
                                throw new RuntimeException("Not able to create column by expression in groupby query", ex);
148
                            }
149
                        } catch (Exception ex) {
150
                            throw new RuntimeException("Not able to create column by expression in groupby query", ex);
151
                        }
152
                    }
153
                }
154
            }
155
            this.helper.processSpecialFunctions(subsqlbuilder, featureType, null);
156
            subsqlbuilder.setProperties(
157
                    ExpressionBuilder.Variable.class,
158
                    PROP_TABLE, table
159
            );
160
            String subsql = subselect.toString();
161
            select.from().table()
162
                    .database(this.table.getDatabase())
163
                    .schema(this.table.getSchema())
164
                    .name(this.table.getTable());
165
            select.from().subquery(subsql);
166

  
140 167
        } else {
141
          select.column().value(sqlbuilder.count().all());
142
          select.from().table()
143
                  .database(this.table.getDatabase())
144
                  .schema(this.table.getSchema())
145
                  .name(this.table.getTable());
146
          select.from().subquery(this.table.getSubquery());
147
          if (!StringUtils.isEmpty(baseFilter)) {
148
              sqlbuilder.select().where().set( expbuilder.custom(baseFilter) );
149
          }
150
          if( this.query!=null ) {
151
            if( this.query.getFilter()!=null && !StringUtils.isBlank(this.query.getFilter().getSQL()) ) {
152
              // El and() hace un set() si no hay un filtro previo
153
              select.where().and(expbuilder.toValue(this.query.getFilter().getSQL()));
168
            select.column().value(sqlbuilder.count().all());
169
            select.from().table()
170
                    .database(this.table.getDatabase())
171
                    .schema(this.table.getSchema())
172
                    .name(this.table.getTable());
173
            select.from().subquery(this.table.getSubquery());
174
            if (!StringUtils.isEmpty(baseFilter)) {
175
                sqlbuilder.select().where().set(expbuilder.custom(baseFilter));
154 176
            }
155
          }
177
            if (this.query != null) {
178
                if (this.query.getFilter() != null && !StringUtils.isBlank(this.query.getFilter().getSQL())) {
179
                    // El and() hace un set() si no hay un filtro previo
180
                    select.where().and(expbuilder.toValue(this.query.getFilter().getSQL()));
181
                }
182
            }
156 183
        }
157 184
        this.helper.processSpecialFunctions(sqlbuilder, featureType, null);
158
        
185

  
159 186
        select.remove_all_columns();
160 187
        select.column().value(sqlbuilder.count().all());
161
        
188

  
162 189
        sqlbuilder.setProperties(
163
                ExpressionBuilder.Variable.class, 
190
                ExpressionBuilder.Variable.class,
164 191
                PROP_TABLE, table
165 192
        );
166 193

  
167 194
        String sql = select.toString();
168 195
        return sql;
169 196
    }
170
    
197

  
171 198
    public long count(Connection conn) throws DataException {
172 199

  
173 200
        String sql = this.getSQL();
......
188 215
            JDBCUtils.closeQuietly(rs);
189 216
        }
190 217
    }
191

  
218
   
192 219
}

Also available in: Unified diff