dcdbquery.cpp

/*
 * dcdbquery.cpp
 *
 *  Created on: Jan 06, 2015
 *      Author: Axel Auweter
 */

/* C++ standard headers */
#include <iostream>
#include <stdexcept>

/* C standard headers */
#include <cstdlib>
#include <unistd.h>

/* Custom headers */
#include "dcdb/timestamp.h"
#include "query.h"

void usage(void)
{
  /*            0---------1---------2---------3---------4---------5---------6---------7--------- */
  std::cout << "Usage:" << std::endl
            << "dcdbquery [-h <hostname>] [-r] [-l] <Sensor 1> [<Sensor 2> ...] <Start> <End>" << std::endl
            << "where" << std::endl
            << "    <hostname> - the name of the DB server" << std::endl
            << "    <Sensor n> - a public sensor name" << std::endl
            << "    <Start>    - start of time series" << std::endl
            << "    <End>      - end of time series" << std::endl
            << "Return the readings for the sensors in the interval from <Start> to <End>." << std::endl
            << "<Start> and <End> times should be supplied as 'yyyy-mm-dd hh:mm:ss' or unix" << std::endl
            << "timestamps." << std::endl
            << "When the -r option is specified, timestamps will be reported in internal (raw)" << std::endl
            << "format (nanoseconds since epoch) insted of ISO human readable form." << std::endl
            << "When the -l option is specified, times for <Start>, <End>, and in the generated" << std::endl
            << "output will be interpreted / printed in the local time zone." << std::endl;
  exit(EXIT_SUCCESS);
}

int main(int argc, char* argv[])
{
  /* Check command line arguments */
  if (argc <= 3) {
      usage();
  }

  /* Allocate our lovely helper class */
  DCDBQuery* myQuery;
  myQuery = new DCDBQuery();

  /* Get the options */
  int ret;
  const char *host = "localhost";
  while ((ret=getopt(argc, argv, "+h:rl"))!=-1) {
      switch(ret) {
          case 'h':
              host = optarg;
              break;
          case 'r':
              myQuery->setRawOutputEnabled(true);
              break;
          case 'l':
              myQuery->setLocalTimeEnabled(true);
              break;
          default:
              usage();
              exit(EXIT_FAILURE);
      }
  }

  /* Try to create DCDBTimeStamp objects from the arguments */
  DCDBTimeStamp start, end;
  try {
      bool local = myQuery->getLocalTimeEnabled();
      start = DCDBTimeStamp(argv[argc-2], local);
      end   = DCDBTimeStamp(argv[argc-1], local);
  }
  catch (std::exception& e) {
      std::cout << "Wrong time format." << std::endl;
      exit(EXIT_FAILURE);
  }

  /* Ensure start < end */
  if(start >= end) {
      std::cout << "Start time must be earlier than end time." << std::endl;
      exit(EXIT_FAILURE);
  }

  /* Build a list of sensornames */
  std::list<std::string> sensors;
  for (int arg = optind; arg < argc-2; arg++) {
      sensors.push_back(argv[arg]);
  }

  myQuery->doQuery(host, sensors, start, end);

  delete myQuery;

  return 0;
}