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 Socket;
  32
  33 /** @class ServerDescription
  34  *  @brief Class to describe a server to which we can send encoding work.
  35  */
  36 class ServerDescription
  37 {
  38 public:
  39         /** @param h Server host name or IP address in string form.
  40          *  @param t Number of threads to use on the server.
  41          */
  42         ServerDescription (std::string h, int t)
  43                 : _host_name (h)
  44                 , _threads (t)
  45         {}
  46
  47         /** @return server's host name or IP address in string form */
  48         std::string host_name () const {
  49                 return _host_name;
  50         }
  51
  52         /** @return number of threads to use on the server */
  53         int threads () const {
  54                 return _threads;
  55         }
  56
  57         void set_host_name (std::string n) {
  58                 _host_name = n;
  59         }
  60
  61         void set_threads (int t) {
  62                 _threads = t;
  63         }
  64
  65         std::string as_metadata () const;
  66
  67         static ServerDescription * create_from_metadata (std::string v);
  68
  69 private:
  70         /** server's host name */
  71         std::string _host_name;
  72         /** number of threads to use on the server */
  73         int _threads;
  74 };
  75
  76 class Server
  77 {
  78 public:
  79         Server (Log* log);
  80
  81         void run (int num_threads);
  82
  83 private:
  84         void worker_thread ();
  85         int process (boost::shared_ptr<Socket> socket);
  86
  87         std::vector<boost::thread *> _worker_threads;
  88         std::list<boost::shared_ptr<Socket> > _queue;
  89         boost::mutex _worker_mutex;
  90         boost::condition _worker_condition;
  91         Log* _log;
  92 };