Sql Support within Solr-类Sql的solr搜索实现(1)
public>public static void main(String[] args) throws
IOException
{
//ConsoleReader conReader = new ConsoleReader();
String filedemo=”testSql2Solr.txt”; // 改为从文件读取sql 语法
BufferedReader fileReader = new BufferedReader( new
InputStreamReader(new FileInputStream(filedemo)));
EvalContext ctx = new EvalContext();
StringBuilder builder = new StringBuilder();
CharSequenceReader reader = new CharSequenceReader(builder);
final String NORMAL_PROMPT = “SQL>”;
final String CONTD_PROMPT = “… “;
String prompt = NORMAL_PROMPT;
String targetUrl=”http://localhost:8080/terminator-search/search4****-0“;
String name =”conn”; //这个name 对应testSql2Solr.txt 中的 信息，然后关联到
targetUrl
UseStatement use = new UseStatement();
use.setUrl( targetUrl );
use.setName( name );
try {
use.execute(ctx);
} catch (Exception e) {
e.printStackTrace();
}
while (true)
{
String line = fileReader.readLine();
//System.out.println(“–”+line);
if ( line == null ) {
System.out.println(“line is null so exit”);
break;
}
// String line =”SELECT * FROM conn WHERE>
builder.append(line + “n”);
reader.reset();
try {
if ( tryEvaluate( reader, ctx ) ) {
prompt = NORMAL_PROMPT;
} else {
prompt = CONTD_PROMPT;
continue;
}
} catch (Exception e) {
e.printStackTrace();
prompt = NORMAL_PROMPT;
}finally{
}
builder.setLength(0);
builder.trimToSize();
}
fileReader.close();
}
private static boolean tryEvaluate(Reader reader, EvalContext
ctx)
throws SyntaxException, EvalException, IOException
{
//String
expr= null;
//ANTLRStringStreamin = new
ANTLRStringStream(expr);
ANTLRReaderStream stream = new ANTLRReaderStream( reader
);
RecognizerSharedState state = new RecognizerSharedState();
SolrqlLexer lexer = new SolrqlLexer(stream, state);//词法分析器
CommonTokenStream tokenStream = new
CommonTokenStream(lexer);//获取lexer构造的记号流
SolrqlParser parser = new SolrqlParser(tokenStream);//语法分析器
try {
stmt_list_return ret = parser.stmt_list();
if (ret != null) {
evaluate((CommonTree)ret.getTree(), ctx);
}
} catch (MismatchedTokenException e) {
if (e.getUnexpectedType() == SolrqlParser.EOF) {
// immature input. continue.
return false;
} else {
throw new SyntaxException(
lexer.getErrorMessage(e, SolrqlParser.tokenNames), e);
}
} catch (RecognitionException e) {
throw new SyntaxException(
lexer.getErrorMessage(e, SolrqlParser.tokenNames), e);
}
return true;
}
private static void evaluate(CommonTree t, EvalContext ctx)
throws EvalException, SyntaxException
{
printTree(t, 0);
List statements = new StatementListBuilder().build(t);
for (Statement stmt: statements) {
Object ret = stmt.execute(ctx);
if ( ret == null ) {
System.out.println(“OK.”);
continue;
}
if ( ret instanceof UpdateResponse ) {
printUpdateResponse((UpdateResponse)ret);
} else if ( ret instanceof SelectStatement.Result ) {
printSelectResult((SelectStatement.Result)ret);
} else {
System.out.println(“Result: ” + ret.toString());
}
}
}
private static void printSelectResult(SelectStatement.Result
res)
{
int i=1;
for (SolrDocument doc: res.getDocuments()) {
System.out.print(Integer.toString(i) + “: “);
int j=0;
for (Map.Entry e: doc.entrySet()) {
if (j > 0) {
System.out.print(“, “);
}
System.out.print(e.getKey() + “=” + e.getValue().toString());
++j;
}
System.out.println();
++i;
}
System.out.println();
System.out.println(“Query=” + res.getQuery());
System.out.println(
“Found ” + res.getResponse().getResults().getNumFound() +
” documents, Offset=” +
res.getResponse().getResults().getStart()
);
System.out.println(
“Status=” + res.getResponse().getStatus() +
“, Elapsed Time=” + res.getResponse().getElapsedTime()
);
}
private static void printUpdateResponse(UpdateResponse res)
{
System.out.println(
“Status=” + res.getStatus() +
“, Elapsed Time=” + res.getElapsedTime()
);
}
private static void printTree(Tree t, int indent)
{
if (t != null) {
StringBuffer sb = new StringBuffer(indent);
for (int i = 0; i < indent; ++i) {
sb =
sb.append(“
“);
}
if (!t.isNil()) {
System.out.println(sb.toString() + t.toString());
++indent;
}
for (int i = 0; i < t.getChildCount(); ++i) {
printTree(t.getChild(i), indent);
}
}
}
}
页:
[1]