src/lib/server.h

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   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; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 /** @file src/server.h
  21  *  @brief Class to describe a server to which we can send
  22  *  encoding work, and a class to implement such a server.
  23  */
  24
  25 #include <string>
  26 #include <boost/thread.hpp>
  27 #include <boost/asio.hpp>
  28 #include <boost/thread/condition.hpp>
  29 #include "log.h"
  30
  31 /** @class ServerDescription
  32  *  @brief Class to describe a server to which we can send encoding work.
  33  */
  34 class ServerDescription
  35 {
  36 public:
  37         /** @param h Server host name or IP address in string form.
  38          *  @param t Number of threads to use on the server.
  39          */
  40         ServerDescription (std::string h, int t)
  41                 : _host_name (h)
  42                 , _threads (t)
  43         {}
  44
  45         /** @return server's host name or IP address in string form */
  46         std::string host_name () const {
  47                 return _host_name;
  48         }
  49
  50         /** @return number of threads to use on the server */
  51         int threads () const {
  52                 return _threads;
  53         }
  54
  55         void set_host_name (std::string n) {
  56                 _host_name = n;
  57         }
  58
  59         void set_threads (int t) {
  60                 _threads = t;
  61         }
  62
  63         std::string as_metadata () const;
  64
  65         static ServerDescription * create_from_metadata (std::string v);
  66
  67 private:
  68         /** server's host name */
  69         std::string _host_name;
  70         /** number of threads to use on the server */
  71         int _threads;
  72 };
  73
  74 class Server
  75 {
  76 public:
  77         Server (Log* log);
  78
  79         void run (int num_threads);
  80
  81 private:
  82         void worker_thread ();
  83         int process (boost::shared_ptr<boost::asio::ip::tcp::socket> socket);
  84
  85         boost::asio::io_service _io_service;
  86         std::vector<boost::thread *> _worker_threads;
  87         std::list<boost::shared_ptr<boost::asio::ip::tcp::socket> > _queue;
  88         boost::mutex _worker_mutex;
  89         boost::condition _worker_condition;
  90         Log* _log;
  91 };