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

Example 16 with Aggregate

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

the class CreateAggregate method update.

@Override
public int update() {
    session.commit(true);
    session.getUser().checkAdmin();
    Database db = session.getDatabase();
    if (db.findAggregate(name) != null || schema.findFunction(name) != null) {
        if (!ifNotExists) {
            throw DbException.get(ErrorCode.FUNCTION_ALIAS_ALREADY_EXISTS_1, name);
        }
    } else {
        int id = getObjectId();
        UserAggregate aggregate = new UserAggregate(db, id, name, javaClassMethod, force);
        db.addDatabaseObject(session, aggregate);
    }
    return 0;
}

Example 17 with Aggregate

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

the class JavaAggregate method getValue.

@Override
public Value getValue(Session session) {
    HashMap<Expression, Object> group = select.getCurrentGroup();
    if (group == null) {
        throw DbException.get(ErrorCode.INVALID_USE_OF_AGGREGATE_FUNCTION_1, getSQL());
    }
    try {
        Aggregate agg = (Aggregate) group.get(this);
        if (agg == null) {
            agg = getInstance();
        }
        Object obj = agg.getResult();
        if (obj == null) {
            return ValueNull.INSTANCE;
        }
        return DataType.convertToValue(session, obj, dataType);
    } catch (SQLException e) {
        throw DbException.convert(e);
    }
}

Example 18 with Aggregate

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

the class JavaAggregate method getInstance.

private Aggregate getInstance() throws SQLException {
    Aggregate agg = userAggregate.getInstance();
    agg.init(userConnection);
    return agg;
}

Example 19 with Aggregate

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

the class JavaAggregate method optimize.

@Override
public Expression optimize(Session session) {
    userConnection = session.createConnection(false);
    int len = args.length;
    argTypes = new int[len];
    for (int i = 0; i < len; i++) {
        Expression expr = args[i];
        args[i] = expr.optimize(session);
        int type = expr.getType();
        argTypes[i] = type;
    }
    try {
        Aggregate aggregate = getInstance();
        dataType = aggregate.getInternalType(argTypes);
    } catch (SQLException e) {
        throw DbException.convert(e);
    }
    if (filterCondition != null) {
        filterCondition = filterCondition.optimize(session);
    }
    return this;
}

Example 20 with Aggregate

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

the class TestFunctions method testAggregateType.

private void testAggregateType() throws SQLException {
    deleteDb("functions");
    Connection conn = getConnection("functions");
    Statement stat = conn.createStatement();
    stat.execute("CREATE AGGREGATE SIMPLE_MEDIAN FOR \"" + MedianStringType.class.getName() + "\"");
    stat.execute("CREATE AGGREGATE IF NOT EXISTS SIMPLE_MEDIAN FOR \"" + MedianStringType.class.getName() + "\"");
    ResultSet rs = stat.executeQuery("SELECT SIMPLE_MEDIAN(X) FROM SYSTEM_RANGE(1, 9)");
    rs.next();
    assertEquals("5", rs.getString(1));
    rs = stat.executeQuery("SELECT SIMPLE_MEDIAN(X) FILTER (WHERE X > 2) FROM SYSTEM_RANGE(1, 9)");
    rs.next();
    assertEquals("6", 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();
}