CounterWithStatsd.java

/*
 * Copyright (c) 2014 Globo.com - ATeam
 * All rights reserved.
 *
 * This source is subject to the Apache License, Version 2.0.
 * Please see the LICENSE file for more information.
 *
 * Authors: See AUTHORS file
 *
 * 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 com.globo.galeb.metrics;

import static org.vertx.java.core.datagram.InternetProtocolFamily.IPv4;
import static com.globo.galeb.verticles.ConfVerticleDictionary.CONF_STATSD_ENABLE;
import static com.globo.galeb.verticles.ConfVerticleDictionary.CONF_STATSD_HOST;
import static com.globo.galeb.verticles.ConfVerticleDictionary.CONF_STATSD_PORT;
import static com.globo.galeb.verticles.ConfVerticleDictionary.CONF_STATSD_PREFIX;

import com.globo.galeb.metrics.StatsdClient.TypeStatsdMessage;

import org.vertx.java.core.Vertx;
import org.vertx.java.core.datagram.DatagramSocket;
import org.vertx.java.core.json.JsonObject;
import org.vertx.java.core.logging.Logger;

/**
 * Class CounterWithStatsd: send metrics to statsd
 *
 * @author: See AUTHORS file.
 * @version: 1.0.0, Oct 23, 2014.
 */
public class CounterWithStatsd implements ICounter {

    /** The statsd client. */
    private final StatsdClient statsdClient;

    /**
     * Cleanup the key.
     *
     * @param aString the a string
     * @param strDefault the str default
     * @return the string
     */
    private String cleanupString(String aString, String strDefault) {
        return !"".equals(aString)?aString.replaceAll("[^\\w]", "_"):strDefault;
    }

    /**
     * Merge virtualhost id with backend id.
     *
     * @param virtualhostId the virtualhost id
     * @param backendId the backend id
     * @return the string merged
     */
    private String mergeVirtualhostIdWithBackendId(String virtualhostId, String backendId) {
        return String.format("%s.%s",cleanupString(virtualhostId,"UNDEF"), cleanupString(backendId, "UNDEF"));
    }

    /**
     * Instantiates a new counter with statsd.
     *
     * @param conf the conf
     * @param vertx the vertx
     * @param log the log
     */
    public CounterWithStatsd(final JsonObject conf, final Vertx vertx, final Logger log) {
        if (conf.getBoolean(CONF_STATSD_ENABLE, false)) {
            String statsdHost = conf.getString(CONF_STATSD_HOST,"127.0.0.1");
            Integer statsdPort = conf.getInteger(CONF_STATSD_PORT, 8125);
            String statsdPrefix = conf.getString(CONF_STATSD_PREFIX,"");
            final DatagramSocket dgram = vertx.createDatagramSocket(IPv4).setReuseAddress(true);
            statsdClient = new StatsdClient(statsdHost, statsdPort, statsdPrefix, dgram, log);
        } else {
            statsdClient = null;
        }
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#httpCode(java.lang.String, java.lang.Integer)
     */
    @Override
    public void httpCode(String key, Integer code) {
        if (statsdClient!=null && key!=null && !("".equals(key))) {
            statsdClient.send(TypeStatsdMessage.TIME,
                    String.format("%s.httpCode%d:%d", key, code, 1));
        }
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#httpCode(java.lang.String, java.lang.String, java.lang.Integer)
     */
    @Override
    public void httpCode(String virtualhostId, String backendId, Integer code) {
        httpCode(mergeVirtualhostIdWithBackendId(virtualhostId, backendId), code);
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#incrHttpCode(java.lang.String, java.lang.Integer)
     */
    @Override
    public void incrHttpCode(String key, Integer code) {
        incrHttpCode(key, code, 1.0);
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#incrHttpCode(java.lang.String, java.lang.Integer, double)
     */
    @Override
    public void incrHttpCode(String key, Integer code, double sample) {
        String srtSample = sample > 0.0 && sample < 1.0 ? String.format("|@%f", sample) : "";
        if (statsdClient!=null && key!=null && !("".equals(key))) {
            statsdClient.send(TypeStatsdMessage.COUNT,
                    String.format("%s.httpCode%d:%d%s", key, code, 1, srtSample));
        }
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#decrHttpCode(java.lang.String, java.lang.Integer)
     */
    @Override
    public void decrHttpCode(String key, Integer code) {
        decrHttpCode(key, code, 1.0);
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#decrHttpCode(java.lang.String, java.lang.Integer, double)
     */
    @Override
    public void decrHttpCode(String key, Integer code, double sample) {
        String srtSample = sample > 0.0 && sample < 1.0 ? String.format("|@%f", sample) : "";
        if (statsdClient!=null && key!=null && !("".equals(key))) {
            statsdClient.send(TypeStatsdMessage.COUNT,
                    String.format("%s.httpCode%d:%d%s", key, code, -1, srtSample));
        }
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#requestTime(java.lang.String, java.lang.Long)
     */
    @Override
    public void requestTime(String key, final Long initialRequestTime) {
        Long requestTime = System.currentTimeMillis() - initialRequestTime;
        if (statsdClient!=null && key!=null && !("".equals(key))) {
            statsdClient.send(TypeStatsdMessage.TIME,
                    String.format("%s.requestTime:%d", key, requestTime));
        }
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#requestTime(java.lang.String, java.lang.String, java.lang.Long)
     */
    @Override
    public void requestTime(String virtualhostId, String backendId,
            Long initialRequestTime) {
        requestTime(mergeVirtualhostIdWithBackendId(virtualhostId, backendId), initialRequestTime);
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#sendActiveSessions(java.lang.String, java.lang.Long)
     */
    @Override
    public void sendActiveSessions(String key, Long initialRequestTime) {
        if (statsdClient!=null && key!=null && !("".equals(key))) {
            statsdClient.send(TypeStatsdMessage.COUNT, String.format("%s.active:%d", key, 1));
        }
    }

    /* (non-Javadoc)
     * @see com.globo.galeb.metrics.ICounter#sendActiveSessions(java.lang.String, java.lang.String, java.lang.Long)
     */
    @Override
    public void sendActiveSessions(String virtualhostId, String backendId, Long initialRequestTime) {
        sendActiveSessions(mergeVirtualhostIdWithBackendId(virtualhostId, backendId), initialRequestTime);
    }
}