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
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