222 lines
8.1 KiB
Java
222 lines
8.1 KiB
Java
package resources;
|
|
import static org.gcube.resources.discovery.icclient.ICFactory.client;
|
|
|
|
import java.io.StringReader;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import javax.xml.parsers.DocumentBuilder;
|
|
import javax.xml.parsers.DocumentBuilderFactory;
|
|
|
|
import org.gcube.common.resources.gcore.utils.XPathHelper;
|
|
import org.gcube.common.scope.api.ScopeProvider;
|
|
import org.gcube.common.scope.impl.ScopeBean;
|
|
import org.gcube.common.scope.impl.ScopeBean.Type;
|
|
import org.gcube.resources.discovery.client.api.DiscoveryClient;
|
|
import org.gcube.resources.discovery.client.queries.api.Query;
|
|
import org.gcube.resources.discovery.client.queries.impl.QueryBox;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.w3c.dom.Document;
|
|
import org.w3c.dom.Element;
|
|
import org.xml.sax.InputSource;
|
|
|
|
|
|
/**
|
|
* The Class GetAllInfrastructureScopes.
|
|
*
|
|
* @author Francesco Mangiacrapa at ISTI-CNR (francesco.mangiacrapa@isti.cnr.it)
|
|
*
|
|
* May 16, 2019
|
|
*/
|
|
public class GetAllInfrastructureScopes {
|
|
|
|
public static Logger logger = LoggerFactory.getLogger(GetAllInfrastructureScopes.class);
|
|
|
|
protected static final String RESOURCE_PROFILE_NAME_TEXT = "/Resource/Profile/Name/text()";
|
|
|
|
|
|
/**
|
|
* Load map of scope name to full scope.
|
|
*
|
|
* @param rootScope the root scope
|
|
* @return the map of binding between (VRE_NAME, FULL_SCOPE_OF_VRE_NAME)
|
|
* @throws Exception the exception
|
|
*/
|
|
public static Map<String, String> loadMapOfScopeNameToFullScope(String rootScope) throws Exception{
|
|
|
|
String originalScope = ScopeProvider.instance.get();
|
|
|
|
try{
|
|
ScopeBean scopeBean = null;
|
|
Map<String, String> scopeNameToFullScopeMap = new HashMap<String,String>();
|
|
ScopeProvider.instance.set(rootScope);
|
|
String secondaryType = Type.INFRASTRUCTURE.name();
|
|
scopeBean = new ScopeBean(rootScope);
|
|
logger.info("Added the couple ({},{}) as {} (NAME, FULL SCOPE) into map", scopeBean.name(), rootScope, secondaryType);
|
|
scopeNameToFullScopeMap.put(scopeBean.name(), rootScope);
|
|
List<String> listVOScopes = getListOfVOScopes(secondaryType);
|
|
logger.info("Searching for secondaryType={} the scope/s found is/are: " +secondaryType, listVOScopes);
|
|
|
|
//int noVOTypeCount = 0;
|
|
for (String voScope : listVOScopes) {
|
|
//int count = voScope.length() - voScope.replace("/", "").length();
|
|
scopeBean = new ScopeBean(voScope);
|
|
//IS A VO
|
|
//if(count==2){
|
|
if(scopeBean.is(Type.VO)){
|
|
secondaryType = Type.VO.name();
|
|
logger.info("{} is a {}...",voScope,secondaryType);
|
|
ScopeProvider.instance.set(voScope);
|
|
scopeBean = new ScopeBean(voScope);
|
|
logger.info("Added the couple ({},{}) as {} (NAME, FULL SCOPE) into map", scopeBean.name(), voScope, secondaryType);
|
|
scopeNameToFullScopeMap.put(scopeBean.name(), voScope);
|
|
secondaryType = Type.VRE.name();
|
|
List<String> listVREs = getListOfResourcesForSecondaryType(secondaryType);
|
|
logger.debug("VREs found for VO "+voScope+ " is/are "+listVREs.size()+ ": "+listVREs);
|
|
for (String vreName : listVREs) {
|
|
String vreScope = String.format("%s/%s", voScope,vreName);
|
|
scopeNameToFullScopeMap.put(vreName, vreScope);
|
|
}
|
|
|
|
}else{
|
|
//noVOTypeCount++;
|
|
logger.info(voScope +" is not a VO, skipping it");
|
|
}
|
|
}
|
|
|
|
/*System.out.println("Total VO is: "+(listVOScopes.size()+noVOTypeCount));
|
|
for (String vreName : vreNameFullScope.keySet()) {
|
|
System.out.println("VRE Name: "+vreName + " has scope: "+vreNameFullScope.get(vreName));
|
|
}*/
|
|
|
|
logger.info("Total VRE is: "+scopeNameToFullScopeMap.size());
|
|
return scopeNameToFullScopeMap;
|
|
|
|
}catch(Exception e ){
|
|
throw new Exception("Error on loading the map of VRE nameto scope: ", e);
|
|
}
|
|
finally{
|
|
if(originalScope!=null && !originalScope.isEmpty()){
|
|
ScopeProvider.instance.set(originalScope);
|
|
}else
|
|
ScopeProvider.instance.reset();
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Gets the list of resources for secondary type.
|
|
*
|
|
* @param secondaryType the secondary type
|
|
* @return the list of resource names for the input secondary type
|
|
*/
|
|
private static List<String> getListOfResourcesForSecondaryType(String secondaryType) {
|
|
|
|
String queryString = "for $profile in collection('/db/Profiles/GenericResource')//Resource " +
|
|
"where $profile/Profile/SecondaryType/string() eq '"+secondaryType+"' return $profile";
|
|
|
|
List<String> listResourceName = new ArrayList<String>();
|
|
|
|
try {
|
|
logger.info("Trying to fetch GenericResource in the scope: "+ScopeProvider.instance.get()+", SecondaryType: " + secondaryType);
|
|
Query q = new QueryBox(queryString);
|
|
DiscoveryClient<String> client = client();
|
|
List<String> listGenericResources = client.submit(q);
|
|
|
|
logger.info("# of GenericResource returned are: "+listGenericResources.size());
|
|
|
|
if (listGenericResources == null || listGenericResources.size() == 0)
|
|
throw new Exception("GenericResource with SecondaryType: " + secondaryType + ", is not registered in the scope: "+ScopeProvider.instance.get());
|
|
else {
|
|
|
|
|
|
for (String genericResource : listGenericResources) {
|
|
try{
|
|
String elem = genericResource;
|
|
DocumentBuilder docBuilder = DocumentBuilderFactory.newInstance().newDocumentBuilder();
|
|
Document document = docBuilder.parse(new InputSource(new StringReader(elem)));
|
|
Element rootElement = document.getDocumentElement();
|
|
XPathHelper helper = new XPathHelper(rootElement);
|
|
List<String> resourceNames = helper.evaluate(RESOURCE_PROFILE_NAME_TEXT);
|
|
|
|
if(resourceNames!=null && resourceNames.size()>0)
|
|
listResourceName.add(resourceNames.get(0));
|
|
|
|
}catch(Exception e){
|
|
throw new Exception("Error during parsing the generic resource: "+genericResource + " in the scope: "+ScopeProvider.instance.get());
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
logger.error("Error while trying to fetch generic resource from the infrastructure", e);
|
|
}
|
|
|
|
return listResourceName;
|
|
|
|
}
|
|
|
|
|
|
/**
|
|
* Gets the list of resources for secondary type.
|
|
*
|
|
* @param secondaryType the secondary type
|
|
* @return the list of resource names for the input secondary type
|
|
*/
|
|
private static List<String> getListOfVOScopes(String secondaryType) {
|
|
|
|
String queryString = "for $profile in collection('/db/Profiles/GenericResource')//Resource " +
|
|
"where $profile/Profile/SecondaryType/string() eq '"+secondaryType+"' return $profile";
|
|
|
|
List<String> listOfVOScopes = new ArrayList<String>();
|
|
|
|
try {
|
|
logger.info("Trying to fetch GenericResource in the scope: "+ScopeProvider.instance.get()+", SecondaryType: " + secondaryType);
|
|
Query q = new QueryBox(queryString);
|
|
DiscoveryClient<String> client = client();
|
|
List<String> listGenericResources = client.submit(q);
|
|
|
|
logger.info("# of GenericResource returned searching for secondaryType= "+secondaryType+" is/are: "+listGenericResources.size());
|
|
|
|
if (listGenericResources == null || listGenericResources.size() == 0)
|
|
throw new Exception("GenericResource with SecondaryType: " + secondaryType + ", is not registered in the scope: "+ScopeProvider.instance.get());
|
|
else {
|
|
|
|
|
|
for (String genericResource : listGenericResources) {
|
|
try{
|
|
String elem = genericResource;
|
|
DocumentBuilder docBuilder = DocumentBuilderFactory.newInstance().newDocumentBuilder();
|
|
Document document = docBuilder.parse(new InputSource(new StringReader(elem)));
|
|
Element rootElement = document.getDocumentElement();
|
|
XPathHelper helper = new XPathHelper(rootElement);
|
|
// List<String> resourceNames = helper.evaluate(RESOURCE_PROFILE_NAME_TEXT);
|
|
//
|
|
// if(resourceNames!=null && resourceNames.size()>0)
|
|
// listResourceName.add(resourceNames.get(0));
|
|
|
|
List<String> scopes = helper.evaluate("/Resource/Profile/Body/infrastructures/infrastructure/vos/vo/scope/text()");
|
|
for (String scopeFound : scopes) {
|
|
listOfVOScopes.add(scopeFound);
|
|
}
|
|
|
|
}catch(Exception e){
|
|
throw new Exception("Error during parsing the generic resource: "+genericResource + " in the scope: "+ScopeProvider.instance.get());
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
logger.error("Error while trying to fetch generic resource from the infrastructure", e);
|
|
}
|
|
|
|
return listOfVOScopes;
|
|
|
|
}
|
|
}
|