org.overlord.dtgov.jbpm.util.HttpClientWorkItemHandler.java Source code

Introduction

Here is the source code for org.overlord.dtgov.jbpm.util.HttpClientWorkItemHandler.java

Source

/**
 * Copyright 2013 JBoss Inc
 *
 * 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.overlord.dtgov.jbpm.util;

import java.net.URLEncoder;
import java.util.HashMap;
import java.util.Map;

import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.impl.client.DefaultHttpClient;
import org.jboss.resteasy.client.ClientRequest;
import org.jboss.resteasy.client.ClientResponse;
import org.jboss.resteasy.client.core.executors.ApacheHttpClient4Executor;
import org.jboss.resteasy.util.GenericType;
import org.kie.api.runtime.process.WorkItem;
import org.kie.api.runtime.process.WorkItemHandler;
import org.kie.api.runtime.process.WorkItemManager;
import org.overlord.dtgov.server.i18n.Messages;
import org.overlord.sramp.governance.Governance;
import org.overlord.sramp.governance.ValueEntity;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class HttpClientWorkItemHandler implements WorkItemHandler {

    Logger log = LoggerFactory.getLogger(this.getClass());

    /**
     * Constructor.
     */
    public HttpClientWorkItemHandler() {
    }

    /**
     * Calls an HTTP endpoint. The address of the endpoint should be set in the
     * parameter map passed into the workItem by the BPMN workflow. Both
     * this parameters 'Url' as well as the method 'Method' are required
     * parameters.
     */
    @Override
    public void executeWorkItem(WorkItem workItem, WorkItemManager manager) {

        Governance governance = new Governance();
        ClientResponse<?> response = null;
        Map<String, Object> results = new HashMap<String, Object>();
        try {
            // extract required parameters
            String urlStr = (String) workItem.getParameter("Url"); //$NON-NLS-1$
            String method = (String) workItem.getParameter("Method"); //$NON-NLS-1$
            if (urlStr == null || method == null) {
                throw new Exception(Messages.i18n.format("HttpClientWorkItemHandler.MissingParams")); //$NON-NLS-1$
            }
            urlStr = urlStr.toLowerCase();
            urlStr = urlStr.replaceAll("\\{governance.url\\}", governance.getGovernanceUrl()); //$NON-NLS-1$
            Map<String, Object> params = workItem.getParameters();

            // replace tokens in the urlStr, the replacement value of the token
            // should be set in the parameters Map
            for (String key : params.keySet()) {
                // break out if there are no (more) tokens in the urlStr
                if (!urlStr.contains("{")) //$NON-NLS-1$
                    break;
                // replace the token if it is referenced in the urlStr
                String variable = "{" + key.toLowerCase() + "}"; //$NON-NLS-1$ //$NON-NLS-2$
                if (urlStr.contains(variable)) {
                    String escapedVariable = "\\{" + key.toLowerCase() + "\\}"; //$NON-NLS-1$ //$NON-NLS-2$
                    String urlEncodedParam = URLEncoder.encode((String) params.get(key), "UTF-8").replaceAll("%2F", //$NON-NLS-1$//$NON-NLS-2$
                            "*2F"); //$NON-NLS-1$
                    urlStr = urlStr.replaceAll(escapedVariable, urlEncodedParam);
                }
            }
            if (urlStr.contains("{")) //$NON-NLS-1$
                throw new Exception(Messages.i18n.format("HttpClientWorkItemHandler.IncorrectParams", urlStr)); //$NON-NLS-1$

            // call http endpoint
            log.info(Messages.i18n.format("HttpClientWorkItemHandler.CallingTo", method, urlStr)); //$NON-NLS-1$
            DefaultHttpClient httpClient = new DefaultHttpClient();

            String username = governance.getOverlordUser();
            String password = governance.getOverlordPassword();
            httpClient.getCredentialsProvider().setCredentials(AuthScope.ANY,
                    new UsernamePasswordCredentials(username, password));
            ApacheHttpClient4Executor executor = new ApacheHttpClient4Executor(httpClient);

            ClientRequest request = new ClientRequest(urlStr, executor);
            request.setHttpMethod(method);
            response = request.execute();
            int responseCode = response.getResponseStatus().getStatusCode();
            if (responseCode >= 200 && responseCode < 300) {
                Map<String, ValueEntity> map = (Map<String, ValueEntity>) response
                        .getEntity(new GenericType<HashMap<String, ValueEntity>>() {
                        });
                for (String key : map.keySet()) {
                    if (map.get(key).getValue() != null) {
                        results.put(key, map.get(key).getValue());
                    }
                }
                log.info("reply=" + results); //$NON-NLS-1$
            } else {
                results.put("Status", "ERROR " + responseCode); //$NON-NLS-1$ //$NON-NLS-2$
                results.put("StatusMsg", //$NON-NLS-1$
                        Messages.i18n.format("HttpClientWorkItemHandler.UnreachableEndpoint", urlStr)); //$NON-NLS-1$
                log.error(Messages.i18n.format("HttpClientWorkItemHandler.UnreachableEndpoint", urlStr)); //$NON-NLS-1$
            }
        } catch (Exception e) {
            log.error(e.getMessage(), e);
        } finally {
            if (response != null)
                response.releaseConnection();
        }

        // notify manager that work item has been completed
        manager.completeWorkItem(workItem.getId(), results);
    }

    @Override
    public void abortWorkItem(WorkItem workItem, WorkItemManager manager) {

        // Do nothing, notifications cannot be aborted

    }

}