def resource_delete(self, id, resource_id):
''' Before deleting a resource from CKAN, we request WSO2 ESB to remove the subscription from the related Topic. '''
import json
subscription_id = None
brokerclient = getBrokerClient()
package_data = get_action('package_show')({'model': model, 'session': model.Session, 'user': c.user or c.author, 'auth_user_obj': c.userobj}, {'id': id})
topic = getTopicFromPackageData(package_data)
resource_url = getResourceUrlFromName(resource_id)
# Cycle through subscriptions until the right one is found, then select its id.
subscriptions = brokerclient.getAllSubscriptions()
subscriptions = json.loads(subscriptions)
for subscription in subscriptions:
# Note here that CKAN appends http:// prefix into URLs if it's not present upon resource creation, causing mismatch here - so we must take that to account.
if subscription['localTopic'] == '/' + topic and subscription['localEventSinkAddress'] == resource_url or "http://" + subscription['localEventSinkAddress'] == resource_url:
subscription_id = subscription['localSubscriptionId']
# Since we now have access to the subscription id, simply use it to unsubscribe.
try:
brokerclient.unsubscribe(subscription_id)
except Py4JJavaError, e:
error_message = str(e)
if 'org.apache.axis2.AxisFault: java.lang.NullPointerException' in error_message:
h.flash_error(_('Warning: removed subscription was not found on the WSO2 ESB.'))
else:
raise e
def new_resource(self, id, data=None, errors=None, error_summary=None):
''' Before creating a resource into CKAN, we request WSO2 ESB to add a subscription to the related Topic. '''
if request.method == 'POST' and not data:
# Recogniced new resource form POST, extract variables.
postdata = data or clean_dict(unflatten(tuplize_dict(parse_params(request.POST))))
if 'save' in postdata and 'url' in postdata:
package_data = get_action('package_show')({'model': model, 'session': model.Session, 'user': c.user or c.author, 'auth_user_obj': c.userobj}, {'id': id})
topic = getTopicFromPackageData(package_data)
# Add a new subscription for the topic named after the dataset, pointing to the URL given.
brokerclient = getBrokerClient()
try:
result = brokerclient.subscribe(topic, postdata['url'])
except Py4JJavaError, e:
# Errors are propagated to the CKAN controller below to prevent new resource creation.
error_message = str(e)
if 'Error While Subscribing :Cannot initialize URI with empty parameters.' in error_message:
error_message = _('Error While Subscribing: Cannot initialize URI with empty parameters.')
if 'org.apache.axis2.AxisFault: Connection refused' in error_message:
error_message = _('Error While Subscribing: Cannot connect to WSO2 ESB.')
if errors and isinstance(errors, dict):
errors.update({ 'error': error_message })
else:
errors = { 'error': error_message }
data = postdata
error_summary = { _('WSO2 ESB'): error_message }
def wso2esb(self):
''' Queries WSO2 ESB for information. '''
try:
brokerclient = getBrokerClient()
subscriptions = brokerclient.getAllSubscriptions()
c.subscriptions = json.loads(subscriptions)
except Py4JJavaError, e:
h.flash_error(str(e))
def publish(self, id):
''' Publishes (sends) a message to WSO2 ESB Topic related to the dataset. '''
data = clean_dict(unflatten(tuplize_dict(parse_params(request.POST))))
if 'message' in data and isinstance(data['message'], basestring):
if c.userobj.sysadmin:
package_data = get_action('package_show')({'model': model, 'session': model.Session, 'user': c.user or c.author, 'auth_user_obj': c.userobj}, {'id': id})
topic = getTopicFromPackageData(package_data)
brokerclient = getBrokerClient()
if brokerclient.publish(topic, data['message']):
h.flash_notice(_('The message was sent.'))
else:
h.flash_error(_('Error sending the message.'))
else:
h.flash_error(_('Error: sysadmin rights required to send a message.'))
else:
h.flash_error(_('Error: no message was provided.'))
return super(package, self).read(id)
