server/supernova/utilities/callback_system.hpp

   1 //  templated callback system
   2 //  Copyright (C) 2008, 2009 Tim Blechmann
   3 //
   4 //  This program is free software; you can redistribute it and/or modify
   5 //  it under the terms of the GNU General Public License as published by
   6 //  the Free Software Foundation; either version 2 of the License, or
   7 //  (at your option) any later version.
   8 //
   9 //  This program is distributed in the hope that it will be useful,
  10 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 //  GNU General Public License for more details.
  13 //
  14 //  You should have received a copy of the GNU General Public License
  15 //  along with this program; see the file COPYING.  If not, write to
  16 //  the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17 //  Boston, MA 02111-1307, USA.
  18
  19 #ifndef UTILITIES_CALLBACK_SYSTEM_HPP
  20 #define UTILITIES_CALLBACK_SYSTEM_HPP
  21
  22 #include <memory>
  23 #include <exception>
  24 #include <iostream>
  25
  26 #include <boost/checked_delete.hpp>
  27
  28 #include <boost/lockfree/spsc_queue.hpp>
  29 #include <boost/lockfree/queue.hpp>
  30
  31
  32 namespace nova {
  33
  34 /** \brief simple templated callback system, using a lockfree fifo */
  35 template <class callback_type,
  36           bool mpmc = true,
  37           class callback_deleter = boost::checked_deleter<callback_type> >
  38 class callback_system:
  39     private callback_deleter
  40 {
  41     typedef typename boost::mpl::if_c<mpmc, boost::lockfree::queue<callback_type*>,
  42                                             boost::lockfree::spsc_queue<callback_type*>
  43                                      >::type queue_type;
  44
  45 public:
  46     callback_system(size_t element_count = 2048):
  47         callbacks(element_count)
  48     {}
  49
  50     /** \brief adds a new Callback to the Scheduler, threadsafe */
  51     inline void add_callback(callback_type * cb)
  52     {
  53         callbacks.push(cb);
  54     }
  55
  56     /** \brief run all callbacks */
  57     inline void run_callbacks(void)
  58     {
  59         for (;;) {
  60             callback_type* runme;
  61
  62             if (not callbacks.pop(runme))
  63                 break;
  64
  65             run_callback(runme);
  66         }
  67     }
  68
  69     /** \brief run one callback
  70      *
  71      * assumes, that the queue contains at least one callback
  72      *
  73      * */
  74     void run_callback(void)
  75     {
  76         callback_type* runme;
  77         bool dequeued = callbacks.pop(&runme);
  78         assert(dequeued);
  79
  80         run_callback(runme);
  81     }
  82
  83 private:
  84     /** run a callback, handle exceptions */
  85     bool run_callback(callback_type * runme)
  86     {
  87         bool ret;
  88         try {
  89             runme->run();
  90             ret = true;
  91         } catch(std::exception const & e) {
  92             std::cerr << "unhandled exception while running callback: "  << e.what() << std::endl;
  93             ret = false;
  94         }
  95         callback_deleter::operator()(runme);
  96         return ret;
  97     }
  98
  99 protected:
 100     queue_type callbacks;   /**< \brief fifo for callbacks */
 101 };
 102
 103 } /* namespace nova */
 104
 105 #endif /* UTILITIES_CALLBACK_SYSTEM_HPP */