Sql Support within Solr-类Sql的solr搜索实现(1)

bco

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;
　　
//ANTLRStringStream  in = 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);
　　
}
　　
}
　　
}
　　
}