msubmit.cc

#include "iputils.hh"
#include <sys/types.h>
#include <sys/stat.h>
#include <signal.h>
#include <boost/lexical_cast.hpp>
#include <boost/algorithm/string.hpp>

using std::cin;
using std::cerr;
using std::endl;


// syntax: msubmit address
int main(int argc, char** argv)
try
{
  if(argc!=2) {
    cerr<<"Syntax: (echo metric1 value1 ; echo metric2 value2 $(date +%s) ) |  msubmit carbon-server-address"<<endl;
    exit(EXIT_FAILURE);
  }
  ComboAddress remote(argv[1], 2003);
  time_t now = time(0);

  int s = SSocket(remote.sin4.sin_family, SOCK_STREAM, 0);
  SConnect(s, remote);
  int ret;
  std::string line;
  while(getline(cin, line)) {
    boost::trim_right(line);
 
    std::vector<std::string> parts;
    stringtok(parts, line, " \t\n");
    if(parts.size() < 2) 
      throw std::runtime_error("Line '"+line+"' contained less than 2 parts");

    string output = parts[0];
    output+=' ';
    output += parts[1];
    output+=' ';
    if(parts.size() > 2)
      output += parts[2];
    else
      output += boost::lexical_cast<string>(now);
    output += "\r\n";

    ret = writen(s, output.c_str(), output.size()); //  format: name value timestamp
    if(!ret)
      throw std::runtime_error("Carbon server '"+remote.toStringWithPort()+"' closed socket while feeding data");
    if(ret < 0)
      throw std::runtime_error("Error writing to Carbon server '"+remote.toStringWithPort()+"': "+strerror(errno));
  }
  shutdown(s, SHUT_WR);
  char c;
  ret=read(s, &c, 1);
  if(ret < 0)
    throw std::runtime_error("Improper shutdown of Carbon session with '"+remote.toStringWithPort()+"': "+strerror(errno));
  if(ret == 1)
    throw std::runtime_error("Improper shutdown of Carbon session with '"+remote.toStringWithPort()+"': they sent us data");
    
}
catch(std::exception& e)
{
  cerr<<"Error: "<<e.what()<<endl;
  exit(EXIT_FAILURE);
}