~drizzle-trunk/drizzle/development

/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-

 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:

 *

 *  Copyright (C) 2008 Sun Microsystems, Inc.

 *

 *  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 program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

/*  Extra functions used by unireg library */

#pragma once

#include <drizzled/visibility.h>

#include <sstream>

namespace drizzled

{

void unireg_exit() __attribute__((noreturn));

DRIZZLED_API void unireg_actual_abort(const char *file, int line, const char *func, const std::string& message) __attribute__((noreturn));

void unireg_startup_finished();

namespace stream {

namespace detail {

template<class Ch, class Tr, class A>

  class _unireg {

  private:

  public:

    typedef std::basic_ostringstream<Ch, Tr, A> stream_buffer;

  public:

    void operator()(const stream_buffer &s, const char *filename, int line, const char *func)

    {

      unireg_actual_abort(filename, line, func, s.str());

    }

  };

template<template <class Ch, class Tr, class A> class OutputPolicy, class Ch = char, class Tr = std::char_traits<Ch>, class A = std::allocator<Ch> >

  class log {

  private:

    typedef OutputPolicy<Ch, Tr, A> output_policy;

    const char *_filename;

    int _line_number;

    const char *_func;

  public:

    log() :

      _filename(NULL),

      _line_number(0),

      _func(NULL)

    { }

    void set_filename(const char *filename, int line_number, const char *func)

    {

      _filename= filename;

      _line_number= line_number;

      _func= func;

    }

    ~log()

    {

      output_policy()(arg, _filename, _line_number, _func);

    }

  public:

    template<class T>

      log &operator<<(const T &x)

      {

        arg << x;

        return *this;

      }

  private:

    typename output_policy::stream_buffer arg;

  };

} // namespace detail

class _unireg : public detail::log<detail::_unireg> {

public:

  _unireg(const char *filename, int line_number, const char *func)

  {

    set_filename(filename, line_number, func);

  }

};

} // namespace stream

#define unireg_abort stream::_unireg(__FILE__, __LINE__, __func__)

} /* namespace drizzled */