提交 cceb5a12 authored 作者: noelgrandin's avatar noelgrandin

Issue 510: Make org.h2.bnf public for consumption by external projects, patch by Nicolas Fortin

上级 5da851fe
...@@ -489,7 +489,7 @@ CREATE DOMAIN EMAIL AS VARCHAR(255) CHECK (POSITION('@', VALUE) > 1) ...@@ -489,7 +489,7 @@ CREATE DOMAIN EMAIL AS VARCHAR(255) CHECK (POSITION('@', VALUE) > 1)
"Commands (DDL)","CREATE INDEX"," "Commands (DDL)","CREATE INDEX","
CREATE CREATE
{ [ UNIQUE ] [ HASH ] INDEX [ [ IF NOT EXISTS ] newIndexName ] { [ UNIQUE ] [ HASH ] [ SPATIAL] INDEX [ [ IF NOT EXISTS ] newIndexName ]
| PRIMARY KEY [ HASH ] } | PRIMARY KEY [ HASH ] }
ON tableName ( indexColumn [,...] ) ON tableName ( indexColumn [,...] )
"," ","
...@@ -500,6 +500,7 @@ Hash indexes are meant for in-memory databases and memory tables (CREATE MEMORY ...@@ -500,6 +500,7 @@ Hash indexes are meant for in-memory databases and memory tables (CREATE MEMORY
For other tables, or if the index contains multiple columns, the HASH keyword is ignored. For other tables, or if the index contains multiple columns, the HASH keyword is ignored.
Hash indexes can only test for equality, and do not support range queries (similar to a hash table). Hash indexes can only test for equality, and do not support range queries (similar to a hash table).
Non-unique keys are supported. Non-unique keys are supported.
Spatial indexes are supported only on Geometry columns.
"," ","
CREATE INDEX IDXNAME ON TEST(NAME) CREATE INDEX IDXNAME ON TEST(NAME)
" "
...@@ -1726,9 +1727,9 @@ multiple lines long. ...@@ -1726,9 +1727,9 @@ multiple lines long.
" "
"Other Grammar","Compare"," "Other Grammar","Compare","
<> | <= | >= | = | < | > | != <> | <= | >= | = | < | > | != | &&
"," ","
Comparison operator. The operator != is the same as <>. Comparison operator. The operator != is the same as <>. && means overlapping, can only be used with geometry types.
"," ","
<> <>
" "
......
...@@ -80,6 +80,7 @@ Change Log ...@@ -80,6 +80,7 @@ Change Log
</li><li>Make org.h2.util.ScriptReader throw a better exception when handling broken scripts which generate </li><li>Make org.h2.util.ScriptReader throw a better exception when handling broken scripts which generate
extremely large statements. extremely large statements.
</li><li>Fix bug with ALLOW_LITERALS=NONE, where the periodic analyze table on insert would throw an exception. </li><li>Fix bug with ALLOW_LITERALS=NONE, where the periodic analyze table on insert would throw an exception.
</li><li>Issue 510: Make org.h2.bnf public for consumption by external projects, patch by Nicolas Fortin
</li></ul> </li></ul>
<h2>Version 1.3.173 (2013-07-28)</h2> <h2>Version 1.3.173 (2013-07-28)</h2>
......
...@@ -48,5 +48,7 @@ Export-Package: org.h2;version="${version}", ...@@ -48,5 +48,7 @@ Export-Package: org.h2;version="${version}",
org.h2.jdbcx;version="${version}", org.h2.jdbcx;version="${version}",
org.h2.tools;version="${version}", org.h2.tools;version="${version}",
org.h2.util;version="${version}", org.h2.util;version="${version}",
org.h2.value;version="${version}" org.h2.value;version="${version}",
org.h2.bnf;version="${version}" ,
org.h2.bnf.context;version="${version}"
Premain-Class: org.h2.util.Profiler Premain-Class: org.h2.util.Profiler
...@@ -15,7 +15,8 @@ import java.sql.SQLException; ...@@ -15,7 +15,8 @@ import java.sql.SQLException;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.HashMap; import java.util.HashMap;
import java.util.StringTokenizer; import java.util.StringTokenizer;
import org.h2.server.web.DbContextRule;
import org.h2.bnf.context.DbContextRule;
import org.h2.tools.Csv; import org.h2.tools.Csv;
import org.h2.util.New; import org.h2.util.New;
import org.h2.util.StringUtils; import org.h2.util.StringUtils;
......
...@@ -13,14 +13,14 @@ import org.h2.util.StringUtils; ...@@ -13,14 +13,14 @@ import org.h2.util.StringUtils;
/** /**
* A single terminal rule in a BNF object. * A single terminal rule in a BNF object.
*/ */
class RuleElement implements Rule { public class RuleElement implements Rule {
private final boolean keyword; private final boolean keyword;
private final String name; private final String name;
private Rule link; private Rule link;
private final int type; private final int type;
RuleElement(String name, String topic) { public RuleElement(String name, String topic) {
this.name = name; this.name = name;
this.keyword = name.length() == 1 || name.equals(StringUtils.toUpperEnglish(name)); this.keyword = name.length() == 1 || name.equals(StringUtils.toUpperEnglish(name));
topic = StringUtils.toLowerEnglish(topic); topic = StringUtils.toLowerEnglish(topic);
......
...@@ -13,13 +13,13 @@ import org.h2.util.New; ...@@ -13,13 +13,13 @@ import org.h2.util.New;
/** /**
* Represents a sequence of BNF rules, or a list of alternative rules. * Represents a sequence of BNF rules, or a list of alternative rules.
*/ */
class RuleList implements Rule { public class RuleList implements Rule {
private final boolean or; private final boolean or;
private final ArrayList<Rule> list; private final ArrayList<Rule> list;
private boolean mapSet; private boolean mapSet;
RuleList(Rule first, Rule next, boolean or) { public RuleList(Rule first, Rule next, boolean or) {
list = New.arrayList(); list = New.arrayList();
if (first instanceof RuleList && ((RuleList) first).or == or) { if (first instanceof RuleList && ((RuleList) first).or == or) {
list.addAll(((RuleList) first).list); list.addAll(((RuleList) first).list);
......
...@@ -11,11 +11,11 @@ import java.util.HashMap; ...@@ -11,11 +11,11 @@ import java.util.HashMap;
/** /**
* Represents an optional BNF rule. * Represents an optional BNF rule.
*/ */
class RuleOptional implements Rule { public class RuleOptional implements Rule {
private final Rule rule; private final Rule rule;
private boolean mapSet; private boolean mapSet;
RuleOptional(Rule rule) { public RuleOptional(Rule rule) {
this.rule = rule; this.rule = rule;
} }
......
...@@ -11,12 +11,12 @@ import java.util.HashMap; ...@@ -11,12 +11,12 @@ import java.util.HashMap;
/** /**
* Represents a loop in a BNF object. * Represents a loop in a BNF object.
*/ */
class RuleRepeat implements Rule { public class RuleRepeat implements Rule {
private final Rule rule; private final Rule rule;
private final boolean comma; private final boolean comma;
RuleRepeat(Rule rule, boolean comma) { public RuleRepeat(Rule rule, boolean comma) {
this.rule = rule; this.rule = rule;
this.comma = comma; this.comma = comma;
} }
......
...@@ -9,8 +9,8 @@ package org.h2.bnf; ...@@ -9,8 +9,8 @@ package org.h2.bnf;
import java.util.HashMap; import java.util.HashMap;
import java.util.HashSet; import java.util.HashSet;
import org.h2.server.web.DbSchema; import org.h2.bnf.context.DbSchema;
import org.h2.server.web.DbTableOrView; import org.h2.bnf.context.DbTableOrView;
import org.h2.util.New; import org.h2.util.New;
import org.h2.util.StringUtils; import org.h2.util.StringUtils;
...@@ -29,12 +29,12 @@ public class Sentence { ...@@ -29,12 +29,12 @@ public class Sentence {
/** /**
* The token type for a keyword. * The token type for a keyword.
*/ */
static final int KEYWORD = 1; public static final int KEYWORD = 1;
/** /**
* The token type for a function name. * The token type for a function name.
*/ */
static final int FUNCTION = 2; public static final int FUNCTION = 2;
private static final long MAX_PROCESSING_TIME = 100; private static final long MAX_PROCESSING_TIME = 100;
...@@ -63,7 +63,7 @@ public class Sentence { ...@@ -63,7 +63,7 @@ public class Sentence {
/** /**
* Start the timer to make sure processing doesn't take too long. * Start the timer to make sure processing doesn't take too long.
*/ */
void start() { public void start() {
stopAt = System.currentTimeMillis() + MAX_PROCESSING_TIME; stopAt = System.currentTimeMillis() + MAX_PROCESSING_TIME;
} }
...@@ -72,7 +72,7 @@ public class Sentence { ...@@ -72,7 +72,7 @@ public class Sentence {
* Processing auto-complete shouldn't take more than a few milliseconds. * Processing auto-complete shouldn't take more than a few milliseconds.
* If processing is stopped, this methods throws an IllegalStateException * If processing is stopped, this methods throws an IllegalStateException
*/ */
void stopIfRequired() { public void stopIfRequired() {
if (System.currentTimeMillis() > stopAt) { if (System.currentTimeMillis() > stopAt) {
throw new IllegalStateException(); throw new IllegalStateException();
} }
...@@ -214,7 +214,7 @@ public class Sentence { ...@@ -214,7 +214,7 @@ public class Sentence {
* *
* @return the next token map * @return the next token map
*/ */
HashMap<String, String> getNext() { public HashMap<String, String> getNext() {
return next; return next;
} }
......
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
* (http://h2database.com/html/license.html). * (http://h2database.com/html/license.html).
* Initial Developer: H2 Group * Initial Developer: H2 Group
*/ */
package org.h2.server.web; package org.h2.bnf.context;
import java.sql.DatabaseMetaData; import java.sql.DatabaseMetaData;
import java.sql.ResultSet; import java.sql.ResultSet;
...@@ -14,30 +14,59 @@ import java.sql.SQLException; ...@@ -14,30 +14,59 @@ import java.sql.SQLException;
* Keeps the meta data information of a column. * Keeps the meta data information of a column.
* This class is used by the H2 Console. * This class is used by the H2 Console.
*/ */
class DbColumn { public class DbColumn {
private final String name;
private final String quotedName;
private final String dataType;
private int position;
/**
* @return The data type name (including precision and the NOT NULL flag if
* applicable).
*/
public String getDataType() {
return dataType;
}
/** /**
* The column name. * @return The column name.
*/ */
final String name; public String getName() {
return name;
}
/** /**
* The quoted table name. * @return The quoted table name.
*/ */
final String quotedName; public String getQuotedName() {
return quotedName;
}
/** /**
* The data type name (including precision and the NOT NULL flag if * @return Column index
* applicable).
*/ */
final String dataType; public int getPosition() {
return position;
}
DbColumn(DbContents contents, ResultSet rs) throws SQLException { public DbColumn(DbContents contents, ResultSet rs) throws SQLException {
name = rs.getString("COLUMN_NAME"); name = rs.getString("COLUMN_NAME");
quotedName = contents.quoteIdentifier(name); quotedName = contents.quoteIdentifier(name);
String type = rs.getString("TYPE_NAME"); String type = rs.getString("TYPE_NAME");
int size = rs.getInt(DbContents.findColumn(rs, "COLUMN_SIZE", 7)); // A procedures column size is identified by PRECISION, for table this is COLUMN_SIZE
boolean isSQLite = contents.isSQLite; int precisionColumnIndex = DbContents.findColumn(rs, "PRECISION", 0);
int size;
if( precisionColumnIndex == 0 ){
size = rs.getInt(DbContents.findColumn(rs, "COLUMN_SIZE", 7));
} else {
size = rs.getInt(precisionColumnIndex);
}
position = rs.getInt("ORDINAL_POSITION");
boolean isSQLite = contents.isSQLite();
if (size > 0 && !isSQLite) { if (size > 0 && !isSQLite) {
type += "(" + size; type += "(" + size;
int prec = rs.getInt(DbContents.findColumn(rs, "DECIMAL_DIGITS", 9)); int prec = rs.getInt(DbContents.findColumn(rs, "DECIMAL_DIGITS", 9));
......
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
* (http://h2database.com/html/license.html). * (http://h2database.com/html/license.html).
* Initial Developer: H2 Group * Initial Developer: H2 Group
*/ */
package org.h2.server.web; package org.h2.bnf.context;
import java.sql.DatabaseMetaData; import java.sql.DatabaseMetaData;
import java.sql.ResultSet; import java.sql.ResultSet;
...@@ -22,60 +22,94 @@ import org.h2.util.StringUtils; ...@@ -22,60 +22,94 @@ import org.h2.util.StringUtils;
*/ */
public class DbContents { public class DbContents {
private DbSchema[] schemas;
private DbSchema defaultSchema;
private boolean isOracle;
private boolean isH2;
private boolean isPostgreSQL;
private boolean isDerby;
private boolean isSQLite;
private boolean isH2ModeMySQL;
private boolean isMySQL;
private boolean isFirebird;
private boolean isMSSQLServer;
/** /**
* The list of schemas. * @return The default schema.
*/ */
DbSchema[] schemas; public DbSchema getDefaultSchema() {
return defaultSchema;
}
/** /**
* The default schema. * @return True if this is an Apache Derby database.
*/ */
DbSchema defaultSchema; public boolean isDerby() {
return isDerby;
}
/** /**
* True if this is an Oracle database. * @return True if this is a Firebird database.
*/ */
boolean isOracle; public boolean isFirebird() {
return isFirebird;
}
/** /**
* True if this is a H2 database. * @return True if this is a H2 database.
*/ */
boolean isH2; public boolean isH2() {
return isH2;
}
/** /**
* True if this is a PostgreSQL database. * @return True if this is a H2 database in MySQL mode.
*/ */
boolean isPostgreSQL; public boolean isH2ModeMySQL() {
return isH2ModeMySQL;
}
/** /**
* True if this is an Apache Derby database. * @return True if this is a MS SQL Server database.
*/ */
boolean isDerby; public boolean isMSSQLServer() {
return isMSSQLServer;
}
/** /**
* True if this is an SQLite database. * @return True if this is a MySQL database.
*/ */
boolean isSQLite; public boolean isMySQL() {
return isMySQL;
}
/** /**
* True if this is a H2 database in MySQL mode. * @return True if this is an Oracle database.
*/ */
private boolean isH2ModeMySQL; public boolean isOracle() {
return isOracle;
}
/** /**
* True if this is a MySQL database. * @return True if this is a PostgreSQL database.
*/ */
private boolean isMySQL; public boolean isPostgreSQL() {
return isPostgreSQL;
}
/** /**
* True if this is a Firebird database. * @return True if this is an SQLite database.
*/ */
private boolean isFirebird; public boolean isSQLite() {
return isSQLite;
}
/** /**
* True if this is a MS SQL Server database. * @return The list of schemas.
*/ */
private boolean isMSSQLServer; public DbSchema[] getSchemas() {
return schemas;
}
/** /**
* Get the column index of a column in a result set. If the column is not * Get the column index of a column in a result set. If the column is not
...@@ -87,7 +121,7 @@ public class DbContents { ...@@ -87,7 +121,7 @@ public class DbContents {
* @param defaultColumnIndex the default column index * @param defaultColumnIndex the default column index
* @return the column index * @return the column index
*/ */
static int findColumn(ResultSet rs, String columnName, int defaultColumnIndex) { public static int findColumn(ResultSet rs, String columnName, int defaultColumnIndex) {
try { try {
return rs.findColumn(columnName); return rs.findColumn(columnName);
} catch (SQLException e) { } catch (SQLException e) {
...@@ -100,7 +134,7 @@ public class DbContents { ...@@ -100,7 +134,7 @@ public class DbContents {
* *
* @param meta the database meta data * @param meta the database meta data
*/ */
synchronized void readContents(DatabaseMetaData meta) throws SQLException { public synchronized void readContents(DatabaseMetaData meta) throws SQLException {
String prod = StringUtils.toLowerEnglish(meta.getDatabaseProductName()); String prod = StringUtils.toLowerEnglish(meta.getDatabaseProductName());
isSQLite = prod.indexOf("sqlite") >= 0; isSQLite = prod.indexOf("sqlite") >= 0;
String url = meta.getURL(); String url = meta.getURL();
...@@ -139,6 +173,7 @@ public class DbContents { ...@@ -139,6 +173,7 @@ public class DbContents {
String[] tableTypes = { "TABLE", "SYSTEM TABLE", "VIEW", "SYSTEM VIEW", String[] tableTypes = { "TABLE", "SYSTEM TABLE", "VIEW", "SYSTEM VIEW",
"TABLE LINK", "SYNONYM", "EXTERNAL" }; "TABLE LINK", "SYNONYM", "EXTERNAL" };
schema.readTables(meta, tableTypes); schema.readTables(meta, tableTypes);
schema.readProcedures(meta);
} }
if (defaultSchema == null) { if (defaultSchema == null) {
String best = null; String best = null;
...@@ -233,7 +268,7 @@ public class DbContents { ...@@ -233,7 +268,7 @@ public class DbContents {
* @param identifier the identifier * @param identifier the identifier
* @return the quoted identifier * @return the quoted identifier
*/ */
String quoteIdentifier(String identifier) { public String quoteIdentifier(String identifier) {
if (identifier == null) { if (identifier == null) {
return null; return null;
} }
......
...@@ -4,13 +4,15 @@ ...@@ -4,13 +4,15 @@
* (http://h2database.com/html/license.html). * (http://h2database.com/html/license.html).
* Initial Developer: H2 Group * Initial Developer: H2 Group
*/ */
package org.h2.server.web; package org.h2.bnf.context;
import java.util.HashMap; import java.util.HashMap;
import java.util.HashSet; import java.util.HashSet;
import org.h2.bnf.BnfVisitor; import org.h2.bnf.BnfVisitor;
import org.h2.bnf.Rule; import org.h2.bnf.Rule;
import org.h2.bnf.RuleElement;
import org.h2.bnf.RuleHead; import org.h2.bnf.RuleHead;
import org.h2.bnf.RuleList;
import org.h2.bnf.Sentence; import org.h2.bnf.Sentence;
import org.h2.command.Parser; import org.h2.command.Parser;
import org.h2.message.DbException; import org.h2.message.DbException;
...@@ -22,18 +24,32 @@ import org.h2.util.StringUtils; ...@@ -22,18 +24,32 @@ import org.h2.util.StringUtils;
*/ */
public class DbContextRule implements Rule { public class DbContextRule implements Rule {
static final int COLUMN = 0, TABLE = 1, TABLE_ALIAS = 2; public static final int COLUMN = 0, TABLE = 1, TABLE_ALIAS = 2;
static final int NEW_TABLE_ALIAS = 3; public static final int NEW_TABLE_ALIAS = 3;
static final int COLUMN_ALIAS = 4, SCHEMA = 5; public static final int COLUMN_ALIAS = 4, SCHEMA = 5, PROCEDURE = 6;
private final DbContents contents; private final DbContents contents;
private final int type; private final int type;
DbContextRule(DbContents contents, int type) { private String columnType;
/**
* BNF terminal rule Constructor
* @param contents Extract rule from this component
* @param type Rule type, one of {@link DbContextRule#COLUMN,DbContextRule#TABLE,DbContextRule#TABLE_ALIAS,DbContextRule#NEW_TABLE_ALIAS,DbContextRule#COLUMN_ALIAS,DbContextRule#SCHEMA}
*/
public DbContextRule(DbContents contents, int type) {
this.contents = contents; this.contents = contents;
this.type = type; this.type = type;
} }
/**
* @param columnType COLUMN Auto completion can be filtered by column type
*/
public void setColumnType(String columnType) {
this.columnType = columnType;
}
@Override @Override
public void setLinks(HashMap<String, RuleHead> ruleMap) { public void setLinks(HashMap<String, RuleHead> ruleMap) {
// nothing to do // nothing to do
...@@ -50,7 +66,7 @@ public class DbContextRule implements Rule { ...@@ -50,7 +66,7 @@ public class DbContextRule implements Rule {
String up = sentence.getQueryUpper(); String up = sentence.getQueryUpper();
switch (type) { switch (type) {
case SCHEMA: { case SCHEMA: {
DbSchema[] schemas = contents.schemas; DbSchema[] schemas = contents.getSchemas();
String best = null; String best = null;
DbSchema bestSchema = null; DbSchema bestSchema = null;
for (DbSchema schema: schemas) { for (DbSchema schema: schemas) {
...@@ -76,16 +92,16 @@ public class DbContextRule implements Rule { ...@@ -76,16 +92,16 @@ public class DbContextRule implements Rule {
case TABLE: { case TABLE: {
DbSchema schema = sentence.getLastMatchedSchema(); DbSchema schema = sentence.getLastMatchedSchema();
if (schema == null) { if (schema == null) {
schema = contents.defaultSchema; schema = contents.getDefaultSchema();
} }
DbTableOrView[] tables = schema.tables; DbTableOrView[] tables = schema.getTables();
String best = null; String best = null;
DbTableOrView bestTable = null; DbTableOrView bestTable = null;
for (DbTableOrView table : tables) { for (DbTableOrView table : tables) {
String compare = up; String compare = up;
String name = StringUtils.toUpperEnglish(table.name); String name = StringUtils.toUpperEnglish(table.getName());
if (table.quotedName.length() > name.length()) { if (table.getQuotedName().length() > name.length()) {
name = table.quotedName; name = table.getQuotedName();
compare = query; compare = query;
} }
if (compare.startsWith(name)) { if (compare.startsWith(name)) {
...@@ -95,7 +111,7 @@ public class DbContextRule implements Rule { ...@@ -95,7 +111,7 @@ public class DbContextRule implements Rule {
} }
} else if (s.length() == 0 || name.startsWith(compare)) { } else if (s.length() == 0 || name.startsWith(compare)) {
if (s.length() < name.length()) { if (s.length() < name.length()) {
sentence.add(table.quotedName, table.quotedName.substring(s.length()), Sentence.CONTEXT); sentence.add(table.getQuotedName(), table.getQuotedName().substring(s.length()), Sentence.CONTEXT);
} }
} }
} }
...@@ -137,36 +153,37 @@ public class DbContextRule implements Rule { ...@@ -137,36 +153,37 @@ public class DbContextRule implements Rule {
HashSet<DbTableOrView> set = sentence.getTables(); HashSet<DbTableOrView> set = sentence.getTables();
String best = null; String best = null;
DbTableOrView last = sentence.getLastMatchedTable(); DbTableOrView last = sentence.getLastMatchedTable();
if (last != null && last.columns != null) { if (last != null && last.getColumns() != null) {
for (DbColumn column : last.columns) { for (DbColumn column : last.getColumns()) {
String compare = up; String compare = up;
String name = StringUtils.toUpperEnglish(column.name); String name = StringUtils.toUpperEnglish(column.getName());
if (column.quotedName.length() > name.length()) { if (column.getQuotedName().length() > name.length()) {
name = column.quotedName; name = column.getQuotedName();
compare = query; compare = query;
} }
if (compare.startsWith(name)) { if (compare.startsWith(name) && (columnType == null || column.getDataType().contains(columnType))) {
String b = s.substring(name.length()); String b = s.substring(name.length());
if (best == null || b.length() < best.length()) { if (best == null || b.length() < best.length()) {
best = b; best = b;
} else if (s.length() == 0 || name.startsWith(compare)) { } else if (s.length() == 0 || name.startsWith(compare)) {
if (s.length() < name.length()) { if (s.length() < name.length()) {
sentence.add(column.name, column.name.substring(s.length()), Sentence.CONTEXT); sentence.add(column.getName(), column.getName().substring(s.length()), Sentence.CONTEXT);
} }
} }
} }
} }
} }
for (DbSchema schema : contents.schemas) { for (DbSchema schema : contents.getSchemas()) {
for (DbTableOrView table : schema.tables) { for (DbTableOrView table : schema.getTables()) {
if (table != last && set != null && !set.contains(table)) { if (table != last && set != null && !set.contains(table)) {
continue; continue;
} }
if (table == null || table.columns == null) { if (table == null || table.getColumns() == null) {
continue; continue;
} }
for (DbColumn column : table.columns) { for (DbColumn column : table.getColumns()) {
String name = StringUtils.toUpperEnglish(column.name); String name = StringUtils.toUpperEnglish(column.getName());
if(columnType == null || column.getDataType().contains(columnType)) {
if (up.startsWith(name)) { if (up.startsWith(name)) {
String b = s.substring(name.length()); String b = s.substring(name.length());
if (best == null || b.length() < best.length()) { if (best == null || b.length() < best.length()) {
...@@ -174,7 +191,8 @@ public class DbContextRule implements Rule { ...@@ -174,7 +191,8 @@ public class DbContextRule implements Rule {
} }
} else if (s.length() == 0 || name.startsWith(up)) { } else if (s.length() == 0 || name.startsWith(up)) {
if (s.length() < name.length()) { if (s.length() < name.length()) {
sentence.add(column.name, column.name.substring(s.length()), Sentence.CONTEXT); sentence.add(column.getName(), column.getName().substring(s.length()), Sentence.CONTEXT);
}
} }
} }
} }
...@@ -185,6 +203,9 @@ public class DbContextRule implements Rule { ...@@ -185,6 +203,9 @@ public class DbContextRule implements Rule {
} }
break; break;
} }
case PROCEDURE:
autoCompleteProcedure(sentence);
break;
default: default:
throw DbException.throwInternalError("type=" + type); throw DbException.throwInternalError("type=" + type);
} }
...@@ -197,7 +218,50 @@ public class DbContextRule implements Rule { ...@@ -197,7 +218,50 @@ public class DbContextRule implements Rule {
} }
return false; return false;
} }
private void autoCompleteProcedure(Sentence sentence) {
DbSchema schema = sentence.getLastMatchedSchema();
if (schema == null) {
schema = contents.getDefaultSchema();
}
String incompleteSentence = sentence.getQueryUpper();
String incompleteFunctionName = incompleteSentence;
if(incompleteSentence.contains("(")) {
incompleteFunctionName = incompleteSentence.substring(0, incompleteSentence.indexOf('(')).trim();
}
// Common elements
RuleElement openBracket = new RuleElement("(", "Function");
RuleElement closeBracket = new RuleElement(")", "Function");
RuleElement comma = new RuleElement(",", "Function");
// Fetch all elements
for(DbProcedure procedure : schema.getProcedures()) {
final String procName = procedure.getName();
if(procName.startsWith(incompleteFunctionName)) {
// That's it, build a RuleList from this function
RuleElement procedureElement = new RuleElement(procName, "Function");
RuleList rl = new RuleList(procedureElement, openBracket, false);
// Go further only if the user use open bracket
if(incompleteSentence.contains("(")) {
for(DbColumn parameter : procedure.getParameters()) {
if(parameter.getPosition() > 1) {
rl = new RuleList(rl, comma, false);
}
DbContextRule columnRule = new DbContextRule(contents, COLUMN);
String parameterType = parameter.getDataType();
// Remove precision
if(parameterType.contains("(")) {
parameterType = parameterType.substring(0, parameterType.indexOf('('));
}
columnRule.setColumnType(parameterType);
rl = new RuleList(rl, columnRule , false);
}
rl = new RuleList(rl, closeBracket , false);
}
rl.autoComplete(sentence);
}
}
}
private static String autoCompleteTableAlias(Sentence sentence, boolean newAlias) { private static String autoCompleteTableAlias(Sentence sentence, boolean newAlias) {
String s = sentence.getQuery(); String s = sentence.getQuery();
String up = sentence.getQueryUpper(); String up = sentence.getQueryUpper();
...@@ -233,7 +297,7 @@ public class DbContextRule implements Rule { ...@@ -233,7 +297,7 @@ public class DbContextRule implements Rule {
if (tables != null) { if (tables != null) {
String best = null; String best = null;
for (DbTableOrView table : tables) { for (DbTableOrView table : tables) {
String tableName = StringUtils.toUpperEnglish(table.name); String tableName = StringUtils.toUpperEnglish(table.getName());
if (alias.startsWith(tableName) && (best == null || tableName.length() > best.length())) { if (alias.startsWith(tableName) && (best == null || tableName.length() > best.length())) {
sentence.setLastMatchedTable(table); sentence.setLastMatchedTable(table);
best = tableName; best = tableName;
......
/*
* Copyright 2004-2013 H2 Group. Multiple-Licensed under the H2 License,
* Version 1.0, and under the Eclipse Public License, Version 1.0
* (http://h2database.com/html/license.html).
* Initial Developer: H2 Group
*/
package org.h2.bnf.context;
import org.h2.util.New;
import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
/**
* Contains meta data information about a procedure.
* This class is used by the H2 Console.
*/
public class DbProcedure {
private final DbSchema schema;
private final String name;
private final String quotedName;
private boolean returnsResult;
private DbColumn[] parameters;
public DbProcedure(DbSchema schema, ResultSet rs) throws SQLException {
this.schema = schema;
name = rs.getString("PROCEDURE_NAME");
returnsResult = rs.getShort("PROCEDURE_TYPE") == DatabaseMetaData.procedureReturnsResult;
quotedName = schema.getContents().quoteIdentifier(name);
}
/**
* @return The schema this table belongs to.
*/
public DbSchema getSchema() {
return schema;
}
/**
* @return The column list.
*/
public DbColumn[] getParameters() {
return parameters;
}
/**
* @return The table name.
*/
public String getName() {
return name;
}
/**
* @return The quoted table name.
*/
public String getQuotedName() {
return quotedName;
}
/**
* @return True if this function return a value
*/
public boolean isReturnsResult() {
return returnsResult;
}
/**
* Read the column for this table from the database meta data.
*
* @param meta the database meta data
*/
void readParameters(DatabaseMetaData meta) throws SQLException {
ResultSet rs = meta.getProcedureColumns(null, schema.name, name, null);
ArrayList<DbColumn> list = New.arrayList();
while (rs.next()) {
DbColumn column = new DbColumn(schema.getContents(), rs);
if(column.getPosition()>0) { //Not the return type
list.add(column);
}
}
rs.close();
parameters = new DbColumn[list.size()];
// Store the parameter in the good position [1-n]
for(int i=0;i<parameters.length;i++) {
DbColumn column = list.get(i);
if(column.getPosition()>0 && column.getPosition() <= parameters.length) {
parameters[column.getPosition() - 1] = column;
}
}
}
}
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
* (http://h2database.com/html/license.html). * (http://h2database.com/html/license.html).
* Initial Developer: H2 Group * Initial Developer: H2 Group
*/ */
package org.h2.server.web; package org.h2.bnf.context;
import java.sql.DatabaseMetaData; import java.sql.DatabaseMetaData;
import java.sql.ResultSet; import java.sql.ResultSet;
...@@ -22,42 +22,52 @@ public class DbSchema { ...@@ -22,42 +22,52 @@ public class DbSchema {
/** /**
* Up to this many tables, the column type and indexes are listed. * Up to this many tables, the column type and indexes are listed.
*/ */
static final int MAX_TABLES_LIST_INDEXES = 100; public static final int MAX_TABLES_LIST_INDEXES = 100;
/** /**
* Up to this many tables, the column names are listed. * Up to this many tables, the column names are listed.
*/ */
static final int MAX_TABLES_LIST_COLUMNS = 500; public static final int MAX_TABLES_LIST_COLUMNS = 500;
/**
* Up to this many tables, the column names are listed.
*/
public static final int MAX_PROCEDURES_LIST_COLUMNS = 500;
/** /**
* The database content container. * The database content container.
*/ */
final DbContents contents; private final DbContents contents;
/** /**
* The schema name. * The schema name.
*/ */
final String name; public final String name;
/** /**
* True if this is the default schema for this database. * True if this is the default schema for this database.
*/ */
final boolean isDefault; public final boolean isDefault;
/** /**
* True if this is a system schema (for example the INFORMATION_SCHEMA). * True if this is a system schema (for example the INFORMATION_SCHEMA).
*/ */
final boolean isSystem; public final boolean isSystem;
/** /**
* The quoted schema name. * The quoted schema name.
*/ */
final String quotedName; public final String quotedName;
/** /**
* The table list. * The table list.
*/ */
DbTableOrView[] tables; private DbTableOrView[] tables;
/**
* The procedures list.
*/
private DbProcedure[] procedures;
DbSchema(DbContents contents, String name, boolean isDefault) { DbSchema(DbContents contents, String name, boolean isDefault) {
this.contents = contents; this.contents = contents;
...@@ -66,29 +76,50 @@ public class DbSchema { ...@@ -66,29 +76,50 @@ public class DbSchema {
this.isDefault = isDefault; this.isDefault = isDefault;
if (name.equals("INFORMATION_SCHEMA")) { if (name.equals("INFORMATION_SCHEMA")) {
isSystem = true; isSystem = true;
} else if (!contents.isH2 && StringUtils.toUpperEnglish(name).startsWith("INFO")) { } else if (!contents.isH2() && StringUtils.toUpperEnglish(name).startsWith("INFO")) {
isSystem = true; isSystem = true;
} else if (contents.isPostgreSQL && StringUtils.toUpperEnglish(name).startsWith("PG_")) { } else if (contents.isPostgreSQL() && StringUtils.toUpperEnglish(name).startsWith("PG_")) {
isSystem = true; isSystem = true;
} else if (contents.isDerby && name.startsWith("SYS")) { } else if (contents.isDerby() && name.startsWith("SYS")) {
isSystem = true; isSystem = true;
} else { } else {
isSystem = false; isSystem = false;
} }
} }
/**
* @return The database content container.
*/
public DbContents getContents() {
return contents;
}
/**
* @return The table list.
*/
public DbTableOrView[] getTables() {
return tables;
}
/**
* @return The procedure list.
*/
public DbProcedure[] getProcedures() {
return procedures;
}
/** /**
* Read all tables for this schema from the database meta data. * Read all tables for this schema from the database meta data.
* *
* @param meta the database meta data * @param meta the database meta data
* @param tableTypes the table types to read * @param tableTypes the table types to read
*/ */
void readTables(DatabaseMetaData meta, String[] tableTypes) throws SQLException { public void readTables(DatabaseMetaData meta, String[] tableTypes) throws SQLException {
ResultSet rs = meta.getTables(null, name, null, tableTypes); ResultSet rs = meta.getTables(null, name, null, tableTypes);
ArrayList<DbTableOrView> list = New.arrayList(); ArrayList<DbTableOrView> list = New.arrayList();
while (rs.next()) { while (rs.next()) {
DbTableOrView table = new DbTableOrView(this, rs); DbTableOrView table = new DbTableOrView(this, rs);
if (contents.isOracle && table.name.indexOf('$') > 0) { if (contents.isOracle() && table.getName().indexOf('$') > 0) {
continue; continue;
} }
list.add(table); list.add(table);
...@@ -103,4 +134,24 @@ public class DbSchema { ...@@ -103,4 +134,24 @@ public class DbSchema {
} }
} }
/**
* Read all procedures in the dataBase.
* @param meta the database meta data
* @throws SQLException Error while fetching procedures
*/
public void readProcedures(DatabaseMetaData meta) throws SQLException {
ResultSet rs = meta.getProcedures(null, name, null);
ArrayList<DbProcedure> list = New.arrayList();
while (rs.next()) {
list.add(new DbProcedure(this, rs));
}
rs.close();
procedures = new DbProcedure[list.size()];
list.toArray(procedures);
if (procedures.length < MAX_PROCEDURES_LIST_COLUMNS) {
for (DbProcedure procedure : procedures) {
procedure.readParameters(meta);
}
}
}
} }
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
* (http://h2database.com/html/license.html). * (http://h2database.com/html/license.html).
* Initial Developer: H2 Group * Initial Developer: H2 Group
*/ */
package org.h2.server.web; package org.h2.bnf.context;
import java.sql.DatabaseMetaData; import java.sql.DatabaseMetaData;
import java.sql.ResultSet; import java.sql.ResultSet;
...@@ -21,34 +21,69 @@ public class DbTableOrView { ...@@ -21,34 +21,69 @@ public class DbTableOrView {
/** /**
* The schema this table belongs to. * The schema this table belongs to.
*/ */
final DbSchema schema; private final DbSchema schema;
/** /**
* The table name. * The table name.
*/ */
final String name; private final String name;
/** /**
* The quoted table name. * The quoted table name.
*/ */
final String quotedName; private final String quotedName;
/** /**
* True if this represents a view. * True if this represents a view.
*/ */
final boolean isView; private final boolean isView;
/** /**
* The column list. * The column list.
*/ */
DbColumn[] columns; private DbColumn[] columns;
DbTableOrView(DbSchema schema, ResultSet rs) throws SQLException { public DbTableOrView(DbSchema schema, ResultSet rs) throws SQLException {
this.schema = schema; this.schema = schema;
name = rs.getString("TABLE_NAME"); name = rs.getString("TABLE_NAME");
String type = rs.getString("TABLE_TYPE"); String type = rs.getString("TABLE_TYPE");
isView = "VIEW".equals(type); isView = "VIEW".equals(type);
quotedName = schema.contents.quoteIdentifier(name); quotedName = schema.getContents().quoteIdentifier(name);
}
/**
* @return The schema this table belongs to.
*/
public DbSchema getSchema() {
return schema;
}
/**
* @return The column list.
*/
public DbColumn[] getColumns() {
return columns;
}
/**
* @return The table name.
*/
public String getName() {
return name;
}
/**
* @return True if this represents a view.
*/
public boolean isView() {
return isView;
}
/**
* @return The quoted table name.
*/
public String getQuotedName() {
return quotedName;
} }
/** /**
...@@ -56,11 +91,11 @@ public class DbTableOrView { ...@@ -56,11 +91,11 @@ public class DbTableOrView {
* *
* @param meta the database meta data * @param meta the database meta data
*/ */
void readColumns(DatabaseMetaData meta) throws SQLException { public void readColumns(DatabaseMetaData meta) throws SQLException {
ResultSet rs = meta.getColumns(null, schema.name, name, null); ResultSet rs = meta.getColumns(null, schema.name, name, null);
ArrayList<DbColumn> list = New.arrayList(); ArrayList<DbColumn> list = New.arrayList();
while (rs.next()) { while (rs.next()) {
DbColumn column = new DbColumn(schema.contents, rs); DbColumn column = new DbColumn(schema.getContents(), rs);
list.add(column); list.add(column);
} }
rs.close(); rs.close();
......
...@@ -167,7 +167,7 @@ CREATE DOMAIN [ IF NOT EXISTS ] newDomainName AS dataType ...@@ -167,7 +167,7 @@ CREATE DOMAIN [ IF NOT EXISTS ] newDomainName AS dataType
Creates a new data type (domain)." Creates a new data type (domain)."
"Commands (DDL)","CREATE INDEX"," "Commands (DDL)","CREATE INDEX","
CREATE CREATE
{ [ UNIQUE ] [ HASH ] INDEX [ [ IF NOT EXISTS ] newIndexName ] { [ UNIQUE ] [ HASH ] [ SPATIAL] INDEX [ [ IF NOT EXISTS ] newIndexName ]
| PRIMARY KEY [ HASH ] } | PRIMARY KEY [ HASH ] }
ON tableName ( indexColumn [,...] ) ON tableName ( indexColumn [,...] )
"," ","
...@@ -541,7 +541,7 @@ Default expressions are used if no explicit value was used when adding a row." ...@@ -541,7 +541,7 @@ Default expressions are used if no explicit value was used when adding a row."
"," ","
Comments can be used anywhere in a command and are ignored by the database." Comments can be used anywhere in a command and are ignored by the database."
"Other Grammar","Compare"," "Other Grammar","Compare","
<> | <= | >= | = | < | > | != <> | <= | >= | = | < | > | != | &&
"," ","
Comparison operator." Comparison operator."
"Other Grammar","Condition"," "Other Grammar","Condition","
......
...@@ -34,6 +34,10 @@ import java.util.Map; ...@@ -34,6 +34,10 @@ import java.util.Map;
import java.util.Properties; import java.util.Properties;
import java.util.Random; import java.util.Random;
import org.h2.bnf.Bnf; import org.h2.bnf.Bnf;
import org.h2.bnf.context.DbColumn;
import org.h2.bnf.context.DbContents;
import org.h2.bnf.context.DbSchema;
import org.h2.bnf.context.DbTableOrView;
import org.h2.constant.ErrorCode; import org.h2.constant.ErrorCode;
import org.h2.constant.SysProperties; import org.h2.constant.SysProperties;
import org.h2.engine.Constants; import org.h2.engine.Constants;
...@@ -469,20 +473,20 @@ public class WebApp { ...@@ -469,20 +473,20 @@ public class WebApp {
private static int addColumns(boolean mainSchema, DbTableOrView table, private static int addColumns(boolean mainSchema, DbTableOrView table,
StringBuilder buff, int treeIndex, boolean showColumnTypes, StringBuilder buff, int treeIndex, boolean showColumnTypes,
StringBuilder columnsBuffer) { StringBuilder columnsBuffer) {
DbColumn[] columns = table.columns; DbColumn[] columns = table.getColumns();
for (int i = 0; columns != null && i < columns.length; i++) { for (int i = 0; columns != null && i < columns.length; i++) {
DbColumn column = columns[i]; DbColumn column = columns[i];
if (columnsBuffer.length() > 0) { if (columnsBuffer.length() > 0) {
columnsBuffer.append(' '); columnsBuffer.append(' ');
} }
columnsBuffer.append(column.name); columnsBuffer.append(column.getName());
String col = escapeIdentifier(column.name); String col = escapeIdentifier(column.getName());
String level = mainSchema ? ", 1, 1" : ", 2, 2"; String level = mainSchema ? ", 1, 1" : ", 2, 2";
buff.append("setNode(" + treeIndex + level + ", 'column', '" + PageParser.escapeJavaScript(column.name) buff.append("setNode(" + treeIndex + level + ", 'column', '" + PageParser.escapeJavaScript(column.getName())
+ "', 'javascript:ins(\\'" + col + "\\')');\n"); + "', 'javascript:ins(\\'" + col + "\\')');\n");
treeIndex++; treeIndex++;
if (mainSchema && showColumnTypes) { if (mainSchema && showColumnTypes) {
buff.append("setNode(" + treeIndex + ", 2, 2, 'type', '" + PageParser.escapeJavaScript(column.dataType) buff.append("setNode(" + treeIndex + ", 2, 2, 'type', '" + PageParser.escapeJavaScript(column.getDataType())
+ "', null);\n"); + "', null);\n");
treeIndex++; treeIndex++;
} }
...@@ -585,57 +589,57 @@ public class WebApp { ...@@ -585,57 +589,57 @@ public class WebApp {
boolean showColumns = mainSchema || !schema.isSystem; boolean showColumns = mainSchema || !schema.isSystem;
String indentation = ", " + level + ", " + (showColumns ? "1" : "2") + ", "; String indentation = ", " + level + ", " + (showColumns ? "1" : "2") + ", ";
String indentNode = ", " + (level + 1) + ", 2, "; String indentNode = ", " + (level + 1) + ", 2, ";
DbTableOrView[] tables = schema.tables; DbTableOrView[] tables = schema.getTables();
if (tables == null) { if (tables == null) {
return treeIndex; return treeIndex;
} }
boolean isOracle = schema.contents.isOracle; boolean isOracle = schema.getContents().isOracle();
boolean notManyTables = tables.length < DbSchema.MAX_TABLES_LIST_INDEXES; boolean notManyTables = tables.length < DbSchema.MAX_TABLES_LIST_INDEXES;
for (DbTableOrView table : tables) { for (DbTableOrView table : tables) {
if (table.isView) { if (table.isView()) {
continue; continue;
} }
int tableId = treeIndex; int tableId = treeIndex;
String tab = table.quotedName; String tab = table.getQuotedName();
if (!mainSchema) { if (!mainSchema) {
tab = schema.quotedName + "." + tab; tab = schema.quotedName + "." + tab;
} }
tab = escapeIdentifier(tab); tab = escapeIdentifier(tab);
buff.append("setNode(" + treeIndex + indentation + " 'table', '" + PageParser.escapeJavaScript(table.name) buff.append("setNode(" + treeIndex + indentation + " 'table', '" + PageParser.escapeJavaScript(table.getName())
+ "', 'javascript:ins(\\'" + tab + "\\',true)');\n"); + "', 'javascript:ins(\\'" + tab + "\\',true)');\n");
treeIndex++; treeIndex++;
if (mainSchema || showColumns) { if (mainSchema || showColumns) {
StringBuilder columnsBuffer = new StringBuilder(); StringBuilder columnsBuffer = new StringBuilder();
treeIndex = addColumns(mainSchema, table, buff, treeIndex, notManyTables, columnsBuffer); treeIndex = addColumns(mainSchema, table, buff, treeIndex, notManyTables, columnsBuffer);
if (!isOracle && notManyTables) { if (!isOracle && notManyTables) {
treeIndex = addIndexes(mainSchema, meta, table.name, schema.name, buff, treeIndex); treeIndex = addIndexes(mainSchema, meta, table.getName(), schema.name, buff, treeIndex);
} }
buff.append("addTable('" + PageParser.escapeJavaScript(table.name) + "', '" buff.append("addTable('" + PageParser.escapeJavaScript(table.getName()) + "', '"
+ PageParser.escapeJavaScript(columnsBuffer.toString()) + "', " + tableId + ");\n"); + PageParser.escapeJavaScript(columnsBuffer.toString()) + "', " + tableId + ");\n");
} }
} }
tables = schema.tables; tables = schema.getTables();
for (DbTableOrView view : tables) { for (DbTableOrView view : tables) {
if (!view.isView) { if (!view.isView()) {
continue; continue;
} }
int tableId = treeIndex; int tableId = treeIndex;
String tab = view.quotedName; String tab = view.getQuotedName();
if (!mainSchema) { if (!mainSchema) {
tab = view.schema.quotedName + "." + tab; tab = view.getSchema().quotedName + "." + tab;
} }
tab = escapeIdentifier(tab); tab = escapeIdentifier(tab);
buff.append("setNode(" + treeIndex + indentation + " 'view', '" + PageParser.escapeJavaScript(view.name) buff.append("setNode(" + treeIndex + indentation + " 'view', '" + PageParser.escapeJavaScript(view.getName())
+ "', 'javascript:ins(\\'" + tab + "\\',true)');\n"); + "', 'javascript:ins(\\'" + tab + "\\',true)');\n");
treeIndex++; treeIndex++;
if (mainSchema) { if (mainSchema) {
StringBuilder columnsBuffer = new StringBuilder(); StringBuilder columnsBuffer = new StringBuilder();
treeIndex = addColumns(mainSchema, view, buff, treeIndex, notManyTables, columnsBuffer); treeIndex = addColumns(mainSchema, view, buff, treeIndex, notManyTables, columnsBuffer);
if (schema.contents.isH2) { if (schema.getContents().isH2()) {
PreparedStatement prep = null; PreparedStatement prep = null;
try { try {
prep = conn.prepareStatement("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME=?"); prep = conn.prepareStatement("SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME=?");
prep.setString(1, view.name); prep.setString(1, view.getName());
ResultSet rs = prep.executeQuery(); ResultSet rs = prep.executeQuery();
if (rs.next()) { if (rs.next()) {
String sql = rs.getString("SQL"); String sql = rs.getString("SQL");
...@@ -648,7 +652,7 @@ public class WebApp { ...@@ -648,7 +652,7 @@ public class WebApp {
JdbcUtils.closeSilently(prep); JdbcUtils.closeSilently(prep);
} }
} }
buff.append("addTable('" + PageParser.escapeJavaScript(view.name) + "', '" buff.append("addTable('" + PageParser.escapeJavaScript(view.getName()) + "', '"
+ PageParser.escapeJavaScript(columnsBuffer.toString()) + "', " + tableId + ");\n"); + PageParser.escapeJavaScript(columnsBuffer.toString()) + "', " + tableId + ");\n");
} }
} }
...@@ -663,16 +667,16 @@ public class WebApp { ...@@ -663,16 +667,16 @@ public class WebApp {
session.loadBnf(); session.loadBnf();
Connection conn = session.getConnection(); Connection conn = session.getConnection();
DatabaseMetaData meta = session.getMetaData(); DatabaseMetaData meta = session.getMetaData();
isH2 = contents.isH2; isH2 = contents.isH2();
StringBuilder buff = new StringBuilder(); StringBuilder buff = new StringBuilder();
buff.append("setNode(0, 0, 0, 'database', '" + PageParser.escapeJavaScript((String) session.get("url")) buff.append("setNode(0, 0, 0, 'database', '" + PageParser.escapeJavaScript((String) session.get("url"))
+ "', null);\n"); + "', null);\n");
int treeIndex = 1; int treeIndex = 1;
DbSchema defaultSchema = contents.defaultSchema; DbSchema defaultSchema = contents.getDefaultSchema();
treeIndex = addTablesAndViews(defaultSchema, true, buff, treeIndex); treeIndex = addTablesAndViews(defaultSchema, true, buff, treeIndex);
DbSchema[] schemas = contents.schemas; DbSchema[] schemas = contents.getSchemas();
for (DbSchema schema : schemas) { for (DbSchema schema : schemas) {
if (schema == defaultSchema || schema == null) { if (schema == defaultSchema || schema == null) {
continue; continue;
...@@ -983,7 +987,7 @@ public class WebApp { ...@@ -983,7 +987,7 @@ public class WebApp {
result = buff.toString(); result = buff.toString();
session.put("result", result); session.put("result", result);
} catch (Throwable e) { } catch (Throwable e) {
session.put("result", getStackTrace(0, e, session.getContents().isH2)); session.put("result", getStackTrace(0, e, session.getContents().isH2()));
} }
return "result.jsp"; return "result.jsp";
} }
...@@ -1034,7 +1038,7 @@ public class WebApp { ...@@ -1034,7 +1038,7 @@ public class WebApp {
// cancel // cancel
} }
} catch (Throwable e) { } catch (Throwable e) {
result = "<br />" + getStackTrace(0, e, session.getContents().isH2); result = "<br />" + getStackTrace(0, e, session.getContents().isH2());
error = formatAsError(e.getMessage()); error = formatAsError(e.getMessage());
} }
String sql = "@edit " + (String) session.get("resultSetSQL"); String sql = "@edit " + (String) session.get("resultSetSQL");
...@@ -1212,7 +1216,7 @@ public class WebApp { ...@@ -1212,7 +1216,7 @@ public class WebApp {
} }
Statement stat; Statement stat;
DbContents contents = session.getContents(); DbContents contents = session.getContents();
if (forceEdit || (allowEdit && contents.isH2)) { if (forceEdit || (allowEdit && contents.isH2())) {
stat = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_UPDATABLE); stat = conn.createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_UPDATABLE);
} else { } else {
stat = conn.createStatement(); stat = conn.createStatement();
...@@ -1342,7 +1346,7 @@ public class WebApp { ...@@ -1342,7 +1346,7 @@ public class WebApp {
return buff.toString(); return buff.toString();
} catch (Throwable e) { } catch (Throwable e) {
// throwable: including OutOfMemoryError and so on // throwable: including OutOfMemoryError and so on
return getStackTrace(id, e, session.getContents().isH2); return getStackTrace(id, e, session.getContents().isH2());
} finally { } finally {
session.executingStatement = null; session.executingStatement = null;
} }
...@@ -1405,7 +1409,7 @@ public class WebApp { ...@@ -1405,7 +1409,7 @@ public class WebApp {
prep.setInt(j + 1, i); prep.setInt(j + 1, i);
} }
} }
if (session.getContents().isSQLite) { if (session.getContents().isSQLite()) {
// SQLite currently throws an exception on prep.execute() // SQLite currently throws an exception on prep.execute()
prep.executeUpdate(); prep.executeUpdate();
} else { } else {
...@@ -1746,7 +1750,7 @@ public class WebApp { ...@@ -1746,7 +1750,7 @@ public class WebApp {
} }
ResultSetMetaData meta = rs.getMetaData(); ResultSetMetaData meta = rs.getMetaData();
int type = meta.getColumnType(columnIndex); int type = meta.getColumnType(columnIndex);
if (session.getContents().isH2) { if (session.getContents().isH2()) {
rs.updateString(columnIndex, x); rs.updateString(columnIndex, x);
return; return;
} }
......
...@@ -17,6 +17,8 @@ import java.util.HashMap; ...@@ -17,6 +17,8 @@ import java.util.HashMap;
import java.util.Locale; import java.util.Locale;
import org.h2.bnf.Bnf; import org.h2.bnf.Bnf;
import org.h2.bnf.context.DbContents;
import org.h2.bnf.context.DbContextRule;
import org.h2.message.TraceSystem; import org.h2.message.TraceSystem;
import org.h2.util.New; import org.h2.util.New;
...@@ -120,12 +122,14 @@ class WebSession { ...@@ -120,12 +122,14 @@ class WebSession {
DbContextRule tableRule = new DbContextRule(contents, DbContextRule.TABLE); DbContextRule tableRule = new DbContextRule(contents, DbContextRule.TABLE);
DbContextRule schemaRule = new DbContextRule(contents, DbContextRule.SCHEMA); DbContextRule schemaRule = new DbContextRule(contents, DbContextRule.SCHEMA);
DbContextRule columnAliasRule = new DbContextRule(contents, DbContextRule.COLUMN_ALIAS); DbContextRule columnAliasRule = new DbContextRule(contents, DbContextRule.COLUMN_ALIAS);
DbContextRule procedureRule = new DbContextRule(contents, DbContextRule.PROCEDURE);
newBnf.updateTopic("column_name", columnRule); newBnf.updateTopic("column_name", columnRule);
newBnf.updateTopic("new_table_alias", newAliasRule); newBnf.updateTopic("new_table_alias", newAliasRule);
newBnf.updateTopic("table_alias", aliasRule); newBnf.updateTopic("table_alias", aliasRule);
newBnf.updateTopic("column_alias", columnAliasRule); newBnf.updateTopic("column_alias", columnAliasRule);
newBnf.updateTopic("table_name", tableRule); newBnf.updateTopic("table_name", tableRule);
newBnf.updateTopic("schema_name", schemaRule); newBnf.updateTopic("schema_name", schemaRule);
newBnf.updateTopic("expression", procedureRule);
newBnf.linkStatements(); newBnf.linkStatements();
bnf = newBnf; bnf = newBnf;
} catch (Exception e) { } catch (Exception e) {
......
...@@ -144,6 +144,7 @@ import org.h2.test.synth.sql.TestSynth; ...@@ -144,6 +144,7 @@ import org.h2.test.synth.sql.TestSynth;
import org.h2.test.synth.thread.TestMulti; import org.h2.test.synth.thread.TestMulti;
import org.h2.test.unit.TestAutoReconnect; import org.h2.test.unit.TestAutoReconnect;
import org.h2.test.unit.TestBitField; import org.h2.test.unit.TestBitField;
import org.h2.test.unit.TestBnf;
import org.h2.test.unit.TestCache; import org.h2.test.unit.TestCache;
import org.h2.test.unit.TestClearReferences; import org.h2.test.unit.TestClearReferences;
import org.h2.test.unit.TestCollation; import org.h2.test.unit.TestCollation;
...@@ -608,6 +609,7 @@ kill -9 `jps -l | grep "org.h2.test." | cut -d " " -f 1` ...@@ -608,6 +609,7 @@ kill -9 `jps -l | grep "org.h2.test." | cut -d " " -f 1`
new TestAlterSchemaRename().runTest(this); new TestAlterSchemaRename().runTest(this);
new TestAutoRecompile().runTest(this); new TestAutoRecompile().runTest(this);
new TestBitField().runTest(this); new TestBitField().runTest(this);
new TestBnf().runTest(this);
new TestBackup().runTest(this); new TestBackup().runTest(this);
new TestBigDb().runTest(this); new TestBigDb().runTest(this);
new TestBigResult().runTest(this); new TestBigResult().runTest(this);
......
/*
* Copyright 2004-2013 H2 Group. Multiple-Licensed under the H2 License,
* Version 1.0, and under the Eclipse Public License, Version 1.0
* (http://h2database.com/html/license.html).
* Initial Developer: H2 Group
*/
package org.h2.test.unit;
import org.h2.bnf.Bnf;
import org.h2.bnf.context.DbContents;
import org.h2.bnf.context.DbContextRule;
import org.h2.bnf.context.DbProcedure;
import org.h2.bnf.context.DbSchema;
import org.h2.test.TestBase;
import java.sql.Connection;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
/**
* Test Bnf Sql parser
* @author Nicolas Fortin
*/
public class TestBnf extends TestBase {
/**
* Run just this test.
*
* @param a ignored
*/
public static void main(String... a) throws Exception {
TestBase.createCaller().init().test();
}
@Override
public void test() throws Exception {
deleteDb("TestBnf");
Connection conn = getConnection("TestBnf");
try {
testProcedures(conn);
} finally {
conn.close();
}
}
private void testProcedures(Connection conn) throws Exception {
// Register a procedure and check if it is present in DbContents
conn.createStatement().execute("DROP ALIAS IF EXISTS CUSTOM_PRINT");
conn.createStatement().execute("CREATE ALIAS CUSTOM_PRINT AS $$ void print(String s) { System.out.println(s); } $$");
conn.createStatement().execute("DROP TABLE IF EXISTS TABLE_WITH_STRING_FIELD");
conn.createStatement().execute("CREATE TABLE TABLE_WITH_STRING_FIELD (STRING_FIELD VARCHAR(50), INT_FIELD integer)");
DbContents dbContents = new DbContents();
dbContents.readContents(conn.getMetaData());
DbSchema defaultSchema = dbContents.getDefaultSchema();
DbProcedure[] procedures = defaultSchema.getProcedures();
Set<String> procedureName = new HashSet<String>(procedures.length);
for(DbProcedure procedure : procedures) {
procedureName.add(procedure.getName());
}
assertTrue(procedureName.contains("CUSTOM_PRINT"));
// Test completion
Bnf bnf = Bnf.getInstance(null);
DbContextRule columnRule = new DbContextRule(dbContents, DbContextRule.COLUMN);
bnf.updateTopic("column_name", columnRule);
bnf.updateTopic("expression", new DbContextRule(dbContents, DbContextRule.PROCEDURE));
bnf.linkStatements();
// Test partial
Map<String, String> tokens = bnf.getNextTokenList("SELECT CUSTOM_PR");
assertTrue(tokens.values().contains("INT"));
// Test parameters
tokens = bnf.getNextTokenList("SELECT CUSTOM_PRINT(");
assertTrue(tokens.values().contains("STRING_FIELD"));
assertFalse(tokens.values().contains("INT_FIELD"));
// Test parameters with spaces
tokens = bnf.getNextTokenList("SELECT CUSTOM_PRINT ( ");
assertTrue(tokens.values().contains("STRING_FIELD"));
assertFalse(tokens.values().contains("INT_FIELD"));
// Test parameters with close bracket
tokens = bnf.getNextTokenList("SELECT CUSTOM_PRINT ( STRING_FIELD");
assertTrue(tokens.values().contains(")"));
}
}
\ No newline at end of file
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论