diff options
Diffstat (limited to 'src/main/java')
-rw-r--r-- | src/main/java/com/iciql/Query.java | 51 | ||||
-rw-r--r-- | src/main/java/com/iciql/QueryWhere.java | 8 | ||||
-rw-r--r-- | src/main/java/com/iciql/ValueCount.java | 71 |
3 files changed, 130 insertions, 0 deletions
diff --git a/src/main/java/com/iciql/Query.java b/src/main/java/com/iciql/Query.java index d26e7b7..20d5520 100644 --- a/src/main/java/com/iciql/Query.java +++ b/src/main/java/com/iciql/Query.java @@ -31,6 +31,7 @@ import java.sql.ResultSet; import java.sql.SQLException; import java.util.ArrayList; import java.util.Arrays; +import java.util.Collections; import java.util.HashMap; import java.util.IdentityHashMap; import java.util.List; @@ -100,6 +101,56 @@ public class Query<T> { } } + public <X> List<ValueCount<X>> selectCount(X x) { + return selectCount(x, false); + } + + public <X> List<ValueCount<X>> selectCountDesc(X x) { + return selectCount(x, true); + } + + <X> List<ValueCount<X>> selectCount(X x, boolean desc) { + List<ValueCount<X>> list = Utils.newArrayList(); + SelectColumn<T> col = getColumnByReference(x); + X alias = x; + if (col == null) { + alias = getPrimitiveAliasByValue(x); + col = getColumnByReference(alias); + } + if (col == null) { + throw new IciqlException("Unmapped column reference!"); + } + groupBy(alias); + + SQLStatement stat = getSelectStatement(false); + col.appendSQL(stat); + stat.appendSQL(", COUNT(*)"); + appendFromWhere(stat); + + ResultSet rs = stat.executeQuery(); + Class<? extends DataTypeAdapter<?>> typeAdapter = col.getFieldDefinition().typeAdapter; + Class<?> clazz = x.getClass(); + try { + // SQLite returns pre-closed ResultSets for query results with 0 rows + if (!rs.isClosed()) { + while (rs.next()) { + X value = (X) db.getDialect().deserialize(rs, 1, clazz, typeAdapter); + long count = rs.getLong(2); + list.add(new ValueCount<X>(value, count)); + } + } + Collections.sort(list); + if (desc) { + Collections.reverse(list); + } + } catch (Exception e) { + throw IciqlException.fromSQL(stat.getSQL(), e); + } finally { + JdbcUtils.closeSilently(rs, true); + } + return list; + } + public List<T> select() { return select(false); } diff --git a/src/main/java/com/iciql/QueryWhere.java b/src/main/java/com/iciql/QueryWhere.java index 16f1397..de6ecd4 100644 --- a/src/main/java/com/iciql/QueryWhere.java +++ b/src/main/java/com/iciql/QueryWhere.java @@ -605,4 +605,12 @@ public class QueryWhere<T> { return query.selectCount();
}
+ public <X> List<ValueCount<X>> selectCount(X x) {
+ return query.selectCount(x, false);
+ }
+
+ public <X> List<ValueCount<X>> selectCountDesc(X x) {
+ return query.selectCount(x, true);
+ }
+
}
diff --git a/src/main/java/com/iciql/ValueCount.java b/src/main/java/com/iciql/ValueCount.java new file mode 100644 index 0000000..dfc8570 --- /dev/null +++ b/src/main/java/com/iciql/ValueCount.java @@ -0,0 +1,71 @@ +/* + * Copyright 2016 James Moger. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.iciql; + +import java.io.Serializable; +import java.util.Date; + +public class ValueCount<X> implements Serializable, Comparable<ValueCount<X>> { + + private final static long serialVersionUID = 1L; + + private final X value; + private final long count; + + ValueCount(X x, long count) { + this.value = x; + this.count = count; + } + + public X getValue() { + return value; + } + + public long getCount() { + return count; + } + + @Override + public String toString() { + return String.format("%s=%d", value, count); + } + + @Override + public int compareTo(ValueCount<X> o) { + return (count < o.count) ? -1 : ((count == o.count) ? compareValue(o.value) : 1); + } + + private int compareValue(X anotherValue) { + if (value == null && anotherValue == null) { + return 0; + } else if (value == null) { + return 1; + } else if (anotherValue == null) { + return -1; + } + + if (Number.class.isAssignableFrom(value.getClass())) { + long n1 = ((Number) value).longValue(); + long n2 = ((Number) anotherValue).longValue(); + return (n1 < n2) ? -1 : ((n1 == n2) ? 0 : 1); + } else if (Date.class.isAssignableFrom(value.getClass())) { + return ((Date) value).compareTo((Date) anotherValue); + } + + return value.toString().compareToIgnoreCase(anotherValue.toString()); + } +} + |