stressjob.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2007 Tommi Maekitalo
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * As a special exception, you may use this file as part of a free
10  * software library without restriction. Specifically, if other files
11  * instantiate templates or use macros or inline functions from this
12  * file, or you compile this file and link it with other files to
13  * produce an executable, this file does not by itself cause the
14  * resulting executable to be covered by the GNU General Public
15  * License. This exception does not however invalidate any other
16  * reasons why the executable file might be covered by the GNU Library
17  * General Public License.
18  *
19  * This library is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22  * Lesser General Public License for more details.
23  *
24  * You should have received a copy of the GNU Lesser General Public
25  * License along with this library; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27  */
28 
29 #ifndef TNT_STRESSJOB_H
30 #define TNT_STRESSJOB_H
31 
32 #include <tnt/job.h>
33 #include <string>
34 #include <cxxtools/mutex.h>
35 #include <iostream>
36 #include <tnt/socketif.h>
37 
47 namespace tnt
48 {
49  class StressHttpClientStreambuf : public std::streambuf
50  {
51  std::string request;
52 
53  public:
54  StressHttpClientStreambuf(const std::string& url);
55  int_type overflow(int_type c);
56  int_type underflow();
57  int sync();
58  };
59 
61  {
62  StressHttpClientStreambuf streambuf;
63 
64  public:
65  explicit StressHttpClient(const std::string& url)
66  : std::iostream(0),
67  streambuf(url)
68  { rdbuf(&streambuf); }
69  };
70 
71  class StressJob : public Job, private SocketIf
72  {
73  StressHttpClient client;
74  bool init;
75  Tntnet& app;
76  std::string url;
77  static cxxtools::Mutex mutex;
78  static unsigned count;
79 
80  virtual std::string getPeerIp() const;
81  virtual std::string getServerIp() const;
82  virtual bool isSsl() const;
83 
84  public:
85  StressJob(Tntnet& app_, const std::string& url_);
87  int getFd() const;
88  void setRead();
89  void setWrite();
90  };
91 
92 }
93 
94 #endif // TNT_STRESSJOB_H