org.cassandraunit.shaded.com.addthis.metrics.reporter.config.GmondConfigParser Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of cassandra-unit-shaded Show documentation
Shaded version of cassandra-unit
There is a newer version: 4.3.1.0
Show newest version
/*
 * 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 com.addthis.metrics.reporter.config;

import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/*
  So I couldn't find any java parsers for libconfuse, which is sad.  Didn't
  want to bundle a full scale parser into this project.  Been a while
  since lex or yacc.

  http://xkcd.com/208/
 */
public class GmondConfigParser
{
    private static final Logger log = LoggerFactory.getLogger(GmondConfigParser.class);

    private static final String C_COMMENT_PATTERN = "//.*?\n"; //
    private static final String CPP_COMMENT_PATTERN = "(?s)//*.*?/*/"; /* */
    private static final String EMPTY_LINE_PATTERN = "(?m)^\\s*\n";
    private static final String UDP_SEND_PATTERN = "(?s)udp_send_channel\\s*\\{(.*?)\\}";


    public List getGmondSendChannels(String fileName)
    {
        try
        {
            String conf = readFile(fileName);
            return getGmondSendChannelsFromConf(conf);
        }
        catch (IOException ioe)
        {
            log.error("Unable to read gmond config from:" + fileName, ioe);
            return null;
        }
        catch (Exception e)
        {
            log.error("Error searching for unicast udp_send_channels.  It is possible none are defined in " + fileName, e);
            return null;
        }
    }

    public List getGmondSendChannelsFromConf(String conf)
    {
        String cleanConf = removeEmptyLines(stripComments(conf));
        List blobs =  findSendChannels(cleanConf);
        log.debug("Found {} channels", blobs.size());
        List hosts = new ArrayList();
        for (String blob : blobs)
        {
            Map chanMap = mapifyChannelString(blob);
            log.debug("Parsed channel from config {}", chanMap);
            HostPort hp = makeHostPort(chanMap);
            if (hp != null)
            {
                hosts.add(hp);
            }
            else
            {
                log.warn("Failure to create HostPort from {}", blob);
            }
        }
        return hosts;
    }


    public String stripComments(String conf)
    {
        String cFree = conf.replaceAll(C_COMMENT_PATTERN, "\n");
        return cFree.replaceAll(CPP_COMMENT_PATTERN, "");
    }


    public String removeEmptyLines(String conf)
    {
        return conf.replaceAll(EMPTY_LINE_PATTERN, "");
    }


    public List findSendChannels(String conf)
    {
        List channelBlobs = new ArrayList();

        Matcher matcher = Pattern.compile(UDP_SEND_PATTERN).matcher(conf);
        while(matcher.find())
        {
            channelBlobs.add(matcher.group(1).trim());
        }

        return channelBlobs;
    }

    public Map mapifyChannelString(String sChannel)
    {
        String[] arr = sChannel.split("\n");
        Map chan = new HashMap();
        for (int i=0; i chan)
    {
        if (chan.containsKey("mcast_join") || chan.containsKey("mcast_if"))
        {
            log.warn("Looks like a multicast send channel, not supported and ignoring: {}", chan);
            return null;
        }
        HostPort hp = null;
        try
        {
            hp = new HostPort(chan.get("host"), Integer.valueOf(chan.get("port")));
        }
        catch (Exception e)
        {
            // beter with slf4j 1.7
            log.warn("Failed to create HostPort for:" + chan, e);
        }
        return hp;
    }


    // java is ridiculous
    public String readFile(String fileName) throws IOException
    {
        FileReader fr = null;
        BufferedReader br = null;
        try
        {
            fr = new FileReader(fileName);
            br = new BufferedReader(fr);
            StringBuilder sb = new StringBuilder();
            String line = br.readLine();
            while (line != null)
            {
                sb.append(line);
                sb.append("\n");
                line = br.readLine();
            }
            return sb.toString();
        }
        finally
        {
            if (fr != null)
            {
                fr.close();
            }
            if (br != null)
            {
                br.close();
            }
        }
    }
}