/**
* Copyright (c) 2009, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.wso2.carbon.mediation.initializer.persistence.registry;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.impl.builder.StAXOMBuilder;
import org.wso2.carbon.registry.core.exceptions.RegistryException;
import org.wso2.carbon.registry.core.session.UserRegistry;
import org.wso2.carbon.registry.core.Resource;
import org.wso2.carbon.registry.core.RegistryConstants;
import org.wso2.carbon.mediation.initializer.ServiceBusConstants;
import org.wso2.carbon.mediation.initializer.persistence.ServiceBusPersistenceException;
import javax.xml.stream.XMLStreamException;
import java.util.Collection;
import java.util.ArrayList;
/**
*
*/
public abstract class AbstractRegistryStore {
protected Log log = LogFactory.getLog(AbstractRegistryStore.class);
protected UserRegistry registry;
protected String configName;
protected AbstractRegistryStore(UserRegistry registry, String configName) {
this.registry = registry;
this.configName = configName;
}
public abstract Collection<OMElement> getElements();
public abstract OMElement getElement(String name);
public abstract void persistElement(String name, OMElement element, String fileName);
public abstract void deleteElement(String name);
protected void createCollection(String path) {
try {
if (!registry.resourceExists(path)) {
org.wso2.carbon.registry.core.Collection collection = registry.newCollection();
registry.put(path, collection);
}
} catch (RegistryException e) {
handleException("Unable to create the collection : " + path, e);
}
}
protected void persistElement(OMElement element, String resourcePath, String fileName)
throws RegistryException {
try {
registry.beginTransaction();
if (registry.resourceExists(resourcePath)) {
registry.delete(resourcePath);
}
Resource resource = registry.newResource();
resource.setContent(element.toString());
if (fileName != null) {
resource.setProperty(ServiceBusConstants.DEFINITION_FILE_NAME, fileName);
}
resource.setMediaType("text/xml");
registry.put(resourcePath, resource);
registry.commitTransaction();
} catch (Exception e) {
registry.rollbackTransaction();
throw new RegistryException("Unable to persist element", e);
}
}
protected void handleException(String message, Exception e) {
log.error(message, e);
throw new ServiceBusPersistenceException(message, e);
}
protected void handleException(String message) {
log.error(message);
throw new ServiceBusPersistenceException(message);
}
protected Collection<OMElement> getChildElementsInPath(String path) throws RegistryException {
Collection<OMElement> childElements = new ArrayList<OMElement>();
if (registry.resourceExists(path)) {
Resource resource = registry.get(path);
if (resource instanceof org.wso2.carbon.registry.core.Collection) {
org.wso2.carbon.registry.core.Collection collection
= (org.wso2.carbon.registry.core.Collection) resource;
String[] children = collection.getChildren();
for (String resourcePath : children) {
OMElement resourceElem = getResourceAsElement(resourcePath);
if (resourceElem != null) {
childElements.add(resourceElem);
}
}
}
} else {
if (log.isDebugEnabled()) {
log.debug("Collection at path " + path + " doesn't exists in the registry");
}
}
return childElements;
}
protected OMElement getResourceAsElement(String resourcePath) throws RegistryException {
OMElement resourceElem = null;
Resource childResource = registry.get(resourcePath);
try {
resourceElem = new StAXOMBuilder(
childResource.getContentStream()).getDocumentElement();
resourceElem.build();
String fileNameProperty
= childResource.getProperty(ServiceBusConstants.DEFINITION_FILE_NAME);
if (fileNameProperty != null) {
resourceElem.addAttribute(ServiceBusConstants.DEFINITION_FILE_NAME,
fileNameProperty, resourceElem.getOMFactory().createOMNamespace("", ""));
}
} catch (XMLStreamException e) {
log.warn("Resource loadded from the registry with path : "
+ resourcePath + " is not well formed XML");
}
return resourceElem;
}
protected String getConfigurationRoot() {
return ServiceBusConstants.SYNAPSE_CONFIG_REGISTRY_SPACE +
RegistryConstants.PATH_SEPARATOR + configName;
}
protected abstract String getConfigurationPath();
}