use of org.h2.engine.UserAggregate in project h2database by h2database.
the class Parser method readJavaAggregate.
private JavaAggregate readJavaAggregate(UserAggregate aggregate) {
ArrayList<Expression> params = New.arrayList();
do {
params.add(readExpression());
} while (readIfMore(true));
Expression filterCondition;
if (readIf("FILTER")) {
read("(");
read("WHERE");
filterCondition = readExpression();
read(")");
} else {
filterCondition = null;
}
Expression[] list = params.toArray(new Expression[0]);
JavaAggregate agg = new JavaAggregate(aggregate, list, currentSelect, filterCondition);
currentSelect.setGroupQuery();
return agg;
}
use of org.h2.engine.UserAggregate in project h2database by h2database.
the class ScriptCommand method query.
@Override
public ResultInterface query(int maxrows) {
session.getUser().checkAdmin();
reset();
Database db = session.getDatabase();
if (schemaNames != null) {
for (String schemaName : schemaNames) {
Schema schema = db.findSchema(schemaName);
if (schema == null) {
throw DbException.get(ErrorCode.SCHEMA_NOT_FOUND_1, schemaName);
}
}
}
try {
result = createResult();
deleteStore();
openOutput();
if (out != null) {
buffer = new byte[Constants.IO_BUFFER_SIZE];
}
if (settings) {
for (Setting setting : db.getAllSettings()) {
if (setting.getName().equals(SetTypes.getTypeName(SetTypes.CREATE_BUILD))) {
// (it is only set when creating the database)
continue;
}
add(setting.getCreateSQL(), false);
}
}
if (out != null) {
add("", true);
}
for (User user : db.getAllUsers()) {
add(user.getCreateSQL(passwords), false);
}
for (Role role : db.getAllRoles()) {
add(role.getCreateSQL(true), false);
}
for (Schema schema : db.getAllSchemas()) {
if (excludeSchema(schema)) {
continue;
}
add(schema.getCreateSQL(), false);
}
for (UserDataType datatype : db.getAllUserDataTypes()) {
if (drop) {
add(datatype.getDropSQL(), false);
}
add(datatype.getCreateSQL(), false);
}
for (SchemaObject obj : db.getAllSchemaObjects(DbObject.CONSTANT)) {
if (excludeSchema(obj.getSchema())) {
continue;
}
Constant constant = (Constant) obj;
add(constant.getCreateSQL(), false);
}
final ArrayList<Table> tables = db.getAllTablesAndViews(false);
// sort by id, so that views are after tables and views on views
// after the base views
Collections.sort(tables, new Comparator<Table>() {
@Override
public int compare(Table t1, Table t2) {
return t1.getId() - t2.getId();
}
});
// Generate the DROP XXX ... IF EXISTS
for (Table table : tables) {
if (excludeSchema(table.getSchema())) {
continue;
}
if (excludeTable(table)) {
continue;
}
if (table.isHidden()) {
continue;
}
table.lock(session, false, false);
String sql = table.getCreateSQL();
if (sql == null) {
// null for metadata tables
continue;
}
if (drop) {
add(table.getDropSQL(), false);
}
}
for (SchemaObject obj : db.getAllSchemaObjects(DbObject.FUNCTION_ALIAS)) {
if (excludeSchema(obj.getSchema())) {
continue;
}
if (drop) {
add(obj.getDropSQL(), false);
}
add(obj.getCreateSQL(), false);
}
for (UserAggregate agg : db.getAllAggregates()) {
if (drop) {
add(agg.getDropSQL(), false);
}
add(agg.getCreateSQL(), false);
}
for (SchemaObject obj : db.getAllSchemaObjects(DbObject.SEQUENCE)) {
if (excludeSchema(obj.getSchema())) {
continue;
}
Sequence sequence = (Sequence) obj;
if (drop) {
add(sequence.getDropSQL(), false);
}
add(sequence.getCreateSQL(), false);
}
// Generate CREATE TABLE and INSERT...VALUES
int count = 0;
for (Table table : tables) {
if (excludeSchema(table.getSchema())) {
continue;
}
if (excludeTable(table)) {
continue;
}
if (table.isHidden()) {
continue;
}
table.lock(session, false, false);
String createTableSql = table.getCreateSQL();
if (createTableSql == null) {
// null for metadata tables
continue;
}
final TableType tableType = table.getTableType();
add(createTableSql, false);
final ArrayList<Constraint> constraints = table.getConstraints();
if (constraints != null) {
for (Constraint constraint : constraints) {
if (Constraint.Type.PRIMARY_KEY == constraint.getConstraintType()) {
add(constraint.getCreateSQLWithoutIndexes(), false);
}
}
}
if (TableType.TABLE == tableType) {
if (table.canGetRowCount()) {
String rowcount = "-- " + table.getRowCountApproximation() + " +/- SELECT COUNT(*) FROM " + table.getSQL();
add(rowcount, false);
}
if (data) {
count = generateInsertValues(count, table);
}
}
final ArrayList<Index> indexes = table.getIndexes();
for (int j = 0; indexes != null && j < indexes.size(); j++) {
Index index = indexes.get(j);
if (!index.getIndexType().getBelongsToConstraint()) {
add(index.getCreateSQL(), false);
}
}
}
if (tempLobTableCreated) {
add("DROP TABLE IF EXISTS SYSTEM_LOB_STREAM", true);
add("CALL SYSTEM_COMBINE_BLOB(-1)", true);
add("DROP ALIAS IF EXISTS SYSTEM_COMBINE_CLOB", true);
add("DROP ALIAS IF EXISTS SYSTEM_COMBINE_BLOB", true);
tempLobTableCreated = false;
}
// Generate CREATE CONSTRAINT ...
final ArrayList<SchemaObject> constraints = db.getAllSchemaObjects(DbObject.CONSTRAINT);
Collections.sort(constraints, new Comparator<SchemaObject>() {
@Override
public int compare(SchemaObject c1, SchemaObject c2) {
return ((Constraint) c1).compareTo((Constraint) c2);
}
});
for (SchemaObject obj : constraints) {
if (excludeSchema(obj.getSchema())) {
continue;
}
Constraint constraint = (Constraint) obj;
if (excludeTable(constraint.getTable())) {
continue;
}
if (constraint.getTable().isHidden()) {
continue;
}
if (Constraint.Type.PRIMARY_KEY != constraint.getConstraintType()) {
add(constraint.getCreateSQLWithoutIndexes(), false);
}
}
// Generate CREATE TRIGGER ...
for (SchemaObject obj : db.getAllSchemaObjects(DbObject.TRIGGER)) {
if (excludeSchema(obj.getSchema())) {
continue;
}
TriggerObject trigger = (TriggerObject) obj;
if (excludeTable(trigger.getTable())) {
continue;
}
add(trigger.getCreateSQL(), false);
}
// Generate GRANT ...
for (Right right : db.getAllRights()) {
DbObject object = right.getGrantedObject();
if (object != null) {
if (object instanceof Schema) {
if (excludeSchema((Schema) object)) {
continue;
}
} else if (object instanceof Table) {
Table table = (Table) object;
if (excludeSchema(table.getSchema())) {
continue;
}
if (excludeTable(table)) {
continue;
}
}
}
add(right.getCreateSQL(), false);
}
// Generate COMMENT ON ...
for (Comment comment : db.getAllComments()) {
add(comment.getCreateSQL(), false);
}
if (out != null) {
out.close();
}
} catch (IOException e) {
throw DbException.convertIOException(e, getFileName());
} finally {
closeIO();
}
result.done();
LocalResult r = result;
reset();
return r;
}
use of org.h2.engine.UserAggregate 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;
}
use of org.h2.engine.UserAggregate in project h2database by h2database.
the class MetaTable method generateRows.
/**
* Generate the data for the given metadata table using the given first and
* last row filters.
*
* @param session the session
* @param first the first row to return
* @param last the last row to return
* @return the generated rows
*/
public ArrayList<Row> generateRows(Session session, SearchRow first, SearchRow last) {
Value indexFrom = null, indexTo = null;
if (indexColumn >= 0) {
if (first != null) {
indexFrom = first.getValue(indexColumn);
}
if (last != null) {
indexTo = last.getValue(indexColumn);
}
}
ArrayList<Row> rows = New.arrayList();
String catalog = identifier(database.getShortName());
boolean admin = session.getUser().isAdmin();
switch(type) {
case TABLES:
{
for (Table table : getAllTables(session)) {
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
if (hideTable(table, session)) {
continue;
}
String storageType;
if (table.isTemporary()) {
if (table.isGlobalTemporary()) {
storageType = "GLOBAL TEMPORARY";
} else {
storageType = "LOCAL TEMPORARY";
}
} else {
storageType = table.isPersistIndexes() ? "CACHED" : "MEMORY";
}
String sql = table.getCreateSQL();
if (!admin) {
if (sql != null && sql.contains(JdbcSQLException.HIDE_SQL)) {
// hide the password of linked tables
sql = "-";
}
}
add(rows, // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
tableName, // TABLE_TYPE
table.getTableType().toString(), // STORAGE_TYPE
storageType, // SQL
sql, // REMARKS
replaceNullWithEmpty(table.getComment()), // LAST_MODIFICATION
"" + table.getMaxDataModificationId(), // ID
"" + table.getId(), // TYPE_NAME
null, // TABLE_CLASS
table.getClass().getName(), // ROW_COUNT_ESTIMATE
"" + table.getRowCountApproximation());
}
break;
}
case COLUMNS:
{
// reduce the number of tables to scan - makes some metadata queries
// 10x faster
final ArrayList<Table> tablesToList;
if (indexFrom != null && indexFrom.equals(indexTo)) {
String tableName = identifier(indexFrom.getString());
tablesToList = getTablesByName(session, tableName);
} else {
tablesToList = getAllTables(session);
}
for (Table table : tablesToList) {
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
if (hideTable(table, session)) {
continue;
}
Column[] cols = table.getColumns();
String collation = database.getCompareMode().getName();
for (int j = 0; j < cols.length; j++) {
Column c = cols[j];
Sequence sequence = c.getSequence();
add(rows, // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
tableName, // COLUMN_NAME
identifier(c.getName()), // ORDINAL_POSITION
String.valueOf(j + 1), // COLUMN_DEFAULT
c.getDefaultSQL(), // IS_NULLABLE
c.isNullable() ? "YES" : "NO", // DATA_TYPE
"" + DataType.convertTypeToSQLType(c.getType()), // CHARACTER_MAXIMUM_LENGTH
"" + c.getPrecisionAsInt(), // CHARACTER_OCTET_LENGTH
"" + c.getPrecisionAsInt(), // NUMERIC_PRECISION
"" + c.getPrecisionAsInt(), // NUMERIC_PRECISION_RADIX
"10", // NUMERIC_SCALE
"" + c.getScale(), // CHARACTER_SET_NAME
CHARACTER_SET_NAME, // COLLATION_NAME
collation, // TYPE_NAME
identifier(DataType.getDataType(c.getType()).name), // NULLABLE
"" + (c.isNullable() ? DatabaseMetaData.columnNullable : DatabaseMetaData.columnNoNulls), // IS_COMPUTED
"" + (c.getComputed() ? "TRUE" : "FALSE"), // SELECTIVITY
"" + (c.getSelectivity()), // CHECK_CONSTRAINT
c.getCheckConstraintSQL(session, c.getName()), // SEQUENCE_NAME
sequence == null ? null : sequence.getName(), // REMARKS
replaceNullWithEmpty(c.getComment()), // SOURCE_DATA_TYPE
null, // COLUMN_TYPE
c.getCreateSQLWithoutName(), // COLUMN_ON_UPDATE
c.getOnUpdateSQL());
}
}
break;
}
case INDEXES:
{
// reduce the number of tables to scan - makes some metadata queries
// 10x faster
final ArrayList<Table> tablesToList;
if (indexFrom != null && indexFrom.equals(indexTo)) {
String tableName = identifier(indexFrom.getString());
tablesToList = getTablesByName(session, tableName);
} else {
tablesToList = getAllTables(session);
}
for (Table table : tablesToList) {
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
if (hideTable(table, session)) {
continue;
}
ArrayList<Index> indexes = table.getIndexes();
ArrayList<Constraint> constraints = table.getConstraints();
for (int j = 0; indexes != null && j < indexes.size(); j++) {
Index index = indexes.get(j);
if (index.getCreateSQL() == null) {
continue;
}
String constraintName = null;
for (int k = 0; constraints != null && k < constraints.size(); k++) {
Constraint constraint = constraints.get(k);
if (constraint.usesIndex(index)) {
if (index.getIndexType().isPrimaryKey()) {
if (constraint.getConstraintType() == Constraint.Type.PRIMARY_KEY) {
constraintName = constraint.getName();
}
} else {
constraintName = constraint.getName();
}
}
}
IndexColumn[] cols = index.getIndexColumns();
String indexClass;
if (index instanceof MultiVersionIndex) {
indexClass = ((MultiVersionIndex) index).getBaseIndex().getClass().getName();
} else {
indexClass = index.getClass().getName();
}
for (int k = 0; k < cols.length; k++) {
IndexColumn idxCol = cols[k];
Column column = idxCol.column;
add(rows, // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
tableName, // NON_UNIQUE
index.getIndexType().isUnique() ? "FALSE" : "TRUE", // INDEX_NAME
identifier(index.getName()), // ORDINAL_POSITION
"" + (k + 1), // COLUMN_NAME
identifier(column.getName()), // CARDINALITY
"0", // PRIMARY_KEY
index.getIndexType().isPrimaryKey() ? "TRUE" : "FALSE", // INDEX_TYPE_NAME
index.getIndexType().getSQL(), // IS_GENERATED
index.getIndexType().getBelongsToConstraint() ? "TRUE" : "FALSE", // INDEX_TYPE
"" + DatabaseMetaData.tableIndexOther, // ASC_OR_DESC
(idxCol.sortType & SortOrder.DESCENDING) != 0 ? "D" : "A", // PAGES
"0", // FILTER_CONDITION
"", // REMARKS
replaceNullWithEmpty(index.getComment()), // SQL
index.getCreateSQL(), // ID
"" + index.getId(), // SORT_TYPE
"" + idxCol.sortType, // CONSTRAINT_NAME
constraintName, // INDEX_CLASS
indexClass, // AFFINITY
index.getIndexType().isAffinity() ? "TRUE" : "FALSE");
}
}
}
break;
}
case TABLE_TYPES:
{
add(rows, TableType.TABLE.toString());
add(rows, TableType.TABLE_LINK.toString());
add(rows, TableType.SYSTEM_TABLE.toString());
add(rows, TableType.VIEW.toString());
add(rows, TableType.EXTERNAL_TABLE_ENGINE.toString());
break;
}
case CATALOGS:
{
add(rows, catalog);
break;
}
case SETTINGS:
{
for (Setting s : database.getAllSettings()) {
String value = s.getStringValue();
if (value == null) {
value = "" + s.getIntValue();
}
add(rows, identifier(s.getName()), value);
}
add(rows, "info.BUILD_ID", "" + Constants.BUILD_ID);
add(rows, "info.VERSION_MAJOR", "" + Constants.VERSION_MAJOR);
add(rows, "info.VERSION_MINOR", "" + Constants.VERSION_MINOR);
add(rows, "info.VERSION", "" + Constants.getFullVersion());
if (admin) {
String[] settings = { "java.runtime.version", "java.vm.name", "java.vendor", "os.name", "os.arch", "os.version", "sun.os.patch.level", "file.separator", "path.separator", "line.separator", "user.country", "user.language", "user.variant", "file.encoding" };
for (String s : settings) {
add(rows, "property." + s, Utils.getProperty(s, ""));
}
}
add(rows, "EXCLUSIVE", database.getExclusiveSession() == null ? "FALSE" : "TRUE");
add(rows, "MODE", database.getMode().getName());
add(rows, "MULTI_THREADED", database.isMultiThreaded() ? "1" : "0");
add(rows, "MVCC", database.isMultiVersion() ? "TRUE" : "FALSE");
add(rows, "QUERY_TIMEOUT", "" + session.getQueryTimeout());
add(rows, "RETENTION_TIME", "" + database.getRetentionTime());
add(rows, "LOG", "" + database.getLogMode());
// database settings
ArrayList<String> settingNames = New.arrayList();
HashMap<String, String> s = database.getSettings().getSettings();
settingNames.addAll(s.keySet());
Collections.sort(settingNames);
for (String k : settingNames) {
add(rows, k, s.get(k));
}
if (database.isPersistent()) {
PageStore store = database.getPageStore();
if (store != null) {
add(rows, "info.FILE_WRITE_TOTAL", "" + store.getWriteCountTotal());
add(rows, "info.FILE_WRITE", "" + store.getWriteCount());
add(rows, "info.FILE_READ", "" + store.getReadCount());
add(rows, "info.PAGE_COUNT", "" + store.getPageCount());
add(rows, "info.PAGE_SIZE", "" + store.getPageSize());
add(rows, "info.CACHE_MAX_SIZE", "" + store.getCache().getMaxMemory());
add(rows, "info.CACHE_SIZE", "" + store.getCache().getMemory());
}
Store mvStore = database.getMvStore();
if (mvStore != null) {
FileStore fs = mvStore.getStore().getFileStore();
add(rows, "info.FILE_WRITE", "" + fs.getWriteCount());
add(rows, "info.FILE_READ", "" + fs.getReadCount());
long size;
try {
size = fs.getFile().size();
} catch (IOException e) {
throw DbException.convertIOException(e, "Can not get size");
}
int pageSize = 4 * 1024;
long pageCount = size / pageSize;
add(rows, "info.PAGE_COUNT", "" + pageCount);
add(rows, "info.PAGE_SIZE", "" + pageSize);
add(rows, "info.CACHE_MAX_SIZE", "" + mvStore.getStore().getCacheSize());
add(rows, "info.CACHE_SIZE", "" + mvStore.getStore().getCacheSizeUsed());
}
}
break;
}
case TYPE_INFO:
{
for (DataType t : DataType.getTypes()) {
if (t.hidden || t.sqlType == Value.NULL) {
continue;
}
add(rows, // TYPE_NAME
t.name, // DATA_TYPE
String.valueOf(t.sqlType), // PRECISION
String.valueOf(MathUtils.convertLongToInt(t.maxPrecision)), // PREFIX
t.prefix, // SUFFIX
t.suffix, // PARAMS
t.params, // AUTO_INCREMENT
String.valueOf(t.autoIncrement), // MINIMUM_SCALE
String.valueOf(t.minScale), // MAXIMUM_SCALE
String.valueOf(t.maxScale), // RADIX
t.decimal ? "10" : null, // POS
String.valueOf(t.sqlTypePos), // CASE_SENSITIVE
String.valueOf(t.caseSensitive), // NULLABLE
"" + DatabaseMetaData.typeNullable, // SEARCHABLE
"" + DatabaseMetaData.typeSearchable);
}
break;
}
case HELP:
{
String resource = "/org/h2/res/help.csv";
try {
byte[] data = Utils.getResource(resource);
Reader reader = new InputStreamReader(new ByteArrayInputStream(data));
Csv csv = new Csv();
csv.setLineCommentCharacter('#');
ResultSet rs = csv.read(reader, null);
for (int i = 0; rs.next(); i++) {
add(rows, // ID
String.valueOf(i), // SECTION
rs.getString(1).trim(), // TOPIC
rs.getString(2).trim(), // SYNTAX
rs.getString(3).trim(), // TEXT
rs.getString(4).trim());
}
} catch (Exception e) {
throw DbException.convert(e);
}
break;
}
case SEQUENCES:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.SEQUENCE)) {
Sequence s = (Sequence) obj;
add(rows, // SEQUENCE_CATALOG
catalog, // SEQUENCE_SCHEMA
identifier(s.getSchema().getName()), // SEQUENCE_NAME
identifier(s.getName()), // CURRENT_VALUE
String.valueOf(s.getCurrentValue()), // INCREMENT
String.valueOf(s.getIncrement()), // IS_GENERATED
s.getBelongsToTable() ? "TRUE" : "FALSE", // REMARKS
replaceNullWithEmpty(s.getComment()), // CACHE
String.valueOf(s.getCacheSize()), // MIN_VALUE
String.valueOf(s.getMinValue()), // MAX_VALUE
String.valueOf(s.getMaxValue()), // IS_CYCLE
s.getCycle() ? "TRUE" : "FALSE", // ID
"" + s.getId());
}
break;
}
case USERS:
{
for (User u : database.getAllUsers()) {
if (admin || session.getUser() == u) {
add(rows, // NAME
identifier(u.getName()), // ADMIN
String.valueOf(u.isAdmin()), // REMARKS
replaceNullWithEmpty(u.getComment()), // ID
"" + u.getId());
}
}
break;
}
case ROLES:
{
for (Role r : database.getAllRoles()) {
if (admin || session.getUser().isRoleGranted(r)) {
add(rows, // NAME
identifier(r.getName()), // REMARKS
replaceNullWithEmpty(r.getComment()), // ID
"" + r.getId());
}
}
break;
}
case RIGHTS:
{
if (admin) {
for (Right r : database.getAllRights()) {
Role role = r.getGrantedRole();
DbObject grantee = r.getGrantee();
String rightType = grantee.getType() == DbObject.USER ? "USER" : "ROLE";
if (role == null) {
DbObject object = r.getGrantedObject();
Schema schema = null;
Table table = null;
if (object != null) {
if (object instanceof Schema) {
schema = (Schema) object;
} else if (object instanceof Table) {
table = (Table) object;
schema = table.getSchema();
}
}
String tableName = (table != null) ? identifier(table.getName()) : "";
String schemaName = (schema != null) ? identifier(schema.getName()) : "";
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
add(rows, // GRANTEE
identifier(grantee.getName()), // GRANTEETYPE
rightType, // GRANTEDROLE
"", // RIGHTS
r.getRights(), // TABLE_SCHEMA
schemaName, // TABLE_NAME
tableName, // ID
"" + r.getId());
} else {
add(rows, // GRANTEE
identifier(grantee.getName()), // GRANTEETYPE
rightType, // GRANTEDROLE
identifier(role.getName()), // RIGHTS
"", // TABLE_SCHEMA
"", // TABLE_NAME
"", // ID
"" + r.getId());
}
}
}
break;
}
case FUNCTION_ALIASES:
{
for (SchemaObject aliasAsSchemaObject : database.getAllSchemaObjects(DbObject.FUNCTION_ALIAS)) {
FunctionAlias alias = (FunctionAlias) aliasAsSchemaObject;
JavaMethod[] methods;
try {
methods = alias.getJavaMethods();
} catch (DbException e) {
methods = new JavaMethod[0];
}
for (FunctionAlias.JavaMethod method : methods) {
int returnsResult = method.getDataType() == Value.NULL ? DatabaseMetaData.procedureNoResult : DatabaseMetaData.procedureReturnsResult;
add(rows, // ALIAS_CATALOG
catalog, // ALIAS_SCHEMA
alias.getSchema().getName(), // ALIAS_NAME
identifier(alias.getName()), // JAVA_CLASS
alias.getJavaClassName(), // JAVA_METHOD
alias.getJavaMethodName(), // DATA_TYPE
"" + DataType.convertTypeToSQLType(method.getDataType()), // TYPE_NAME
DataType.getDataType(method.getDataType()).name, // COLUMN_COUNT INT
"" + method.getParameterCount(), // RETURNS_RESULT SMALLINT
"" + returnsResult, // REMARKS
replaceNullWithEmpty(alias.getComment()), // ID
"" + alias.getId(), // SOURCE
alias.getSource());
}
}
for (UserAggregate agg : database.getAllAggregates()) {
int returnsResult = DatabaseMetaData.procedureReturnsResult;
add(rows, // ALIAS_CATALOG
catalog, // ALIAS_SCHEMA
Constants.SCHEMA_MAIN, // ALIAS_NAME
identifier(agg.getName()), // JAVA_CLASS
agg.getJavaClassName(), // JAVA_METHOD
"", // DATA_TYPE
"" + DataType.convertTypeToSQLType(Value.NULL), // TYPE_NAME
DataType.getDataType(Value.NULL).name, // COLUMN_COUNT INT
"1", // RETURNS_RESULT SMALLINT
"" + returnsResult, // REMARKS
replaceNullWithEmpty(agg.getComment()), // ID
"" + agg.getId(), // SOURCE
"");
}
break;
}
case FUNCTION_COLUMNS:
{
for (SchemaObject aliasAsSchemaObject : database.getAllSchemaObjects(DbObject.FUNCTION_ALIAS)) {
FunctionAlias alias = (FunctionAlias) aliasAsSchemaObject;
JavaMethod[] methods;
try {
methods = alias.getJavaMethods();
} catch (DbException e) {
methods = new JavaMethod[0];
}
for (FunctionAlias.JavaMethod method : methods) {
// Add return column index 0
if (method.getDataType() != Value.NULL) {
DataType dt = DataType.getDataType(method.getDataType());
add(rows, // ALIAS_CATALOG
catalog, // ALIAS_SCHEMA
alias.getSchema().getName(), // ALIAS_NAME
identifier(alias.getName()), // JAVA_CLASS
alias.getJavaClassName(), // JAVA_METHOD
alias.getJavaMethodName(), // COLUMN_COUNT
"" + method.getParameterCount(), // POS INT
"0", // COLUMN_NAME
"P0", // DATA_TYPE
"" + DataType.convertTypeToSQLType(method.getDataType()), // TYPE_NAME
dt.name, // PRECISION INT
"" + MathUtils.convertLongToInt(dt.defaultPrecision), // SCALE
"" + dt.defaultScale, // RADIX
"10", // NULLABLE SMALLINT
"" + DatabaseMetaData.columnNullableUnknown, // COLUMN_TYPE
"" + DatabaseMetaData.procedureColumnReturn, // REMARKS
"", // COLUMN_DEFAULT
null);
}
Class<?>[] columnList = method.getColumnClasses();
for (int k = 0; k < columnList.length; k++) {
if (method.hasConnectionParam() && k == 0) {
continue;
}
Class<?> clazz = columnList[k];
int dataType = DataType.getTypeFromClass(clazz);
DataType dt = DataType.getDataType(dataType);
int nullable = clazz.isPrimitive() ? DatabaseMetaData.columnNoNulls : DatabaseMetaData.columnNullable;
add(rows, // ALIAS_CATALOG
catalog, // ALIAS_SCHEMA
alias.getSchema().getName(), // ALIAS_NAME
identifier(alias.getName()), // JAVA_CLASS
alias.getJavaClassName(), // JAVA_METHOD
alias.getJavaMethodName(), // COLUMN_COUNT
"" + method.getParameterCount(), // POS INT
"" + (k + (method.hasConnectionParam() ? 0 : 1)), // COLUMN_NAME
"P" + (k + 1), // DATA_TYPE
"" + DataType.convertTypeToSQLType(dt.type), // TYPE_NAME
dt.name, // PRECISION INT
"" + MathUtils.convertLongToInt(dt.defaultPrecision), // SCALE
"" + dt.defaultScale, // RADIX
"10", // NULLABLE SMALLINT
"" + nullable, // COLUMN_TYPE
"" + DatabaseMetaData.procedureColumnIn, // REMARKS
"", // COLUMN_DEFAULT
null);
}
}
}
break;
}
case SCHEMATA:
{
String collation = database.getCompareMode().getName();
for (Schema schema : database.getAllSchemas()) {
add(rows, // CATALOG_NAME
catalog, // SCHEMA_NAME
identifier(schema.getName()), // SCHEMA_OWNER
identifier(schema.getOwner().getName()), // DEFAULT_CHARACTER_SET_NAME
CHARACTER_SET_NAME, // DEFAULT_COLLATION_NAME
collation, // IS_DEFAULT
Constants.SCHEMA_MAIN.equals(schema.getName()) ? "TRUE" : "FALSE", // REMARKS
replaceNullWithEmpty(schema.getComment()), // ID
"" + schema.getId());
}
break;
}
case TABLE_PRIVILEGES:
{
for (Right r : database.getAllRights()) {
DbObject object = r.getGrantedObject();
if (!(object instanceof Table)) {
continue;
}
Table table = (Table) object;
if (hideTable(table, session)) {
continue;
}
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
addPrivileges(rows, r.getGrantee(), catalog, table, null, r.getRightMask());
}
break;
}
case COLUMN_PRIVILEGES:
{
for (Right r : database.getAllRights()) {
DbObject object = r.getGrantedObject();
if (!(object instanceof Table)) {
continue;
}
Table table = (Table) object;
if (hideTable(table, session)) {
continue;
}
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
DbObject grantee = r.getGrantee();
int mask = r.getRightMask();
for (Column column : table.getColumns()) {
addPrivileges(rows, grantee, catalog, table, column.getName(), mask);
}
}
break;
}
case COLLATIONS:
{
for (Locale l : Collator.getAvailableLocales()) {
add(rows, // NAME
CompareMode.getName(l), // KEY
l.toString());
}
break;
}
case VIEWS:
{
for (Table table : getAllTables(session)) {
if (table.getTableType() != TableType.VIEW) {
continue;
}
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
TableView view = (TableView) table;
add(rows, // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
tableName, // VIEW_DEFINITION
table.getCreateSQL(), // CHECK_OPTION
"NONE", // IS_UPDATABLE
"NO", // STATUS
view.isInvalid() ? "INVALID" : "VALID", // REMARKS
replaceNullWithEmpty(view.getComment()), // ID
"" + view.getId());
}
break;
}
case IN_DOUBT:
{
ArrayList<InDoubtTransaction> prepared = database.getInDoubtTransactions();
if (prepared != null && admin) {
for (InDoubtTransaction prep : prepared) {
add(rows, // TRANSACTION
prep.getTransactionName(), // STATE
prep.getState());
}
}
break;
}
case CROSS_REFERENCES:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.CONSTRAINT)) {
Constraint constraint = (Constraint) obj;
if (constraint.getConstraintType() != Constraint.Type.REFERENTIAL) {
continue;
}
ConstraintReferential ref = (ConstraintReferential) constraint;
IndexColumn[] cols = ref.getColumns();
IndexColumn[] refCols = ref.getRefColumns();
Table tab = ref.getTable();
Table refTab = ref.getRefTable();
String tableName = identifier(refTab.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
int update = getRefAction(ref.getUpdateAction());
int delete = getRefAction(ref.getDeleteAction());
for (int j = 0; j < cols.length; j++) {
add(rows, // PKTABLE_CATALOG
catalog, // PKTABLE_SCHEMA
identifier(refTab.getSchema().getName()), // PKTABLE_NAME
identifier(refTab.getName()), // PKCOLUMN_NAME
identifier(refCols[j].column.getName()), // FKTABLE_CATALOG
catalog, // FKTABLE_SCHEMA
identifier(tab.getSchema().getName()), // FKTABLE_NAME
identifier(tab.getName()), // FKCOLUMN_NAME
identifier(cols[j].column.getName()), // ORDINAL_POSITION
String.valueOf(j + 1), // UPDATE_RULE SMALLINT
String.valueOf(update), // DELETE_RULE SMALLINT
String.valueOf(delete), // FK_NAME
identifier(ref.getName()), // PK_NAME
identifier(ref.getUniqueIndex().getName()), // DEFERRABILITY
"" + DatabaseMetaData.importedKeyNotDeferrable);
}
}
break;
}
case CONSTRAINTS:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.CONSTRAINT)) {
Constraint constraint = (Constraint) obj;
Constraint.Type constraintType = constraint.getConstraintType();
String checkExpression = null;
IndexColumn[] indexColumns = null;
Table table = constraint.getTable();
if (hideTable(table, session)) {
continue;
}
Index index = constraint.getUniqueIndex();
String uniqueIndexName = null;
if (index != null) {
uniqueIndexName = index.getName();
}
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
if (constraintType == Constraint.Type.CHECK) {
checkExpression = ((ConstraintCheck) constraint).getExpression().getSQL();
} else if (constraintType == Constraint.Type.UNIQUE || constraintType == Constraint.Type.PRIMARY_KEY) {
indexColumns = ((ConstraintUnique) constraint).getColumns();
} else if (constraintType == Constraint.Type.REFERENTIAL) {
indexColumns = ((ConstraintReferential) constraint).getColumns();
}
String columnList = null;
if (indexColumns != null) {
StatementBuilder buff = new StatementBuilder();
for (IndexColumn col : indexColumns) {
buff.appendExceptFirst(",");
buff.append(col.column.getName());
}
columnList = buff.toString();
}
add(rows, // CONSTRAINT_CATALOG
catalog, // CONSTRAINT_SCHEMA
identifier(constraint.getSchema().getName()), // CONSTRAINT_NAME
identifier(constraint.getName()), // CONSTRAINT_TYPE
constraintType == Constraint.Type.PRIMARY_KEY ? constraintType.getSqlName() : constraintType.name(), // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
tableName, // UNIQUE_INDEX_NAME
uniqueIndexName, // CHECK_EXPRESSION
checkExpression, // COLUMN_LIST
columnList, // REMARKS
replaceNullWithEmpty(constraint.getComment()), // SQL
constraint.getCreateSQL(), // ID
"" + constraint.getId());
}
break;
}
case CONSTANTS:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.CONSTANT)) {
Constant constant = (Constant) obj;
ValueExpression expr = constant.getValue();
add(rows, // CONSTANT_CATALOG
catalog, // CONSTANT_SCHEMA
identifier(constant.getSchema().getName()), // CONSTANT_NAME
identifier(constant.getName()), // CONSTANT_TYPE
"" + DataType.convertTypeToSQLType(expr.getType()), // REMARKS
replaceNullWithEmpty(constant.getComment()), // SQL
expr.getSQL(), // ID
"" + constant.getId());
}
break;
}
case DOMAINS:
{
for (UserDataType dt : database.getAllUserDataTypes()) {
Column col = dt.getColumn();
add(rows, // DOMAIN_CATALOG
catalog, // DOMAIN_SCHEMA
Constants.SCHEMA_MAIN, // DOMAIN_NAME
identifier(dt.getName()), // COLUMN_DEFAULT
col.getDefaultSQL(), // IS_NULLABLE
col.isNullable() ? "YES" : "NO", // DATA_TYPE
"" + col.getDataType().sqlType, // PRECISION INT
"" + col.getPrecisionAsInt(), // SCALE INT
"" + col.getScale(), // TYPE_NAME
col.getDataType().name, // SELECTIVITY INT
"" + col.getSelectivity(), // CHECK_CONSTRAINT
"" + col.getCheckConstraintSQL(session, "VALUE"), // REMARKS
replaceNullWithEmpty(dt.getComment()), // SQL
"" + dt.getCreateSQL(), // ID
"" + dt.getId());
}
break;
}
case TRIGGERS:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.TRIGGER)) {
TriggerObject trigger = (TriggerObject) obj;
Table table = trigger.getTable();
add(rows, // TRIGGER_CATALOG
catalog, // TRIGGER_SCHEMA
identifier(trigger.getSchema().getName()), // TRIGGER_NAME
identifier(trigger.getName()), // TRIGGER_TYPE
trigger.getTypeNameList(), // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
identifier(table.getName()), // BEFORE BIT
"" + trigger.isBefore(), // JAVA_CLASS
trigger.getTriggerClassName(), // QUEUE_SIZE INT
"" + trigger.getQueueSize(), // NO_WAIT BIT
"" + trigger.isNoWait(), // REMARKS
replaceNullWithEmpty(trigger.getComment()), // SQL
trigger.getCreateSQL(), // ID
"" + trigger.getId());
}
break;
}
case SESSIONS:
{
long now = System.currentTimeMillis();
for (Session s : database.getSessions(false)) {
if (admin || s == session) {
Command command = s.getCurrentCommand();
long start = s.getCurrentCommandStart();
if (start == 0) {
start = now;
}
add(rows, // ID
"" + s.getId(), // USER_NAME
s.getUser().getName(), // SESSION_START
new Timestamp(s.getSessionStart()).toString(), // STATEMENT
command == null ? null : command.toString(), // STATEMENT_START
new Timestamp(start).toString(), // CONTAINS_UNCOMMITTED
"" + s.containsUncommitted());
}
}
break;
}
case LOCKS:
{
for (Session s : database.getSessions(false)) {
if (admin || s == session) {
for (Table table : s.getLocks()) {
add(rows, // TABLE_SCHEMA
table.getSchema().getName(), // TABLE_NAME
table.getName(), // SESSION_ID
"" + s.getId(), // LOCK_TYPE
table.isLockedExclusivelyBy(s) ? "WRITE" : "READ");
}
}
}
break;
}
case SESSION_STATE:
{
for (String name : session.getVariableNames()) {
Value v = session.getVariable(name);
add(rows, // KEY
"@" + name, // SQL
"SET @" + name + " " + v.getSQL());
}
for (Table table : session.getLocalTempTables()) {
add(rows, // KEY
"TABLE " + table.getName(), // SQL
table.getCreateSQL());
}
String[] path = session.getSchemaSearchPath();
if (path != null && path.length > 0) {
StatementBuilder buff = new StatementBuilder("SET SCHEMA_SEARCH_PATH ");
for (String p : path) {
buff.appendExceptFirst(", ");
buff.append(StringUtils.quoteIdentifier(p));
}
add(rows, // KEY
"SCHEMA_SEARCH_PATH", // SQL
buff.toString());
}
String schema = session.getCurrentSchemaName();
if (schema != null) {
add(rows, // KEY
"SCHEMA", // SQL
"SET SCHEMA " + StringUtils.quoteIdentifier(schema));
}
break;
}
case QUERY_STATISTICS:
{
QueryStatisticsData control = database.getQueryStatisticsData();
if (control != null) {
for (QueryStatisticsData.QueryEntry entry : control.getQueries()) {
add(rows, // SQL_STATEMENT
entry.sqlStatement, // EXECUTION_COUNT
"" + entry.count, // MIN_EXECUTION_TIME
"" + entry.executionTimeMinNanos / 1000d / 1000, // MAX_EXECUTION_TIME
"" + entry.executionTimeMaxNanos / 1000d / 1000, // CUMULATIVE_EXECUTION_TIME
"" + entry.executionTimeCumulativeNanos / 1000d / 1000, // AVERAGE_EXECUTION_TIME
"" + entry.executionTimeMeanNanos / 1000d / 1000, // STD_DEV_EXECUTION_TIME
"" + entry.getExecutionTimeStandardDeviation() / 1000d / 1000, // MIN_ROW_COUNT
"" + entry.rowCountMin, // MAX_ROW_COUNT
"" + entry.rowCountMax, // CUMULATIVE_ROW_COUNT
"" + entry.rowCountCumulative, // AVERAGE_ROW_COUNT
"" + entry.rowCountMean, // STD_DEV_ROW_COUNT
"" + entry.getRowCountStandardDeviation());
}
}
break;
}
case SYNONYMS:
{
for (TableSynonym synonym : database.getAllSynonyms()) {
add(rows, // SYNONYM_CATALOG
catalog, // SYNONYM_SCHEMA
identifier(synonym.getSchema().getName()), // SYNONYM_NAME
identifier(synonym.getName()), // SYNONYM_FOR
synonym.getSynonymForName(), // SYNONYM_FOR_SCHEMA
synonym.getSynonymForSchema().getName(), // TYPE NAME
"SYNONYM", // STATUS
"VALID", // REMARKS
replaceNullWithEmpty(synonym.getComment()), // ID
"" + synonym.getId());
}
break;
}
case TABLE_CONSTRAINTS:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.CONSTRAINT)) {
Constraint constraint = (Constraint) obj;
Constraint.Type constraintType = constraint.getConstraintType();
Table table = constraint.getTable();
if (hideTable(table, session)) {
continue;
}
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
add(rows, // CONSTRAINT_CATALOG
catalog, // CONSTRAINT_SCHEMA
identifier(constraint.getSchema().getName()), // CONSTRAINT_NAME
identifier(constraint.getName()), // CONSTRAINT_TYPE
constraintType.getSqlName(), // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
tableName, // IS_DEFERRABLE
"NO", // INITIALLY_DEFERRED
"NO");
}
break;
}
case KEY_COLUMN_USAGE:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.CONSTRAINT)) {
Constraint constraint = (Constraint) obj;
Constraint.Type constraintType = constraint.getConstraintType();
IndexColumn[] indexColumns = null;
Table table = constraint.getTable();
if (hideTable(table, session)) {
continue;
}
String tableName = identifier(table.getName());
if (!checkIndex(session, tableName, indexFrom, indexTo)) {
continue;
}
if (constraintType == Constraint.Type.UNIQUE || constraintType == Constraint.Type.PRIMARY_KEY) {
indexColumns = ((ConstraintUnique) constraint).getColumns();
} else if (constraintType == Constraint.Type.REFERENTIAL) {
indexColumns = ((ConstraintReferential) constraint).getColumns();
}
if (indexColumns == null) {
continue;
}
ConstraintUnique referenced;
if (constraintType == Constraint.Type.REFERENTIAL) {
referenced = lookupUniqueForReferential((ConstraintReferential) constraint);
} else {
referenced = null;
}
for (int i = 0; i < indexColumns.length; i++) {
IndexColumn indexColumn = indexColumns[i];
String ordinalPosition = Integer.toString(i + 1);
String positionInUniqueConstraint;
if (constraintType == Constraint.Type.REFERENTIAL) {
positionInUniqueConstraint = ordinalPosition;
if (referenced != null) {
Column c = ((ConstraintReferential) constraint).getRefColumns()[i].column;
IndexColumn[] refColumns = referenced.getColumns();
for (int j = 0; j < refColumns.length; j++) {
if (refColumns[j].column.equals(c)) {
positionInUniqueConstraint = Integer.toString(j + 1);
break;
}
}
}
} else {
positionInUniqueConstraint = null;
}
add(rows, // CONSTRAINT_CATALOG
catalog, // CONSTRAINT_SCHEMA
identifier(constraint.getSchema().getName()), // CONSTRAINT_NAME
identifier(constraint.getName()), // TABLE_CATALOG
catalog, // TABLE_SCHEMA
identifier(table.getSchema().getName()), // TABLE_NAME
tableName, // COLUMN_NAME
indexColumn.columnName, // ORDINAL_POSITION
ordinalPosition, // POSITION_IN_UNIQUE_CONSTRAINT
positionInUniqueConstraint);
}
}
break;
}
case REFERENTIAL_CONSTRAINTS:
{
for (SchemaObject obj : database.getAllSchemaObjects(DbObject.CONSTRAINT)) {
if (((Constraint) obj).getConstraintType() != Constraint.Type.REFERENTIAL) {
continue;
}
ConstraintReferential constraint = (ConstraintReferential) obj;
Table table = constraint.getTable();
if (hideTable(table, session)) {
continue;
}
// Should be referenced unique constraint, but H2 uses indexes instead.
// So try to find matching unique constraint first and there is no such
// constraint use index name to return something.
SchemaObject unique = lookupUniqueForReferential(constraint);
if (unique == null) {
unique = constraint.getUniqueIndex();
}
add(rows, // CONSTRAINT_CATALOG
catalog, // CONSTRAINT_SCHEMA
identifier(constraint.getSchema().getName()), // CONSTRAINT_NAME
identifier(constraint.getName()), // UNIQUE_CONSTRAINT_CATALOG
catalog, // UNIQUE_CONSTRAINT_SCHEMA
identifier(unique.getSchema().getName()), // UNIQUE_CONSTRAINT_NAME
unique.getName(), // MATCH_OPTION
"NONE", // UPDATE_RULE
constraint.getUpdateAction().getSqlName(), // DELETE_RULE
constraint.getDeleteAction().getSqlName());
}
break;
}
default:
DbException.throwInternalError("type=" + type);
}
return rows;
}
use of org.h2.engine.UserAggregate in project h2database by h2database.
the class Parser method readFunction.
private Expression readFunction(Schema schema, String name) {
if (schema != null) {
return readJavaFunction(schema, name, true);
}
boolean allowOverride = database.isAllowBuiltinAliasOverride();
if (allowOverride) {
JavaFunction jf = readJavaFunction(null, name, false);
if (jf != null) {
return jf;
}
}
AggregateType agg = getAggregateType(name);
if (agg != null) {
return readAggregate(agg, name);
}
Function function = Function.getFunction(database, name);
if (function == null) {
UserAggregate aggregate = database.findAggregate(name);
if (aggregate != null) {
return readJavaAggregate(aggregate);
}
if (allowOverride) {
throw DbException.get(ErrorCode.FUNCTION_NOT_FOUND_1, name);
}
return readJavaFunction(null, name, true);
}
switch(function.getFunctionType()) {
case Function.CAST:
{
function.setParameter(0, readExpression());
read("AS");
Column type = parseColumnWithType(null);
function.setDataType(type);
read(")");
break;
}
case Function.CONVERT:
{
if (database.getMode().swapConvertFunctionParameters) {
Column type = parseColumnWithType(null);
function.setDataType(type);
read(",");
function.setParameter(0, readExpression());
read(")");
} else {
function.setParameter(0, readExpression());
read(",");
Column type = parseColumnWithType(null);
function.setDataType(type);
read(")");
}
break;
}
case Function.EXTRACT:
{
function.setParameter(0, ValueExpression.get(ValueString.get(currentToken)));
read();
read("FROM");
function.setParameter(1, readExpression());
read(")");
break;
}
case Function.DATE_ADD:
case Function.DATE_DIFF:
{
if (DateTimeFunctions.isDatePart(currentToken)) {
function.setParameter(0, ValueExpression.get(ValueString.get(currentToken)));
read();
} else {
function.setParameter(0, readExpression());
}
read(",");
function.setParameter(1, readExpression());
read(",");
function.setParameter(2, readExpression());
read(")");
break;
}
case Function.SUBSTRING:
{
// Different variants include:
// SUBSTRING(X,1)
// SUBSTRING(X,1,1)
// SUBSTRING(X FROM 1 FOR 1) -- Postgres
// SUBSTRING(X FROM 1) -- Postgres
// SUBSTRING(X FOR 1) -- Postgres
function.setParameter(0, readExpression());
if (readIf("FROM")) {
function.setParameter(1, readExpression());
if (readIf("FOR")) {
function.setParameter(2, readExpression());
}
} else if (readIf("FOR")) {
function.setParameter(1, ValueExpression.get(ValueInt.get(0)));
function.setParameter(2, readExpression());
} else {
read(",");
function.setParameter(1, readExpression());
if (readIf(",")) {
function.setParameter(2, readExpression());
}
}
read(")");
break;
}
case Function.POSITION:
{
// can't read expression because IN would be read too early
function.setParameter(0, readConcat());
if (!readIf(",")) {
read("IN");
}
function.setParameter(1, readExpression());
read(")");
break;
}
case Function.TRIM:
{
Expression space = null;
if (readIf("LEADING")) {
function = Function.getFunction(database, "LTRIM");
if (!readIf("FROM")) {
space = readExpression();
read("FROM");
}
} else if (readIf("TRAILING")) {
function = Function.getFunction(database, "RTRIM");
if (!readIf("FROM")) {
space = readExpression();
read("FROM");
}
} else if (readIf("BOTH")) {
if (!readIf("FROM")) {
space = readExpression();
read("FROM");
}
}
Expression p0 = readExpression();
if (readIf(",")) {
space = readExpression();
} else if (readIf("FROM")) {
space = p0;
p0 = readExpression();
}
function.setParameter(0, p0);
if (space != null) {
function.setParameter(1, space);
}
read(")");
break;
}
case Function.TABLE:
case Function.TABLE_DISTINCT:
{
int i = 0;
ArrayList<Column> columns = New.arrayList();
do {
String columnName = readAliasIdentifier();
Column column = parseColumnWithType(columnName);
columns.add(column);
read("=");
function.setParameter(i, readExpression());
i++;
} while (readIfMore(true));
TableFunction tf = (TableFunction) function;
tf.setColumns(columns);
break;
}
case Function.ROW_NUMBER:
read(")");
read("OVER");
read("(");
read(")");
if (currentSelect == null && currentPrepared == null) {
throw getSyntaxError();
}
return new Rownum(currentSelect == null ? currentPrepared : currentSelect);
default:
if (!readIf(")")) {
int i = 0;
do {
function.setParameter(i++, readExpression());
} while (readIfMore(true));
}
}
function.doneWithParameters();
return function;
}
Aggregations