litvinovg/vitro
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

code modification to get whole list of url.

Browse source
This commit is contained in:
ass92 2011-01-21 17:10:35 +00:00
parent ae22aed70a
commit d220c7e9af
1 changed files with 51 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

66
webapp/src/edu/cornell/mannlib/vitro/webapp/controller/EntityURLController.java
View file

@ -19,6 +19,15 @@ import javax.servlet.http.HttpSession;
import org.apache.commons.logging.Log; import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory; import org.apache.commons.logging.LogFactory;
import org.apache.lucene.document.Document;
import org.apache.lucene.index.Term;
import org.apache.lucene.search.BooleanClause;
import org.apache.lucene.search.BooleanQuery;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.ScoreDoc;
import org.apache.lucene.search.Sort;
import org.apache.lucene.search.TermQuery;
import org.apache.lucene.search.TopDocs;
import com.hp.hpl.jena.ontology.OntClass; import com.hp.hpl.jena.ontology.OntClass;
import com.hp.hpl.jena.ontology.OntModel; import com.hp.hpl.jena.ontology.OntModel;
@ -44,12 +53,15 @@ import edu.cornell.mannlib.vitro.webapp.beans.VClass;
import edu.cornell.mannlib.vitro.webapp.dao.Classes2ClassesDao; import edu.cornell.mannlib.vitro.webapp.dao.Classes2ClassesDao;
import edu.cornell.mannlib.vitro.webapp.dao.OntologyDao; import edu.cornell.mannlib.vitro.webapp.dao.OntologyDao;
import edu.cornell.mannlib.vitro.webapp.dao.VClassDao; import edu.cornell.mannlib.vitro.webapp.dao.VClassDao;
import edu.cornell.mannlib.vitro.webapp.search.lucene.Entity2LuceneDoc;
import edu.cornell.mannlib.vitro.webapp.search.lucene.LuceneIndexFactory;
import edu.cornell.mannlib.vitro.webapp.web.ContentType; import edu.cornell.mannlib.vitro.webapp.web.ContentType;
public class EntityURLController extends VitroHttpServlet { public class EntityURLController extends VitroHttpServlet {
private static final Log log = LogFactory.getLog(EntityURLController.class.getName()); private static final Log log = LogFactory.getLog(EntityURLController.class.getName());
public static final int ENTITY_LIST_CONTROLLER_MAX_RESULTS = 30000;
public void doGet (HttpServletRequest req, HttpServletResponse res) throws IOException, ServletException{ public void doGet (HttpServletRequest req, HttpServletResponse res) throws IOException, ServletException{
@ -84,21 +96,45 @@ public void doGet (HttpServletRequest req, HttpServletResponse res) throws IOExc
return; return;
} }
List<Individual> inds = (List<Individual>)getServletContext().getAttribute("inds"); String classUri = (String) getServletContext().getAttribute("classuri");
Model model = ModelFactory.createDefaultModel(); BooleanQuery query = new BooleanQuery();
if(inds != null){ query.add(
System.out.println("Into the loop"); new TermQuery( new Term(Entity2LuceneDoc.term.RDFTYPE, classUri)),
Iterator<Individual> itr = (Iterator<Individual>)inds.iterator(); BooleanClause.Occur.MUST );
Individual ind = null;
Resource resource = null; IndexSearcher index = LuceneIndexFactory.getIndexSearcher(getServletContext());
while(itr.hasNext()){ TopDocs docs = index.search(query, null,
ind = itr.next(); ENTITY_LIST_CONTROLLER_MAX_RESULTS,
resource = ResourceFactory.createResource(ind.getURI()); new Sort(Entity2LuceneDoc.term.NAMEUNANALYZED));
RDFNode node = (RDFNode) ResourceFactory.createResource((String) getServletContext().getAttribute("classuri"));
model.add(resource, RDF.type, node); if( docs == null ){
} log.error("Search of lucene index returned null");
} throw new ServletException("Search of lucene index returned null");
}
int ii = 0;
int size = docs.totalHits;
Resource resource = null;
RDFNode node = null;
Model model = ModelFactory.createDefaultModel();
while( ii < size ){
ScoreDoc hit = docs.scoreDocs[ii];
if (hit != null) {
Document doc = index.doc(hit.doc);
if (doc != null) {
String uri = doc.getField(Entity2LuceneDoc.term.URI).stringValue();
resource = ResourceFactory.createResource(uri);
node = (RDFNode) ResourceFactory.createResource(classUri);
model.add(resource, RDF.type, node);
} else {
log.warn("no document found for lucene doc id " + hit.doc);
}
} else {
log.debug("hit was null");
}
ii++;
}
String format = ""; String format = "";
if(contentType != null){ if(contentType != null){
if ( RDFXML_MIMETYPE.equals(contentType.getMediaType())) if ( RDFXML_MIMETYPE.equals(contentType.getMediaType()))