t_db_query.cpp

Go to the documentation of this file.

/*****************************************************************************\
*                                                                             *
*  Name   : db_query using freetds db-lib                                     *
*  Author : Chris Koeritz                                                     *
*                                                                             *
*******************************************************************************
* Copyright (c) 2008-$now By Author.  This program is free software; you can  *
* redistribute it and/or modify it under the terms of the GNU General Public  *
* License as published by the Free Software Foundation; either version 2 of   *
* the License or (at your option) any later version.  This is online at:      *
*     http://www.fsf.org/copyleft/gpl.html                                    *
* Please send any updates to: fred@gruntose.com                               *
\*****************************************************************************/


#include <db_freetds/common.h>
#include <db_freetds/query_handler.h>
#include <db_freetds/login_info.h>
#include <loggers/file_logger.h>
#include <opsystem/application_base.h>
#include <opsystem/path_configuration.h>

HOOPLE_STARTUP_CODE;

#define LOG(s) STAMPED_EMERGENCY_LOG(program_wide_logger(), s)

const char *DB_SECTION = "db_query";

database_login_info _global_db_info(DB_SECTION);

// we use this to set up an error logger, but share the db engine log.
#define SET_LOGGER { \
  log_base *old_log = set_PW_logger_for_combo \
      (path_configuration::make_logfile_name("db_engine.log"), \
      file_logger::LOG_FILE_SIZE, true); \
  WHACK(old_log); \
}


int main(int argc, char **argv)
{
  SET_LOGGER;

  if (argc < 3) {
    printf("\
This program needs two parameters.  The first is the configuration file\n\
where the database authentication parameters resides.  The second parameter\n\
is the database query in SQL to run against the database.\n");
    return 1;
  }
  const istring config_file = argv[1];
  const istring query = argv[2];

  LOG("=== db_query starting up ===");

  query_handler *dbmang = new query_handler;
  // get the database open.
  bool db_open = dbmang->prepare_for_query(config_file, _global_db_info);
  if (!db_open) {
    LOG(istring("db_query ERROR: failed to open database ") + _global_db_info.database);
    return 1;
  }

  console_logger out;  // used for producing results.
  bool query_ok = dbmang->process_query(query.s(), out);

  WHACK(dbmang);

  LOG("=== db_query finished ===");

  return !query_ok;  // reverse sense of bool for exit value.
}

---