111 lines
3.6 KiB
Java
111 lines
3.6 KiB
Java
|
|
package eu.dnetlib.dhp.oa.graph.dump;
|
|
|
|
import java.io.StringReader;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import org.dom4j.Document;
|
|
import org.dom4j.DocumentException;
|
|
import org.dom4j.Element;
|
|
import org.dom4j.Node;
|
|
import org.dom4j.io.SAXReader;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.xml.sax.SAXException;
|
|
|
|
import eu.dnetlib.dhp.oa.graph.dump.community.CommunityMap;
|
|
import eu.dnetlib.dhp.oa.graph.dump.csv.Constants;
|
|
import eu.dnetlib.dhp.oa.graph.dump.csv.DumpCommunities;
|
|
import eu.dnetlib.dhp.utils.DHPUtils;
|
|
import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;
|
|
import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;
|
|
|
|
public class QueryInformationSystem {
|
|
|
|
private ISLookUpService isLookUp;
|
|
|
|
private static final Logger log = LoggerFactory.getLogger(QueryInformationSystem.class);
|
|
|
|
private static final String XQUERY_ALL = "for $x in collection('/db/DRIVER/ContextDSResources/ContextDSResourceType') "
|
|
+
|
|
" where $x//CONFIGURATION/context[./@type='community' or ./@type='ri'] " +
|
|
" and ($x//context/param[./@name = 'status']/text() = 'all') "
|
|
+
|
|
" return " +
|
|
"<community> " +
|
|
"{$x//CONFIGURATION/context/@id}" +
|
|
"{$x//CONFIGURATION/context/@label}" +
|
|
"</community>";
|
|
|
|
private static final String XQUERY_CI = "for $x in collection('/db/DRIVER/ContextDSResources/ContextDSResourceType') "
|
|
+
|
|
" where $x//CONFIGURATION/context[./@type='community' or ./@type='ri'] " +
|
|
" and $x//CONFIGURATION/context[./@id=%s] "
|
|
+
|
|
" return " +
|
|
"<community> " +
|
|
"{$x//CONFIGURATION/context/@id}" +
|
|
"{$x//CONFIGURATION/context/@label}" +
|
|
"</community>";
|
|
|
|
public CommunityMap getCommunityMap(boolean singleCommunity, String communityId)
|
|
throws ISLookUpException, DocumentException, SAXException {
|
|
if (singleCommunity)
|
|
return getMap(isLookUp.quickSearchProfile(XQUERY_CI.replace("%s", "'" + communityId + "'")));
|
|
return getMap(isLookUp.quickSearchProfile(XQUERY_ALL));
|
|
|
|
}
|
|
|
|
public ISLookUpService getIsLookUp() {
|
|
return isLookUp;
|
|
}
|
|
|
|
public void setIsLookUp(ISLookUpService isLookUpService) {
|
|
this.isLookUp = isLookUpService;
|
|
}
|
|
|
|
private CommunityMap getMap(List<String> communityMap) throws DocumentException, SAXException {
|
|
final CommunityMap map = new CommunityMap();
|
|
|
|
for (String xml : communityMap) {
|
|
final Document doc;
|
|
final SAXReader reader = new SAXReader();
|
|
reader.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);
|
|
doc = reader.read(new StringReader(xml));
|
|
Element root = doc.getRootElement();
|
|
map.put(root.attribute("id").getValue(), root.attribute("label").getValue());
|
|
}
|
|
|
|
return map;
|
|
}
|
|
|
|
public List<String> getCommunityCsv(String toString) throws ISLookUpException, SAXException, DocumentException {
|
|
List<String> communities = new ArrayList<>();
|
|
|
|
for (String xml : isLookUp.quickSearchProfile(toString)) {
|
|
log.info(xml);
|
|
final Document doc;
|
|
final SAXReader reader = new SAXReader();
|
|
reader.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);
|
|
doc = reader.read(new StringReader(xml));
|
|
Element root = doc.getRootElement();
|
|
StringBuilder builder = new StringBuilder();
|
|
builder.append(DHPUtils.md5(root.attribute("id").getValue()));
|
|
builder.append(Constants.SEP);
|
|
builder.append(root.attribute("label").getValue());
|
|
builder.append(Constants.SEP);
|
|
builder.append(root.attribute("id").getValue());
|
|
builder.append(Constants.SEP);
|
|
builder
|
|
.append(
|
|
((Node) (root.selectNodes("//description").get(0)))
|
|
.getText()
|
|
.replace("\n", " ")
|
|
.replace("\t", " "));
|
|
communities.add(builder.toString());
|
|
}
|
|
return communities;
|
|
}
|
|
}
|