Examples with Aggregate org.h2.api.Aggregate used on opensource projects

Example 6 with Aggregate

use of org.h2.api.Aggregate in project h2database by h2database.

the class DropAggregate method update.

@Override
public int update() {
    session.getUser().checkAdmin();
    session.commit(true);
    Database db = session.getDatabase();
    UserAggregate aggregate = db.findAggregate(name);
    if (aggregate == null) {
        if (!ifExists) {
            throw DbException.get(ErrorCode.AGGREGATE_NOT_FOUND_1, name);
        }
    } else {
        db.removeDatabaseObject(session, aggregate);
    }
    return 0;
}

Example 7 with Aggregate

use of org.h2.api.Aggregate in project h2database by h2database.

the class AggregateDataMedian method getMedianColumnIndex.

/**
 * Get the index (if any) for the column specified in the median aggregate.
 *
 * @param on the expression (usually a column expression)
 * @return the index, or null
 */
static Index getMedianColumnIndex(Expression on) {
    if (on instanceof ExpressionColumn) {
        ExpressionColumn col = (ExpressionColumn) on;
        Column column = col.getColumn();
        TableFilter filter = col.getTableFilter();
        if (filter != null) {
            Table table = filter.getTable();
            ArrayList<Index> indexes = table.getIndexes();
            Index result = null;
            if (indexes != null) {
                boolean nullable = column.isNullable();
                for (int i = 1, size = indexes.size(); i < size; i++) {
                    Index index = indexes.get(i);
                    if (!index.canFindNext()) {
                        continue;
                    }
                    if (!index.isFirstColumn(column)) {
                        continue;
                    }
                    // Prefer index without nulls last for nullable columns
                    if (result == null || result.getColumns().length > index.getColumns().length || nullable && isNullsLast(result) && !isNullsLast(index)) {
                        result = index;
                    }
                }
            }
            return result;
        }
    }
    return null;
}

Example 8 with Aggregate

use of org.h2.api.Aggregate in project h2database by h2database.

the class JavaAggregate method updateAggregate.

@Override
public void updateAggregate(Session session) {
    HashMap<Expression, Object> group = select.getCurrentGroup();
    if (group == null) {
        // this is a different level (the enclosing query)
        return;
    }
    int groupRowId = select.getCurrentGroupRowId();
    if (lastGroupRowId == groupRowId) {
        // already visited
        return;
    }
    lastGroupRowId = groupRowId;
    if (filterCondition != null) {
        if (!filterCondition.getBooleanValue(session)) {
            return;
        }
    }
    Aggregate agg = (Aggregate) group.get(this);
    try {
        if (agg == null) {
            agg = getInstance();
            group.put(this, agg);
        }
        Object[] argValues = new Object[args.length];
        Object arg = null;
        for (int i = 0, len = args.length; i < len; i++) {
            Value v = args[i].getValue(session);
            v = v.convertTo(argTypes[i]);
            arg = v.getObject();
            argValues[i] = arg;
        }
        if (args.length == 1) {
            agg.add(arg);
        } else {
            agg.add(argValues);
        }
    } catch (SQLException e) {
        throw DbException.convert(e);
    }
}

Example 9 with Aggregate

use of org.h2.api.Aggregate in project h2database by h2database.

the class TestFunctions method testAggregate.

private void testAggregate() throws SQLException {
    deleteDb("functions");
    Connection conn = getConnection("functions");
    Statement stat = conn.createStatement();
    stat.execute("CREATE AGGREGATE SIMPLE_MEDIAN FOR \"" + MedianString.class.getName() + "\"");
    stat.execute("CREATE AGGREGATE IF NOT EXISTS SIMPLE_MEDIAN FOR \"" + MedianString.class.getName() + "\"");
    ResultSet rs = stat.executeQuery("SELECT SIMPLE_MEDIAN(X) FROM SYSTEM_RANGE(1, 9)");
    rs.next();
    assertEquals("5", rs.getString(1));
    conn.close();
    if (config.memory) {
        return;
    }
    conn = getConnection("functions");
    stat = conn.createStatement();
    stat.executeQuery("SELECT SIMPLE_MEDIAN(X) FROM SYSTEM_RANGE(1, 9)");
    DatabaseMetaData meta = conn.getMetaData();
    rs = meta.getProcedures(null, null, "SIMPLE_MEDIAN");
    assertTrue(rs.next());
    assertFalse(rs.next());
    rs = stat.executeQuery("SCRIPT");
    boolean found = false;
    while (rs.next()) {
        String sql = rs.getString(1);
        if (sql.contains("SIMPLE_MEDIAN")) {
            found = true;
        }
    }
    assertTrue(found);
    stat.execute("DROP AGGREGATE SIMPLE_MEDIAN");
    stat.execute("DROP AGGREGATE IF EXISTS SIMPLE_MEDIAN");
    conn.close();
}

Example 10 with Aggregate

use of org.h2.api.Aggregate in project h2database by h2database.

the class TestFunctions method testPrecision.

private void testPrecision() throws SQLException {
    Connection conn = getConnection("functions");
    Statement stat = conn.createStatement();
    stat.execute("create alias no_op for \"" + getClass().getName() + ".noOp\"");
    PreparedStatement prep = conn.prepareStatement("select * from dual where no_op(1.6)=?");
    prep.setBigDecimal(1, new BigDecimal("1.6"));
    ResultSet rs = prep.executeQuery();
    assertTrue(rs.next());
    stat.execute("create aggregate agg_sum for \"" + getClass().getName() + "\"");
    rs = stat.executeQuery("select agg_sum(1), sum(1.6) from dual");
    rs.next();
    assertEquals(1, rs.getMetaData().getScale(2));
    assertEquals(32767, rs.getMetaData().getScale(1));
    stat.executeQuery("select * from information_schema.function_aliases");
    conn.close();
}