public class ResultSetImpl_TopDocs implements ResultSet {
protected boolean normalize = true;
protected String keyword = null;
protected SchemaFactory schemaFactory = SchemaFactoryImpl.getInstance();
protected int topcount;
protected DocStream resultStream;
protected IndexReader indexReader = null;
protected TopDocs topDocs;
protected ResultSetImpl_TopDocs(DocStream resultStream, IndexReader indexReader)
throws Exception {
this(resultStream, indexReader, 50, true);
}
protected ResultSetImpl_TopDocs(
DocStream resultStream, IndexReader indexReader, int topCount, boolean normalize)
throws Exception {
this.normalize = normalize;
this.resultStream = resultStream;
this.indexReader = indexReader;
topcount = topCount;
generateTopDocs();
}
private TopDocCollector collector;
private void generateTopDocs() throws Exception {
collector = new TopDocCollector(topcount);
resultStream.init();
for (int i = 0; i < resultStream.getLen(); i++) {
collector.collect(resultStream.doc(), (float) (resultStream.score()));
resultStream.next();
}
;
topDocs = collector.topDocs();
}
public int getDocID(int index) throws Exception {
if (index >= topcount) {
topcount = index * 2;
generateTopDocs();
}
return topDocs.scoreDocs[index].doc;
}
public int getEstimatedNumberOfCompleteResults() {
return resultStream.getLen();
}
public int getLength() {
return resultStream.getLen();
}
protected SchemaObjectInfo getInfo(IndexReader indexReader, int docID) throws IOException {
FieldType[] fields =
new FieldType[] {
FieldType.ID, FieldType.URI, FieldType.LABEL, FieldType.SUMMARY, FieldType.TEXT
};
String[] values = indexReader.getFieldValues(docID, fields);
return schemaFactory.createSchemaObjectInfo(
Long.valueOf(values[0]), values[1], values[2], values[3], values[4]);
}
public SchemaObjectInfo getResult(int index) throws Exception {
int id = getDocID(index);
return getInfo(indexReader, id);
}
public double getScore(int index) throws Exception {
getDocID(index);
if (normalize) return topDocs.scoreDocs[index].score / topDocs.scoreDocs[0].score;
else return topDocs.scoreDocs[index].score;
}
public String getSnippet(int index) throws Exception {
StandardAnalyzer analyzer = new StandardAnalyzer();
QueryParser qp = new QueryParser(FieldType.TEXT.toString(), analyzer);
SchemaObjectInfo info = getResult(index);
String text = info.getTextDescription();
String label = info.getLabel().toLowerCase();
// remove abundant labels
if (!label.equals("")) {
while (text.indexOf(label) == 0) text = text.substring(label.length()).trim();
}
if (keyword != null) {
try {
Highlighter hl = new Highlighter(new QueryScorer(qp.parse(keyword)));
TokenStream ts = analyzer.tokenStream(text, new StringReader(text));
String snippet = hl.getBestFragment(ts, text);
if (snippet != null) return snippet;
} catch (Exception e) {
// e.printStackTrace();
}
}
return text;
}
public ResultSet setSnippetKeyword(String keyword) {
this.keyword = keyword;
return this;
}
public DocStream getResultStream() throws IOException {
return resultStream;
}
}
public class ForServlet {
DOMParser parser = new DOMParser();
XFacetedSearchable searcher = null;
SchemaFactory schemaFactory = SchemaFactoryImpl.getInstance();
SearchFactory searchFactory = SearchFactoryImpl.getInstance();
Hashtable<String, XFacetedResultSet> resultCache = new Hashtable<String, XFacetedResultSet>();
private GeneralCategory createGeneralCategory(Node node)
throws NoSuchAlgorithmException, DOMException {
NamedNodeMap map = node.getAttributes();
CompoundCategory cat = schemaFactory.createCompoundCategory(CompoundCategory.TYPE_AND);
Node n = map.getNamedItem("instanceConstraint");
if (n != null && !n.getNodeValue().equals("")) {
cat.addComponentCategory(
schemaFactory
.createEnumerationCategory()
.addInstanceElement(schemaFactory.createInstance(n.getNodeValue())));
return cat;
}
n = map.getNamedItem("conceptConstraint");
if (n != null && !n.getNodeValue().equals("")) {
cat.addComponentCategory(schemaFactory.createCategory(n.getNodeValue()));
}
n = map.getNamedItem("keyword");
if (n != null && !n.getNodeValue().equals("")) {
cat.addComponentCategory(schemaFactory.createKeywordCategory(n.getNodeValue()));
}
return cat;
}
private void addRelations(CatRelGraph graph, int fromNodeIndex, NodeList nodes)
throws NoSuchAlgorithmException, DOMException {
if (nodes == null) return;
for (int i = 0; i < nodes.getLength(); i++) {
Node node = nodes.item(i);
if (node.getNodeName().equals("#text")) continue;
// System.out.println("name="+node.getNodeName());
graph.add(createGeneralCategory(node));
int toNodeIndex = graph.numOfNodes() - 1;
NamedNodeMap map = node.getAttributes();
if (map.getNamedItem("inverse").getNodeValue().equals("0")) {
graph.add(
schemaFactory.createRelation(map.getNamedItem("relation").getNodeValue()),
fromNodeIndex,
toNodeIndex);
} else {
graph.add(
schemaFactory.createRelation(map.getNamedItem("relation").getNodeValue()),
toNodeIndex,
fromNodeIndex);
}
addRelations(graph, toNodeIndex, node.getChildNodes());
}
}
private XFacetedQuery XML2Query(Document doc) {
try {
Node node = null;
NamedNodeMap map = null;
node = doc.getElementsByTagName("action").item(0);
map = node.getAttributes();
// create XFacetedQuery
node = doc.getElementsByTagName("targetNode").item(0);
map = node.getAttributes();
int target = new Integer(map.getNamedItem("index").getNodeValue()).intValue();
CatRelGraph graph = schemaFactory.createCatRelGraph();
node = doc.getElementsByTagName("queryNode").item(0);
graph.add(createGeneralCategory(node));
// System.out.println("name="+node.getNodeName());
addRelations(graph, 0, node.getChildNodes());
System.out.println(graph);
System.out.println("target=" + target);
XFacetedQuery query = searchFactory.createXFacetedQuery();
query.setSearchTarget(target);
query.setQueryConstraint(graph);
query.setResultSpec(searchFactory.createXFacetedResultSpec());
return query;
} catch (Exception ex) {
ex.printStackTrace();
}
return null;
}
private String nomalizeLabel(String label) {
return label.replaceAll("[0-9]*", "").replaceFirst("Category:", "");
}
private String result2XML(XFacetedResultSet resultSet, Document doc2, long queryTime) {
try {
} catch (Exception ex) {
System.out.println(ex);
}
try {
Node node = null;
NamedNodeMap map = null;
node = doc2.getElementsByTagName("action").item(0);
map = node.getAttributes();
DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
DocumentBuilder db = dbf.newDocumentBuilder();
Document doc = db.newDocument();
Element root = doc.createElement("result"); // Create Root Element
if (map.getNamedItem("type").getNodeValue().equals("all")
|| map.getNamedItem("type").getNodeValue().equals("instanceList")) {
Element item = doc.createElement("instanceList"); // Create
// element
item.setAttribute("count", Integer.toString(resultSet.getLength()));
item.setAttribute("time", Long.toString(queryTime));
int start = new Integer(map.getNamedItem("start").getNodeValue()).intValue();
int count = new Integer(map.getNamedItem("count").getNodeValue()).intValue();
int end = start + count;
if (end > resultSet.getLength()) end = resultSet.getLength();
for (int i = start; i < end; i++) {
SchemaObjectInfo result = resultSet.getResult(i);
Element subitem = doc.createElement("instance");
try {
subitem.setAttribute("name", java.net.URLDecoder.decode(result.getLabel()));
} catch (Exception e) {
subitem.setAttribute("name", result.getLabel());
}
subitem.setAttribute("uri", result.getURI());
// subitem.setAttribute("snippet",
// result.getTextDescription());
try {
subitem.setAttribute("snippet", java.net.URLDecoder.decode(resultSet.getSnippet(i)));
} catch (Exception e) {
subitem.setAttribute("snippet", resultSet.getSnippet(i));
}
item.appendChild(subitem);
// for debug
// System.out.println("result "+i+":
// "+resultSet.getResult(i).getLabel());
// System.out.println("//score:"+resultSet.getScore(i));
// System.out.println("//uri:"+resultSet.getResult(i).getURI());
// System.out.println("//id:"+resultSet.getDocID(i));
// System.out.println("//text:"+resultSet.getResult(i).getTextDescription());
// System.out.println("//text:"+resultSet.getSnippet(i));
// for debug
}
root.appendChild(item); // atach element to Root element
}
if (map.getNamedItem("type").getNodeValue().equals("all")
|| map.getNamedItem("type").getNodeValue().equals("conceptList")) {
Element item = doc.createElement("conceptList"); // Create
// element
int start = new Integer(map.getNamedItem("start").getNodeValue()).intValue();
int count = new Integer(map.getNamedItem("count").getNodeValue()).intValue();
int end = start + count;
Facet[] facets = resultSet.getCategoryFacets();
if (end > facets.length) end = facets.length;
item.setAttribute("count", Integer.toString(facets.length));
for (int i = start; i < end; i++) {
Facet result = facets[i];
Element subitem = doc.createElement("concept");
subitem.setAttribute("name", nomalizeLabel(result.getInfo().getLabel()));
subitem.setAttribute("uri", result.getInfo().getURI());
subitem.setAttribute("count", String.valueOf(result.getCount()));
item.appendChild(subitem);
}
root.appendChild(item); // atach element to Root element
}
if (map.getNamedItem("type").getNodeValue().equals("all")
|| map.getNamedItem("type").getNodeValue().equals("relationList")) {
Element item = doc.createElement("relationList"); // Create
// element
int start = new Integer(map.getNamedItem("start").getNodeValue()).intValue();
int count = new Integer(map.getNamedItem("count").getNodeValue()).intValue();
int end = start + count;
Facet[] facets = resultSet.getRelationFacets();
if (end > facets.length) end = facets.length;
item.setAttribute("count", Integer.toString(facets.length));
for (int i = start; i < end; i++) {
Facet result = facets[i];
Element subitem = doc.createElement("relation");
subitem.setAttribute("name", result.getInfo().getLabel());
subitem.setAttribute("uri", result.getInfo().getURI());
subitem.setAttribute("count", String.valueOf(result.getCount()));
if (result.isInverseRelation()) subitem.setAttribute("inverse", "1");
else subitem.setAttribute("inverse", "0");
item.appendChild(subitem);
}
root.appendChild(item); // atach element to Root element
}
doc.appendChild(root); // Add Root to Document
OutputFormat format = new OutputFormat(doc); // Serialize DOM
StringWriter stringOut = new StringWriter(); // Writer will be a
// String
XMLSerializer serial = new XMLSerializer(stringOut, format);
serial.asDOMSerializer(); // As a DOM Serializer
serial.serialize(doc.getDocumentElement());
// System.out.println( "STRXML = " + stringOut.toString() ); //Spit
// out DOM as a String
return stringOut.toString();
} catch (Exception ex) {
ex.printStackTrace();
}
return null;
}
public String query(String queryXML, String uid) {
return query(new InputSource(new StringReader(queryXML)), uid);
}
public String query(File queryXMLFile, String uid) {
try {
return query(new InputSource(new FileReader(queryXMLFile)), uid);
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
public String query(InputSource input, String uid) {
try {
XFacetedResultSet resultSet = null;
parser.parse(input);
Document doc = parser.getDocument();
Node node = null;
NamedNodeMap map = null;
long time_b = System.currentTimeMillis();
// create result set
node = doc.getElementsByTagName("action").item(0);
map = node.getAttributes();
if (map.getNamedItem("type").getNodeValue().equals("all")) { // a new
// query
XFacetedQuery query = XML2Query(doc);
time_b = System.currentTimeMillis();
resultSet = searcher.search(query);
resultCache.put(uid, resultSet);
} else { // "more_instance","more_concept","more_relation"
resultSet = resultCache.get(uid);
if (resultSet == null) {
XFacetedQuery query = XML2Query(doc);
// System.out.println(query);
time_b = System.currentTimeMillis();
resultSet = searcher.search(query);
}
}
long time_e = System.currentTimeMillis();
long queryTime = time_e - time_b;
// result set to xml
return result2XML(resultSet, doc, queryTime);
} catch (Exception ex) {
ex.printStackTrace();
}
return null;
}
public void init(String configFile) throws Exception {
Properties config = Config.readConfigFile(configFile);
XFacetedSearchService searchService = searchFactory.getXFacetedSearchService(config);
searcher = searchService.getXFacetedSearchable();
}
public static void main(String[] args) {
try {
ForServlet forServ = new ForServlet();
String configFile =
args[0]; // "E:\\User\\AllisQM\\Semplore\\SSModel\\config\\semplore-dbpedia-yago.cfg";
forServ.init(configFile); // config file
String queryXML =
"<?xml version=\"1.0\" encoding=\"UTF-8\"?><query>"
+ " <action type=\"all\" start=\"0\" count=\"10\"/>"
+ " <graph>"
+ " <queryNode insName=\"\" instanceConstraint=\"\" relName=\"\" relation=\"\" inverse=\"0\" conName=\"\" conceptConstraint=\"\" keyword=\"academy award\">"
+ " <queryNode relation=\"<http://dbpedia.org/property/starring>\" inverse=\"0\" relName=\"starring\" insName=\"\" conceptConstraint=\"\" instanceConstraint=\"\" keyword=\"star wars\"/>"
+ " <queryNode relation=\"<http://dbpedia.org/property/director>\" inverse=\"0\" relName=\"director\" insName=\"\" conceptConstraint=\"<http://dbpedia.org/resource/Category:Jewish_American_film_directors>\" instanceConstraint=\"\" keyword=\"\" conName=\"TOP Category\"/>"
+ " </queryNode>"
+ " </graph>"
+ " <targetNode index=\"2\"/>"
+ "</query>";
String uid = "127.0.0.1";
String resultXML = forServ.query(queryXML, uid);
System.out.println("uid:" + uid);
System.out.println("queryXML:" + queryXML);
System.out.println("resultXML:" + resultXML);
} catch (Exception e) {
e.printStackTrace();
}
}
}