Beanstalk.java

package com.urbanairship.octobot;

import com.surftools.BeanstalkClientImpl.ClientImpl;

import org.apache.log4j.Logger;

// This class handles all interfacing with a Beanstalk in Octobot.
// It is responsible for connection initialization and management.

public class Beanstalk {

    private static final Logger logger = Logger.getLogger("Beanstalk");

    public static ClientImpl getBeanstalkChannel(String host, Integer port, String tube) {
        int attempts = 0;
        ClientImpl client = null;
        logger.info("Opening connection to Beanstalk tube: '" + tube + "'...");

        while (true) {
            attempts++;
            logger.debug("Attempt #" + attempts);
            try {
                client = new ClientImpl(host, port);
                client.useTube(tube);
                client.watch(tube);
                logger.info("Connected to Beanstalk");
                break;
            } catch (Exception e) {
                logger.error("Unable to connect to Beanstalk. Retrying in 5 seconds", e);
                try { Thread.sleep(1000 * 5); }
                catch (InterruptedException ex) { }
            }
        }

        return client;
    }

}