com.cognifide.actions.msg.push.active.PushClientRunnable.java Source code

Introduction

Here is the source code for com.cognifide.actions.msg.push.active.PushClientRunnable.java

Source

/*--
 * #%L
 * Cognifide Actions
 * %%
 * Copyright (C) 2015 Cognifide
 * %%
 * 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.
 * #L%
 */

package com.cognifide.actions.msg.push.active;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Set;

import org.apache.commons.httpclient.Credentials;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpConnectionManager;
import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.MultiThreadedHttpConnectionManager;
import org.apache.commons.httpclient.UsernamePasswordCredentials;
import org.apache.commons.httpclient.auth.AuthScope;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.PostMethod;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.cognifide.actions.msg.push.api.PushReceiver;

public class PushClientRunnable implements Runnable {

    private static final Logger LOG = LoggerFactory.getLogger(PushClientRunnable.class);

    private static final String SERVLET_PATH = "/bin/cognifide/cq-actions.txt";

    private final Set<PushReceiver> receivers;

    private final String serverUrl;

    private final HttpClient client;

    private volatile boolean shouldStop = false;

    public PushClientRunnable(Set<PushReceiver> receivers, String serverUrl, String username, String password)
            throws URISyntaxException {
        this.receivers = receivers;
        this.serverUrl = serverUrl;

        final HttpConnectionManager connManager = new MultiThreadedHttpConnectionManager();
        client = new HttpClient(connManager);
        client.getParams().setAuthenticationPreemptive(true);
        final Credentials defaultcreds = new UsernamePasswordCredentials(username, password);
        final URI serverUri = new URI(serverUrl);
        final int port = serverUri.getPort() == -1 ? 80 : serverUri.getPort();
        client.getState().setCredentials(new AuthScope(serverUri.getHost(), port, AuthScope.ANY_REALM),
                defaultcreds);
    }

    @Override
    public void run() {
        while (!shouldStop) {
            try {
                connect();
            } catch (IOException e) {
                LOG.error("Can't connect to the server " + serverUrl, e);
            }
            try {
                Thread.sleep(1000 * 5);
            } catch (InterruptedException e1) {
                LOG.error("Interrupted", e1);
                return;
            }
        }
    }

    private void connect() throws IOException {
        final HttpMethod method = new GetMethod(serverUrl + SERVLET_PATH);
        if (client.executeMethod(method) != 200) {
            return;
        }
        final InputStreamReader reader = new InputStreamReader(method.getResponseBodyAsStream());
        final BufferedReader bufferedReader = new BufferedReader(reader);

        String msgId;
        while ((msgId = bufferedReader.readLine()) != null) {
            final String topic = bufferedReader.readLine();
            final String msg = bufferedReader.readLine();
            LOG.debug("Got message with id " + msgId);
            for (PushReceiver r : receivers) {
                r.gotMessage(topic, msg);
            }
            try {
                confirm(msgId);
                LOG.debug("Message " + msgId + " confirmed");
            } catch (IOException e) {
                LOG.error("Can't confirm message " + msgId, e);
            }
        }
        method.releaseConnection();
    }

    private void confirm(String msgId) throws IOException {
        final HttpMethod method = new PostMethod(serverUrl + SERVLET_PATH + "/" + msgId);
        client.executeMethod(method);
        method.releaseConnection();
    }

    public void stop() {
        shouldStop = true;
        ((MultiThreadedHttpConnectionManager) client.getHttpConnectionManager()).shutdown();
    }

}