private void runSQL(Dbms dbms, List<String> data, boolean failOnError) throws Exception {
StringBuffer sb = new StringBuffer();
for (String row : data) {
if (!row.toUpperCase().startsWith("REM") && !row.startsWith("--") && !row.trim().equals("")) {
sb.append(" ");
sb.append(row);
if (row.endsWith(";")) {
String sql = sb.toString();
sql = sql.substring(0, sql.length() - 1);
if (Log.isDebugEnabled(Geonet.DB)) Log.debug(Geonet.DB, "Executing " + sql);
try {
if (sql.trim().startsWith("SELECT")) {
dbms.select(sql);
} else {
dbms.execute(sql);
}
} catch (SQLException e) {
Log.warning(Geonet.DB, "SQL failure for: " + sql + ", error is:" + e.getMessage());
if (failOnError) throw e;
}
sb = new StringBuffer();
}
}
}
dbms.commit();
}
public void insertData(
ServletContext servletContext, Dbms dbms, String appPath, String filePath, String filePrefix)
throws Exception {
if (Log.isDebugEnabled(Geonet.DB)) Log.debug(Geonet.DB, "Filling database tables");
List<String> data = loadSqlDataFile(servletContext, dbms, appPath, filePath, filePrefix);
runSQL(dbms, data);
}
/**
* Create database schema.
*
* @param servletContext
* @param dbms
*/
public void createSchema(
ServletContext servletContext, Dbms dbms, String appPath, String filePath, String filePrefix)
throws Exception {
if (Log.isDebugEnabled(Geonet.DB)) Log.debug(Geonet.DB, "Creating database schema");
List<String> schema = loadSchemaFile(servletContext, dbms, appPath, filePath, filePrefix);
runSQL(dbms, schema);
}
/**
* TODO javadoc.
*
* @param session
* @param id
* @return
*/
protected static Element getMetadataFromSession(UserSession session, String id) {
if (Log.isDebugEnabled(Geonet.EDITOR)) {
Log.debug(Geonet.EDITOR, "Retrieving metadata from session " + session.getUserId());
}
Element md = (Element) session.getProperty(Geonet.Session.METADATA_EDITING + id);
md.detach();
return md;
}
/**
* Adds missing namespace (ie. GML) to XML inputs. It should be done by the client side but add a
* check in here.
*
* @param fragment The fragment to be checked and processed.
* @return The updated fragment.
*/
protected static String addNamespaceToFragment(String fragment) {
// add the gml namespace if its missing
if (fragment.contains("<gml:") && !fragment.contains("xmlns:gml=\"")) {
if (Log.isDebugEnabled(Geonet.EDITOR))
Log.debug(Geonet.EDITOR, " Add missing GML namespace.");
fragment =
fragment.replaceFirst(
"<gml:([^ >]+)", "<gml:$1 xmlns:gml=\"http://www.opengis.net/gml\"");
}
return fragment;
}
/**
* Check if db specific SQL script exist, if not return default SQL script path.
*
* @param filePath
* @param prefix
* @param type
* @return
*/
private String checkFilePath(String filePath, String prefix, String type) {
String dbFilePath = filePath + "/" + prefix + type + SQL_EXTENSION;
File dbFile = new File(dbFilePath);
if (dbFile.exists()) return dbFilePath;
String defaultFilePath = filePath + "/" + prefix + "default" + SQL_EXTENSION;
File defaultFile = new File(defaultFilePath);
if (defaultFile.exists()) return defaultFilePath;
else if (Log.isDebugEnabled(Geonet.DB))
Log.debug(Geonet.DB, " No default SQL script found: " + defaultFilePath);
return "";
}
/**
* Execute query and commit
*
* @param dbms
* @param query
* @return
*/
private boolean safeExecute(Dbms dbms, String query) {
try {
dbms.execute(query);
// --- as far as I remember, PostgreSQL needs a commit even for DDL
dbms.commit();
return true;
} catch (SQLException e) {
if (Log.isDebugEnabled(Geonet.DB))
Log.debug(Geonet.DB, "Safe execute error: " + query + ", error is:" + e.getMessage());
dbms.abort();
return false;
}
}
/**
* Remove all objects in the database. Read the SQL file and check all CREATE TABLE statements to
* collect the list of table to remove.
*
* @param dbms
* @throws FileNotFoundException
* @throws IOException
*/
public void removeObjects(
ServletContext servletContext, Dbms dbms, String appPath, String filePath, String filePrefix)
throws FileNotFoundException, IOException {
if (Log.isDebugEnabled(Geonet.DB)) Log.debug(Geonet.DB, "Removing database objects");
List<String> schema = loadSchemaFile(servletContext, dbms, appPath, filePath, filePrefix);
// --- step 1 : collect objects to remove
ArrayList<ObjectInfo> objects = new ArrayList<ObjectInfo>();
for (String row : schema)
if (row.toUpperCase().startsWith("CREATE ")) {
ObjectInfo oi = new ObjectInfo();
oi.name = getObjectName(row);
oi.type = getObjectType(row);
if (!oi.type.toLowerCase().equals("index")) objects.add(oi);
}
// --- step 2 : remove objects
while (true) {
boolean removed = false;
for (Iterator<ObjectInfo> i = objects.iterator(); i.hasNext(); ) {
ObjectInfo oi = i.next();
if (Log.isDebugEnabled(Geonet.DB)) Log.debug(Geonet.DB, " * Dropping " + oi.name);
String query = "DROP " + oi.type + " " + oi.name;
if (safeExecute(dbms, query)) {
removed = true;
i.remove();
}
}
if (objects.size() == 0) return;
// --- if no object was removed then we have a cyclic loop
if (!removed) {
ArrayList<String> al = new ArrayList<String>();
for (ObjectInfo oi : objects) al.add(oi.name);
return;
}
}
}
public GNResultSet(GNXMLQuery query, Object userInfo, Observer[] observers, ServiceContext srvctx)
throws Exception {
super(observers);
this.query = query;
this.srvxtx = srvctx;
try {
GeonetContext gc = (GeonetContext) this.srvxtx.getHandlerContext(Geonet.CONTEXT_NAME);
SearchManager searchMan = gc.getBean(SearchManager.class);
metasearcher = searchMan.newSearcher(SearchManager.LUCENE, Geonet.File.SEARCH_Z3950_SERVER);
} catch (Exception e) {
if (Log.isDebugEnabled(Geonet.Z3950_SERVER))
Log.debug(Geonet.Z3950_SERVER, "error constructing GNresult set: " + e);
e.printStackTrace();
}
}
/**
* @param dbms
* @return
* @throws FileNotFoundException
* @throws IOException
*/
private List<String> loadSchemaFile(
ServletContext servletContext,
Dbms dbms,
String appPath,
String filePath,
String filePrefix) // FIXME :
// use
// resource
// dir
// instead
// of
// appPath
throws FileNotFoundException, IOException {
// --- find out which dbms schema to load
String file = checkFilePath(filePath, filePrefix, DatabaseType.lookup(dbms).toString());
if (Log.isDebugEnabled(Geonet.DB)) Log.debug(Geonet.DB, " Loading script:" + file);
// --- load the dbms schema
return Lib.text.load(servletContext, appPath, file);
}
public LDAPInfo lookUp(String username, String password) {
try {
String uidFilter = "(" + uidAttr + "=" + username + ")";
String usersBaseDN = usersDN + "," + baseDN;
String path = null;
try {
path = LDAPUtil.findUserDN(getUrl(), uidFilter, usersBaseDN);
} catch (NamingException ex) {
Log.warning(Geonet.LDAP, ex.getMessage());
}
if (path == null || path.length() == 0) {
path = uidAttr + "=" + username + "," + usersDN + "," + baseDN;
}
DirContext dc = LDAPUtil.openContext(getUrl(), path, password);
Map<String, ? extends List<Object>> attr = LDAPUtil.getNodeInfo(dc, path);
dc.close();
if (attr == null) {
Log.warning(Geonet.LDAP, "Username not found :'" + username + "'");
return null;
} else {
LDAPInfo info = new LDAPInfo();
info.username = username;
info.password = password;
info.name = get(attr, nameAttr);
info.profile = (profileAttr == null) ? defProfile : get(attr, profileAttr);
if (info.profile.equals("Reviewer")) {
info.profile = "Hoofdeditor";
}
info.email = get(attr, emailAttr);
info.groups = (groupAttr == null) ? new String[] {defGroup} : getAll(attr, groupAttr);
if (!profiles.contains(info.profile)) {
Log.warning(Geonet.LDAP, "Skipping user with unknown profile");
Log.warning(Geonet.LDAP, " (C) Username :'******'");
Log.warning(Geonet.LDAP, " (C) Profile :'" + info.profile + "'");
return null;
}
return info;
}
} catch (NamingException e) {
Log.warning(Geonet.LDAP, "Raised exception during LDAP access");
Log.warning(Geonet.LDAP, " (C) Message :" + e.getMessage());
return null;
}
}
public int evaluate(int timeout) {
try {
if (Log.isDebugEnabled(Geonet.Z3950_SERVER))
Log.debug(Geonet.Z3950_SERVER, "INCOMING XML QUERY:\n" + query);
Element request = new Element("request");
request.addContent(query.toGNXMLRep());
List<String> categories = query.getCollections();
for (String category : categories) {
if (!category.equals("geonetwork") && !category.equals("Default"))
request.addContent(new Element("category").setText(category));
}
ServiceConfig config = new ServiceConfig();
// perform the search and save search results
metasearcher.search(this.srvxtx, request, config);
// System.out.println("summary:\n" + Xml.getString(s.getSummary()));
// // DEBUG
// Random number of records.. Set up the result set
setFragmentCount(metasearcher.getSize());
setTaskStatusCode(IRResultSetStatus.COMPLETE);
this.srvxtx.getResourceManager().close();
} catch (Throwable e) {
Log.error(Geonet.Z3950_SERVER, "error evaluating query.." + e);
e.printStackTrace();
try {
this.srvxtx.getResourceManager().abort();
} catch (Exception e2) {
e2.printStackTrace();
}
}
return (getStatus());
}
private String[] getAll(Map<String, ? extends List<Object>> attr, String name) {
List<Object> values = attr.get(name);
if (values == null) {
if (Log.isDebugEnabled(Geonet.LDAP))
Log.debug(Geonet.LDAP, "Attribute '" + name + "' does not exist");
return null;
}
ArrayList<String> objs = new ArrayList<String>();
for (Object obj : values) {
if (obj != null) {
if (Log.isDebugEnabled(Geonet.LDAP)) {
Log.debug(
Geonet.LDAP,
"Attribute '" + name + "' is of type : " + obj.getClass().getSimpleName());
}
objs.add(obj.toString());
} else {
if (Log.isDebugEnabled(Geonet.LDAP)) {
Log.debug(Geonet.LDAP, "Attribute '" + name + "' is null");
}
}
}
return objs.toArray(new String[0]);
}
/**
* For Editing : adds an attribute from a metadata ([add] link). FIXME: Modify and use within Ajax
* controls
*
* @param dbms
* @param id
* @param ref
* @param name
* @param currVersion
* @return
* @throws Exception
*/
public synchronized boolean addAttribute(
Dbms dbms, String id, String ref, String name, String currVersion) throws Exception {
Element md = xmlSerializer.select(dbms, "Metadata", id);
// --- check if the metadata has been deleted
if (md == null) return false;
String schema = dataManager.getMetadataSchema(dbms, id);
EditLib editLib = dataManager.getEditLib();
editLib.expandElements(schema, md);
editLib.enumerateTree(md);
// --- get element to add
Element el = editLib.findElement(md, ref);
if (el == null) Log.error(Geonet.DATA_MANAGER, MSG_ELEMENT_NOT_FOUND_AT_REF + ref);
// throw new IllegalStateException("Element not found at ref = " + ref);
// --- remove editing info added by previous call
editLib.removeEditingInfo(md);
if (el != null) {
el.setAttribute(new Attribute(name, ""));
}
editLib.contractElements(md);
String parentUuid = null;
md =
dataManager.updateFixedInfo(
schema, id, null, md, parentUuid, DataManager.UpdateDatestamp.no, dbms);
String changeDate = null;
xmlSerializer.update(dbms, id, md, changeDate, false, context);
// Notifies the metadata change to metatada notifier service
dataManager.notifyMetadataChange(dbms, md, id);
// --- update search criteria
boolean workspace = false;
dataManager.indexInThreadPoolIfPossible(dbms, id, workspace);
return true;
}
public Element execute(Element request, ServiceContext context) throws CatalogException {
checkService(request);
checkVersion(request);
Element response = new Element(getName() + "Response", Csw.NAMESPACE_CSW);
String[] propertyNames = getParameters(request, "PropertyName");
String[] parameterNames = getParameters(request, "ParameterName");
String cswServiceSpecificConstraint = request.getChildText(Geonet.Elem.FILTER);
// PropertyName handled first.
if (propertyNames != null) {
List<Element> domainValues;
try {
domainValues =
handlePropertyName(
propertyNames,
context,
false,
CatalogConfiguration.getMaxNumberOfRecordsForPropertyNames(),
cswServiceSpecificConstraint,
_luceneConfig);
} catch (Exception e) {
Log.error(Geonet.CSW, "Error getting domain value for specified PropertyName : " + e);
throw new NoApplicableCodeEx(
"Raised exception while getting domain value for specified PropertyName : " + e);
}
response.addContent(domainValues);
return response;
}
if (parameterNames != null) {
List<Element> domainValues = handleParameterName(parameterNames);
response.addContent(domainValues);
}
return response;
}
private String get(Map<String, ? extends List<Object>> attr, String name) {
List<Object> values = attr.get(name);
if (values == null) {
if (Log.isDebugEnabled(Geonet.LDAP))
Log.debug(Geonet.LDAP, "Attribute '" + name + "' does not exist");
return null;
}
Object obj = values.get(0);
if (obj != null)
if (Log.isDebugEnabled(Geonet.LDAP))
Log.debug(
Geonet.LDAP, "Attribute '" + name + "' is of type : " + obj.getClass().getSimpleName());
else if (Log.isDebugEnabled(Geonet.LDAP))
Log.debug(Geonet.LDAP, "Attribute '" + name + "' is null");
return (obj == null) ? null : obj.toString();
}
protected void debug(String message) {
Log.debug(Log.DBMSPOOL, message);
}
public void setService(String service) {
this.service = service;
logger = Log.createLogger(Log.WEBAPP + "." + service);
}
public Element execute(LocalServiceRequest request) throws Exception {
ServiceContext context =
new ServiceContext(
request.getService(),
getApplicationContext(),
getXmlCacheManager(),
getMonitorManager(),
getProviderManager(),
getSerialFactory(),
getProfileManager(),
htContexts) {
/* This override causes database connections to be consumed.....
Comment out for now. sppigot May, 2014
public ResourceManager getResourceManager() {
return new ResourceManager(getMonitorManager(), getProviderManager()) {
@Override
public synchronized void abort() throws Exception {
}
@Override
public synchronized void close() throws Exception {
}
@Override
public synchronized void close(String name, Object resource)
throws Exception {
}
@Override
public synchronized void abort(String name, Object resource)
throws Exception {
}
@Override
protected void openMetrics(Object resource) {
}
@Override
protected void closeMetrics(Object resource) {
}
};
}
*/
};
UserSession session = userSession;
if (userSession == null) {
session = new UserSession();
}
try {
servlet.getEngine().getServiceManager().dispatch(request, session, context);
} catch (Exception e) {
Log.error(Log.XLINK_PROCESSOR, "Failed to parse result xml" + request.getService());
throw new ServiceExecutionFailedException(request.getService(), e);
} finally {
// set old context back as thread local
setAsThreadLocal();
}
try {
return request.getResult();
} catch (Exception e) {
Log.error(
Log.XLINK_PROCESSOR,
"Failed to parse result xml from service:"
+ request.getService()
+ "\n"
+ request.getResultString());
throw new ServiceExecutionFailedException(request.getService(), e);
}
}
/**
* @param request
* @param xml
* @param context
* @param toIndex
* @return
* @throws Exception
*/
private int updateTransaction(
Element request, Element xml, ServiceContext context, Set<String> toIndex) throws Exception {
GeonetContext gc = (GeonetContext) context.getHandlerContext(Geonet.CONTEXT_NAME);
DataManager dataMan = gc.getDataManager();
if (context.getUserSession().getUserId() == null)
throw new NoApplicableCodeEx("User not authenticated.");
int totalUpdated = 0;
// Update full metadata
if (xml != null) {
// Retrieve schema and the related Namespaces
String schemaId = gc.getSchemamanager().autodetectSchema(xml);
if (schemaId == null) {
throw new NoApplicableCodeEx("Can't identify metadata schema");
}
// Retrieve the metadata identifier
Dbms dbms = (Dbms) context.getResourceManager().open(Geonet.Res.MAIN_DB);
String uuid = gc.getDataManager().extractUUID(schemaId, xml);
if (uuid.length() == 0) {
throw new NoApplicableCodeEx("Metadata identifier not provided");
}
// Update metadata record
String id = dataMan.getMetadataId(dbms, uuid);
if (id == null) return totalUpdated;
if (!gc.getAccessManager().canEdit(context, id))
throw new NoApplicableCodeEx("User not allowed to update this metadata(" + id + ").");
String changeDate = null;
boolean validate = false;
boolean ufo = false;
boolean index = false;
String language = context.getLanguage();
dataMan.updateMetadata(
context, dbms, id, xml, validate, ufo, index, language, changeDate, false);
dbms.commit();
toIndex.add(id);
totalUpdated++;
return totalUpdated;
// Update properties
} else {
// first, search the record in the database to get the record id
Element constr = (Element) request.getChild("Constraint", Csw.NAMESPACE_CSW).clone();
List<Element> results = getResultsFromConstraints(context, constr);
List<Element> recordProperties =
(List<Element>) request.getChildren("RecordProperty", Csw.NAMESPACE_CSW);
Iterator<Element> it = results.iterator();
if (!it.hasNext()) return totalUpdated;
Dbms dbms = (Dbms) context.getResourceManager().open(Geonet.Res.MAIN_DB);
Set updatedMd = new HashSet<String>();
// Process all records selected
while (it.hasNext()) {
Element result = it.next();
String uuid = result.getChildText("identifier", Csw.NAMESPACE_DC);
String id = dataMan.getMetadataId(dbms, uuid);
String changeDate = null;
if (id == null) continue;
if (!dataMan.getAccessManager().canEdit(context, id))
throw new NoApplicableCodeEx("User not allowed to update this metadata(" + id + ").");
Element metadata = dataMan.getMetadata(context, id, false, false, true);
metadata.removeChild("info", Edit.NAMESPACE);
// Retrieve the schema and Namespaces of metadata to update
String schemaId = gc.getDataManager().autodetectSchema(metadata);
if (schemaId == null) {
throw new NoApplicableCodeEx("Can't identify metadata schema");
}
Map mapNs = retrieveNamepacesForSchema(gc.getDataManager().getSchema(schemaId));
boolean metadataChanged = false;
// Process properties to update
for (Element recordProperty : recordProperties) {
Element propertyNameEl = recordProperty.getChild("Name", Csw.NAMESPACE_CSW);
Element propertyValueEl = recordProperty.getChild("Value", Csw.NAMESPACE_CSW);
String propertyName = propertyNameEl.getText();
String propertyValue = propertyValueEl.getText();
// Get XPath for queriable name, i provided in propertyName.
// Otherwise assume propertyName contains full XPath to property to update
String xpathProperty = FieldMapper.mapXPath(propertyName, schemaId);
if (xpathProperty == null) {
xpathProperty = propertyName;
}
Log.info(Geonet.CSW, "Xpath of property: " + xpathProperty);
XPath xpath = new JDOMXPath(xpathProperty);
xpath.setNamespaceContext(new SimpleNamespaceContext(mapNs));
Object propEl = xpath.selectSingleNode(metadata);
Log.info(Geonet.CSW, "XPath found in metadata: " + (propEl != null));
// If a property is not found in metadata, just ignore it.
if (propEl != null) {
if (propEl instanceof Element) {
((Element) propEl).setText(propertyValue);
metadataChanged = true;
} else if (propEl instanceof Attribute) {
((Attribute) propEl).setValue(propertyValue);
metadataChanged = true;
}
}
} // for(Element recordProperty : recordProperties)
// Update the metadata with changes
if (metadataChanged) {
boolean validate = false;
boolean ufo = false;
boolean index = false;
String language = context.getLanguage();
dataMan.updateMetadata(
context, dbms, id, metadata, validate, ufo, index, language, changeDate, false);
updatedMd.add(id);
totalUpdated++;
}
}
dbms.commit();
toIndex.addAll(updatedMd);
return totalUpdated;
}
}
public Element execute(Element request, ServiceContext context) throws CatalogException {
checkService(request);
checkVersion(request);
// num counter
int totalInserted = 0;
int totalUpdated = 0;
int totalDeleted = 0;
// Response element
Element response = new Element(getName() + "Response", Csw.NAMESPACE_CSW);
List<String> strFileIds = new ArrayList<String>();
// process the transaction from the first to the last
List<Element> childList = request.getChildren();
GeonetContext gc = (GeonetContext) context.getHandlerContext(Geonet.CONTEXT_NAME);
DataManager dataMan = gc.getDataManager();
Set<String> toIndex = new HashSet<String>();
try {
// process the childlist
for (Element transRequest : childList) {
String transactionType = transRequest.getName().toLowerCase();
if (transactionType.equals("insert")
|| transactionType.equals("update")
|| transactionType.equals("delete")) {
List<Element> mdList = transRequest.getChildren();
// insert to database, and get the number of inserted successful
if (transactionType.equals("insert")) {
Iterator<Element> inIt = mdList.iterator();
while (inIt.hasNext()) {
Element metadata = (Element) inIt.next().clone();
boolean insertSuccess = insertTransaction(metadata, strFileIds, context, toIndex);
if (insertSuccess) {
totalInserted++;
}
}
}
// Update
else if (transactionType.equals("update")) {
Iterator<Element> inIt = mdList.iterator();
Element metadata = null;
while (inIt.hasNext()) {
Element reqElem = (Element) inIt.next();
if (reqElem.getNamespace() != Csw.NAMESPACE_CSW) {
metadata = (Element) reqElem.clone();
}
}
totalUpdated = updateTransaction(transRequest, metadata, context, toIndex);
}
// Delete
else {
totalDeleted = deleteTransaction(transRequest, context);
}
}
}
} catch (Exception e) {
Log.error(Geonet.CSW, "Cannot process transaction");
Log.error(Geonet.CSW, " (C) StackTrace\n" + Util.getStackTrace(e));
throw new NoApplicableCodeEx("Cannot process transaction: " + e.getMessage());
} finally {
try {
dataMan.indexInThreadPool(context, new ArrayList<String>(toIndex), null);
} catch (SQLException e) {
Log.error(Geonet.CSW, "cannot index");
Log.error(Geonet.CSW, " (C) StackTrace\n" + Util.getStackTrace(e));
}
getResponseResult(request, response, strFileIds, totalInserted, totalUpdated, totalDeleted);
}
return response;
}
public static List<Element> handlePropertyName(
String[] propertyNames,
ServiceContext context,
boolean freq,
int maxRecords,
String cswServiceSpecificConstraint,
LuceneConfig luceneConfig)
throws Exception {
List<Element> domainValuesList = null;
if (Log.isDebugEnabled(Geonet.CSW))
Log.debug(
Geonet.CSW,
"Handling property names '"
+ Arrays.toString(propertyNames)
+ "' with max records of "
+ maxRecords);
for (int i = 0; i < propertyNames.length; i++) {
if (i == 0) domainValuesList = new ArrayList<Element>();
// Initialize list of values element.
Element listOfValues = null;
// Generate DomainValues element
Element domainValues = new Element("DomainValues", Csw.NAMESPACE_CSW);
// FIXME what should be the type ???
domainValues.setAttribute("type", "csw:Record");
String property = propertyNames[i].trim();
// Set propertyName in any case.
Element pn = new Element("PropertyName", Csw.NAMESPACE_CSW);
domainValues.addContent(pn.setText(property));
GeonetContext gc = (GeonetContext) context.getHandlerContext(Geonet.CONTEXT_NAME);
SearchManager sm = gc.getSearchmanager();
IndexAndTaxonomy indexAndTaxonomy = sm.getNewIndexReader(null);
try {
GeonetworkMultiReader reader = indexAndTaxonomy.indexReader;
BooleanQuery groupsQuery = (BooleanQuery) CatalogSearcher.getGroupsQuery(context);
BooleanQuery query = null;
// Apply CSW service specific constraint
if (StringUtils.isNotEmpty(cswServiceSpecificConstraint)) {
Query constraintQuery =
CatalogSearcher.getCswServiceSpecificConstraintQuery(
cswServiceSpecificConstraint, luceneConfig);
query = new BooleanQuery();
BooleanClause.Occur occur = LuceneUtils.convertRequiredAndProhibitedToOccur(true, false);
query.add(groupsQuery, occur);
query.add(constraintQuery, occur);
} else {
query = groupsQuery;
}
List<Pair<String, Boolean>> sortFields =
Collections.singletonList(Pair.read(Geonet.SearchResult.SortBy.RELEVANCE, true));
Sort sort = LuceneSearcher.makeSort(sortFields, context.getLanguage(), false);
CachingWrapperFilter filter = null;
Pair<TopDocs, Element> searchResults =
LuceneSearcher.doSearchAndMakeSummary(
maxRecords,
0,
maxRecords,
context.getLanguage(),
null,
reader,
query,
filter,
sort,
null,
false,
false,
false,
false // Scoring is useless for GetDomain operation
);
TopDocs hits = searchResults.one();
try {
// Get mapped lucene field in CSW configuration
String indexField = CatalogConfiguration.getFieldMapping().get(property.toLowerCase());
if (indexField != null) property = indexField;
// check if params asked is in the index using getFieldNames ?
FieldInfos fi = new SlowCompositeReaderWrapper(reader).getFieldInfos();
if (fi.fieldInfo(property) == null) continue;
boolean isRange = false;
if (CatalogConfiguration.getGetRecordsRangeFields().contains(property)) isRange = true;
if (isRange) listOfValues = new Element("RangeOfValues", Csw.NAMESPACE_CSW);
else listOfValues = new Element("ListOfValues", Csw.NAMESPACE_CSW);
Set<String> fields = new HashSet<String>();
fields.add(property);
fields.add("_isTemplate");
// parse each document in the index
String[] fieldValues;
SortedSet<String> sortedValues = new TreeSet<String>();
HashMap<String, Integer> duplicateValues = new HashMap<String, Integer>();
for (int j = 0; j < hits.scoreDocs.length; j++) {
DocumentStoredFieldVisitor selector = new DocumentStoredFieldVisitor(fields);
reader.document(hits.scoreDocs[j].doc, selector);
Document doc = selector.getDocument();
// Skip templates and subTemplates
String[] isTemplate = doc.getValues("_isTemplate");
if (isTemplate[0] != null && !isTemplate[0].equals("n")) continue;
// Get doc values for specified property
fieldValues = doc.getValues(property);
if (fieldValues == null) continue;
addtoSortedSet(sortedValues, fieldValues, duplicateValues);
}
SummaryComparator valuesComparator =
new SummaryComparator(SortOption.FREQUENCY, Type.STRING, context.getLanguage(), null);
TreeSet<Map.Entry<String, Integer>> sortedValuesFrequency =
new TreeSet<Map.Entry<String, Integer>>(valuesComparator);
sortedValuesFrequency.addAll(duplicateValues.entrySet());
if (freq) return createValuesByFrequency(sortedValuesFrequency);
else listOfValues.addContent(createValuesElement(sortedValues, isRange));
} finally {
// any children means that the catalog was unable to determine
// anything about the specified parameter
if (listOfValues != null && listOfValues.getChildren().size() != 0)
domainValues.addContent(listOfValues);
// Add current DomainValues to the list
domainValuesList.add(domainValues);
}
} finally {
sm.releaseIndexReader(indexAndTaxonomy);
}
}
return domainValuesList;
}
private void execute(HttpServletRequest req, HttpServletResponse res) throws IOException {
String ip = req.getRemoteAddr();
// if we do have the optional x-forwarded-for request header then
// use whatever is in it to record ip address of client
String forwardedFor = req.getHeader("x-forwarded-for");
if (forwardedFor != null) ip = forwardedFor;
Log.info(Log.REQUEST, "==========================================================");
Log.info(Log.REQUEST, "HTML Request (from " + ip + ") : " + req.getRequestURI());
Log.debug(Log.REQUEST, "Method : " + req.getMethod());
Log.debug(Log.REQUEST, "Content type : " + req.getContentType());
// Log.debug(Log.REQUEST, "Context path : "+ req.getContextPath());
// Log.debug(Log.REQUEST, "Char encoding: "+ req.getCharacterEncoding());
Log.debug(Log.REQUEST, "Accept : " + req.getHeader("Accept"));
// Log.debug(Log.REQUEST, "Server name : "+ req.getServerName());
// Log.debug(Log.REQUEST, "Server port : "+ req.getServerPort());
// for (Enumeration e = req.getHeaderNames(); e.hasMoreElements();) {
// String theHeader = (String)e.nextElement();
// Log.debug(Log.REQUEST, "Got header: "+theHeader);
// Log.debug(Log.REQUEST, "With value: "+req.getHeader(theHeader));
// }
HttpSession httpSession = req.getSession();
Log.debug(Log.REQUEST, "Session id is " + httpSession.getId());
UserSession session = (UserSession) httpSession.getAttribute("session");
// ------------------------------------------------------------------------
// --- create a new session if doesn't exist
if (session == null) {
// --- create session
session = new UserSession();
httpSession.setAttribute("session", session);
Log.debug(Log.REQUEST, "Session created for client : " + ip);
}
// ------------------------------------------------------------------------
// --- build service request
ServiceRequest srvReq = null;
// --- create request
try {
srvReq =
ServiceRequestFactory.create(req, res, jeeves.getUploadDir(), jeeves.getMaxUploadSize());
} catch (FileUploadTooBigEx e) {
StringBuffer sb = new StringBuffer();
sb.append("File upload too big - exceeds " + jeeves.getMaxUploadSize() + " Mb\n");
sb.append("Error : " + e.getClass().getName() + "\n");
res.sendError(400, sb.toString());
// now stick the stack trace on the end and log the whole lot
sb.append("Stack :\n");
sb.append(Util.getStackTrace(e));
Log.error(Log.REQUEST, sb.toString());
return;
} catch (Exception e) {
StringBuffer sb = new StringBuffer();
sb.append("Cannot build ServiceRequest\n");
sb.append("Cause : " + e.getMessage() + "\n");
sb.append("Error : " + e.getClass().getName() + "\n");
res.sendError(400, sb.toString());
// now stick the stack trace on the end and log the whole lot
sb.append("Stack :\n");
sb.append(Util.getStackTrace(e));
Log.error(Log.REQUEST, sb.toString());
return;
}
// --- execute request
jeeves.dispatch(srvReq, session);
}
/**
* TODO javadoc.
*
* @param session
* @param md
* @param id
*/
private void setMetadataIntoSession(UserSession session, Element md, String id) {
if (Log.isDebugEnabled(Geonet.EDITOR)) {
Log.debug(Geonet.EDITOR, "Storing metadata in session " + session.getUserId());
}
session.setProperty(Geonet.Session.METADATA_EDITING + id, md);
}
/**
* For Ajax Editing : removes metadata from session.
*
* @param session
* @param id
*/
public void removeMetadataEmbedded(UserSession session, String id) {
if (Log.isDebugEnabled(Geonet.EDITOR))
Log.debug(Geonet.EDITOR, "Removing metadata from session " + session.getUserId());
session.removeProperty(Geonet.Session.METADATA_EDITING + id);
session.removeProperty(Geonet.Session.VALIDATION_REPORT + id);
}
protected void info(String message) {
Log.info(Log.DBMSPOOL, message);
}
/**
* Apply a list of changes to the metadata record in current editing session.
*
* <p>The changes are a list of KVP. A key contains at least the element identifier from the
* meta-document. A key starting with an "X" should contain an XML fragment for the value. The
* following KVP combinations are allowed:
*
* <ul>
* <li>ElementId=ElementValue
* <li>ElementId_AttributeName=AttributeValue
* <li>ElementId_AttributeNamespacePrefixCOLONAttributeName=AttributeValue
* <li>XElementId=ElementValue
* <li>XElementId_ElementName=ElementValue
* </ul>
*
* ElementName MUST contain "{@value #COLON_SEPARATOR}" instead of ":" for prefixed elements.
*
* <p>When using X key, value could contains many XML fragments (eg. <gmd:keywords
* .../>{@value #XML_FRAGMENT_SEPARATOR}<gmd:keywords .../>) separated by {@link
* #XML_FRAGMENT_SEPARATOR}. All those fragments are inserted to the last element of this type in
* its parent if ElementName is set. If not, the element with ElementId is replaced.
*
* <p>
*
* @param dbms
* @param id Metadata internal identifier.
* @param changes List of changes to apply.
* @return The update metadata record
* @throws Exception
*/
protected Element applyChangesEmbedded(Dbms dbms, String id, Hashtable changes) throws Exception {
String schema = dataManager.getMetadataSchema(dbms, id);
EditLib editLib = dataManager.getEditLib();
// --- get metadata from session
Element md = getMetadataFromSession(session, id);
// Store XML fragments to be handled after other elements update
Map<String, String> xmlInputs = new HashMap<String, String>();
// --- update elements
for (Enumeration e = changes.keys(); e.hasMoreElements(); ) {
String ref = ((String) e.nextElement()).trim();
String value = ((String) changes.get(ref)).trim();
String attribute = null;
// Avoid empty key
if (ref.equals("")) {
continue;
}
// Catch element starting with a X to replace XML fragments
if (ref.startsWith("X")) {
ref = ref.substring(1);
xmlInputs.put(ref, value);
continue;
}
if (updatedLocalizedTextElement(md, ref, value, editLib)) {
continue;
}
int at = ref.indexOf('_');
if (at != -1) {
attribute = ref.substring(at + 1);
ref = ref.substring(0, at);
}
Element el = editLib.findElement(md, ref);
if (el == null) {
Log.error(Geonet.EDITOR, MSG_ELEMENT_NOT_FOUND_AT_REF + ref);
continue;
}
// Process attribute
if (attribute != null) {
Pair<Namespace, String> attInfo =
parseAttributeName(attribute, COLON_SEPARATOR, id, md, dbms, editLib);
String localname = attInfo.two();
Namespace attrNS = attInfo.one();
if (el.getAttribute(localname, attrNS) != null) {
el.setAttribute(new Attribute(localname, value, attrNS));
}
} else {
// Process element value
List content = el.getContent();
for (int i = 0; i < content.size(); i++) {
if (content.get(i) instanceof Text) {
el.removeContent((Text) content.get(i));
i--;
}
}
el.addContent(value);
}
}
// Deals with XML fragments to insert or update
if (!xmlInputs.isEmpty()) {
// Loop over each XML fragments to insert or replace
for (String ref : xmlInputs.keySet()) {
String value = xmlInputs.get(ref);
String name = null;
int addIndex = ref.indexOf('_');
if (addIndex != -1) {
name = ref.substring(addIndex + 1);
ref = ref.substring(0, addIndex);
}
// Get element to fill
Element el = editLib.findElement(md, ref);
if (el == null) {
Log.error(Geonet.EDITOR, MSG_ELEMENT_NOT_FOUND_AT_REF + ref);
continue;
}
if (value != null && !value.equals("")) {
String[] fragments = value.split(XML_FRAGMENT_SEPARATOR);
for (String fragment : fragments) {
if (name != null) {
if (Log.isDebugEnabled(Geonet.EDITOR))
Log.debug(
Geonet.EDITOR,
"Add XML fragment; " + fragment + " to element with ref: " + ref);
name = name.replace(COLON_SEPARATOR, ":");
editLib.addFragment(schema, el, name, fragment);
} else {
if (Log.isDebugEnabled(Geonet.EDITOR))
Log.debug(
Geonet.EDITOR,
"Add XML fragment; "
+ fragment
+ " to element with ref: "
+ ref
+ " replacing content.");
// clean before update
el.removeContent();
fragment = addNamespaceToFragment(fragment);
// Add content
el.addContent(Xml.loadString(fragment, false));
}
}
}
}
}
// --- remove editing info
editLib.removeEditingInfo(md);
editLib.contractElements(md);
return (Element) md.detach();
}
protected void error(String message) {
Log.error(Log.DBMSPOOL, message);
}
// --------------------------------------------------------------------------
// ---
// --- Service
// ---
// --------------------------------------------------------------------------
@Override
public Element serviceSpecificExec(Element params, ServiceContext context) throws Exception {
boolean readOnlyMode = super.exec(params, context) == null;
if (readOnlyMode) {
return null;
}
String message = "";
GeonetContext gc = (GeonetContext) context.getHandlerContext(Geonet.CONTEXT_NAME);
// gets the total popularity count (=100)
Dbms dbms = (Dbms) context.getResourceManager().open(Geonet.Res.MAIN_DB);
// wont work if there is no metadata
List l = dbms.select("select sum(popularity) as sumpop from metadata").getChildren();
if (l.size() != 1) {
message = "cannot get popularity count";
return null;
}
int cnt = Integer.parseInt(((Element) l.get(0)).getChildText("sumpop"));
if (Log.isDebugEnabled(Geonet.SEARCH_LOGGER))
Log.debug(Geonet.SEARCH_LOGGER, "query to get popularity by group:\n" + query);
dbms = (Dbms) context.getResourceManager().open(Geonet.Res.MAIN_DB);
DefaultPieDataset dataset = new DefaultPieDataset();
List resultSet = dbms.select(query).getChildren();
for (int i = 0; i < resultSet.size(); i++) {
Element record = (Element) resultSet.get(i);
String popularity = (record).getChildText("popu");
Double d = 0.0;
if (popularity.length() > 0) {
d = (Double.parseDouble(popularity) / cnt) * 100;
}
dataset.setValue(record.getChildText("source"), d);
// System.out.println(record.getChildText("groupname") + ", " + d);
}
// create a chart...
JFreeChart chart =
ChartFactory.createPieChart(
null,
dataset,
true, // legend?
true, // tooltips?
false // URLs?
);
// hard coded values for the moment. should come from a configuration file.
chart.setBackgroundPaint(Color.decode("#E7EDF5"));
String chartFilename = "popubycatalog_" + System.currentTimeMillis() + ".png";
File statFolder =
new File(
gc.getHandlerConfig().getMandatoryValue(Geonet.Config.RESOURCES_DIR)
+ File.separator
+ "images"
+ File.separator
+ "statTmp");
if (!statFolder.exists()) {
statFolder.mkdirs();
}
File f = new File(statFolder, chartFilename);
this.imageMap =
org.fao.geonet.services.statistics.ChartFactory.writeChartImage(
chart,
f,
this.chartWidth,
this.chartHeight,
this.createTooltips,
"graphPopuByCatalogImageMap");
// will return some info to the XSLT:
// dateFrom, dateTo, graphicType, chartUrl, tooltipImageMap,
// message, chartWidth, chartHeight
Element elResp = new Element(Jeeves.Elem.RESPONSE);
Element elchartUrl =
new Element("popuByCatalogUrl")
.setText(context.getBaseUrl() + "/images/statTmp/" + chartFilename);
Element elTooltipImageMap =
new Element("tooltipImageMap").addContent(this.createTooltips ? this.imageMap : "");
Element elMessage = new Element("message").setText(message);
Element elChartWidth = new Element("chartWidth").setText("" + this.chartWidth);
Element elChartHeight = new Element("chartHeight").setText("" + this.chartHeight);
elResp.addContent(elchartUrl);
elResp.addContent(elTooltipImageMap);
elResp.addContent(elMessage);
elResp.addContent(elChartWidth);
elResp.addContent(elChartHeight);
return elResp;
}
protected void warning(String message) {
Log.warning(Log.DBMSPOOL, message);
}
