95 lines
3.7 KiB
Java
95 lines
3.7 KiB
Java
package org.gcube.data.analysis.rconnector;
|
|
|
|
import static org.gcube.data.analysis.tabulardata.clientlibrary.plugin.AbstractPlugin.history;
|
|
import static org.gcube.data.analysis.tabulardata.clientlibrary.plugin.AbstractPlugin.query;
|
|
import static org.gcube.data.analysis.tabulardata.clientlibrary.plugin.AbstractPlugin.tabularResource;
|
|
|
|
import java.util.regex.Matcher;
|
|
import java.util.regex.Pattern;
|
|
|
|
import javax.inject.Inject;
|
|
import javax.inject.Singleton;
|
|
|
|
import org.gcube.common.authorization.library.provider.SecurityTokenProvider;
|
|
import org.gcube.common.database.DatabaseEndpointIdentifier;
|
|
import org.gcube.common.database.DatabaseProvider;
|
|
import org.gcube.common.database.endpoint.DatabaseEndpoint;
|
|
import org.gcube.data.analysis.rconnector.Info.DataBaseInfo;
|
|
import org.gcube.data.analysis.tabulardata.commons.webservice.types.TabularResource;
|
|
import org.gcube.data.analysis.tabulardata.model.column.Column;
|
|
import org.gcube.data.analysis.tabulardata.model.column.type.IdColumnType;
|
|
import org.gcube.data.analysis.tabulardata.model.datatype.GeometryType;
|
|
import org.gcube.data.analysis.tabulardata.model.metadata.common.NamesMetadata;
|
|
import org.gcube.data.analysis.tabulardata.model.metadata.table.DatasetViewTableMetadata;
|
|
import org.gcube.data.analysis.tabulardata.model.table.Table;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
@Slf4j
|
|
@Singleton
|
|
public class InfoRetriever {
|
|
|
|
@Inject
|
|
DatabaseProvider databaseProvider;
|
|
|
|
@SuppressWarnings("unchecked")
|
|
public Info retrieve(String userName, Long tabularResourceId){
|
|
try {
|
|
TabularResource tr = tabularResource().build().getTabularResource(tabularResourceId);
|
|
Table table = history().build().getLastTable(tabularResourceId);
|
|
if (table.contains(DatasetViewTableMetadata.class))
|
|
table= query().build().getTable(table.getMetadata(DatasetViewTableMetadata.class).getTargetDatasetViewTableId().getValue());
|
|
|
|
Info info =new Info();
|
|
info.setTableName(table.getName());
|
|
StringBuilder queryOrder = new StringBuilder();;
|
|
StringBuilder fields = new StringBuilder();
|
|
for (Column col:table.getColumnsExceptTypes(IdColumnType.class)){
|
|
fields.append(col.getMetadata(NamesMetadata.class).getTextWithLocale("en").getValue()).append(",");
|
|
if (col.getDataType().getClass().equals(GeometryType.class))
|
|
queryOrder.append("ST_AsText(").append(col.getName()).append(")").append(",");
|
|
else queryOrder.append(col.getName()).append(",");
|
|
}
|
|
fields.deleteCharAt(fields.lastIndexOf(","));
|
|
queryOrder.deleteCharAt(queryOrder.lastIndexOf(","));
|
|
|
|
info.setFields(fields.toString());
|
|
info.setQueryColumns(queryOrder.toString());
|
|
info.setUsername(userName);
|
|
info.setUserTableName(tr.getName());
|
|
info.setToken(SecurityTokenProvider.instance.get());
|
|
info.setDatabase(retrievedatabaseInfo());
|
|
|
|
log.info("retrieved info are "+info);
|
|
|
|
return info;
|
|
} catch (Exception e) {
|
|
log.error("error retrieving info ",e);
|
|
throw new RuntimeException(e.getMessage());
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
DataBaseInfo retrievedatabaseInfo(){
|
|
DatabaseEndpointIdentifier endpointId = new DatabaseEndpointIdentifier("TabularData Database",
|
|
"Data-ReadOnly");
|
|
DatabaseEndpoint endpoint = databaseProvider.get(endpointId);
|
|
DataBaseInfo databaseInfo = new DataBaseInfo();
|
|
databaseInfo.setDatabaseUsername(endpoint.getCredentials().getUsername());
|
|
databaseInfo.setDatabasePassword(endpoint.getCredentials().getPassword());
|
|
|
|
Pattern pattern = Pattern.compile("[^/]*//([^:]*)[^/]*/(.*)");
|
|
Matcher m = pattern.matcher(endpoint.getConnectionString());
|
|
m.find();
|
|
String databaseAddress = m.group(1);
|
|
String databaseName = m.group(2);
|
|
|
|
databaseInfo.setDatabaseAddress(databaseAddress);
|
|
databaseInfo.setDatabaseName(databaseName);
|
|
return databaseInfo;
|
|
}
|
|
|
|
}
|