com.cisco.oss.foundation.message.RabbitMQAdmin Maven / Gradle / Ivy

Go to download
package com.cisco.oss.foundation.message;

import com.cisco.oss.foundation.configuration.ConfigUtil;
import com.cisco.oss.foundation.configuration.ConfigurationFactory;
import com.cisco.oss.foundation.http.HttpClient;
import com.cisco.oss.foundation.http.HttpMethod;
import com.cisco.oss.foundation.http.HttpRequest;
import com.cisco.oss.foundation.http.HttpResponse;
import com.cisco.oss.foundation.http.apache.ApacheHttpClientFactory;
import com.google.common.collect.Lists;
import com.jayway.jsonpath.JsonPath;
import com.rabbitmq.client.Address;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.configuration.Configuration;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

/**
 * Created by Yair Ogen (yaogen) on 09/03/2016.
 */
public enum RabbitMQAdmin {
    INSTANCE;

    private RabbitMQAdmin(){
        Configuration configuration = ConfigurationFactory.getConfiguration();

        final Map> serverConnections = ConfigUtil.parseComplexArrayStructure("service.rabbitmq.admin.connections");
        final ArrayList serverConnectionKeys = Lists.newArrayList(serverConnections.keySet());
        Collections.sort(serverConnectionKeys);

        for (int i = 0; i < serverConnectionKeys.size(); i++) {


            Map serverConnection = serverConnections.get(serverConnectionKeys.get(i));
            String host = serverConnection.get("host");
            int port = Integer.parseInt(serverConnection.get("port"));
            configuration.setProperty("rabbitMqAdmin." + i + ".host", host);
            configuration.setProperty("rabbitMqAdmin." + i + ".port", port);
        }

        configuration.setProperty("rabbitMqAdmin.http.exposeStatisticsToMonitor", "false");
        configuration.setProperty("rabbitMqAdmin.http.autoEncodeUri", "false");

    }

    private HttpClient httpClient;

    public void initAdmin(){
        if(httpClient== null){
            httpClient = ApacheHttpClientFactory.createHttpClient("rabbitMqAdmin");
        }
    }

    public int getQueueLength(String queueName){

        String adminUser = ConfigurationFactory.getConfiguration().getString("service.rabbitmq.admin.userName");
        String adminPwd = ConfigurationFactory.getConfiguration().getString("service.rabbitmq.admin.password");
        String authString = adminUser + ":" + adminPwd;
        byte[] authEncBytes = Base64.encodeBase64(authString.getBytes());
        String authStringEnc = new String(authEncBytes);


        String uri = "api/queues/%2f/" + queueName;
        HttpRequest httpRequest = HttpRequest.newBuilder()
                .uri(uri)
                .httpMethod(HttpMethod.GET)
                .header("Accept", "application/json")
                .header("Authorization", "Basic " + authStringEnc)
                .build();

        HttpResponse httpResponse = null;
        try {
            httpResponse = httpClient.execute(httpRequest);
        } catch (Exception e) {
            throw new RabbitMQAdminException("Can't execute rabbitmq admin request. error is: " + e, e);
        }
        int status = httpResponse.getStatus();
        String responseAsString = httpResponse.getResponseAsString();
        if(status == 200){
            Object queueLength = JsonPath.read(responseAsString, "$.backing_queue_status.len");
            return (Integer)queueLength;
        }else{
            throw new RabbitMQAdminException("Got bad response from rabbitmq server. Code: " + status + ". Reason: " + responseAsString);
        }
    }
}