← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/multi_thread/multi_thread.cc

  • Committer: Monty Taylor
  • Date: 2010-11-08 18:26:08 UTC
  • mto: This revision was merged to the branch mainline in revision 1931.
  • Revision ID: mordred@inaugust.com-20101108182608-lci86acl7r53sbi3
Replaced auto_ptr with scoped_ptr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/multi_thread/multi_thread.cc
13
13
   along with this program; if not, write to the Free Software
14
14
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
15
15
 
16
 
#include <config.h>
17
 
 
18
 
#include <iostream>
19
 
 
20
 
#include <drizzled/pthread_globals.h>
 
16
#include "config.h"
 
17
#include <plugin/multi_thread/multi_thread.h>
 
18
#include "drizzled/pthread_globals.h"
 
19
#include <boost/program_options.hpp>
21
20
#include <drizzled/module/option_map.h>
22
21
#include <drizzled/errmsg_print.h>
23
 
#include <drizzled/session.h>
24
 
#include <drizzled/session/cache.h>
25
 
#include <drizzled/abort_exception.h>
26
 
#include <drizzled/transaction_services.h>
27
 
#include <drizzled/gettext.h>
28
22
 
29
23
#include <boost/thread.hpp>
30
24
#include <boost/bind.hpp>
31
 
#include <boost/program_options.hpp>
32
 
 
33
 
#include "multi_thread.h"
34
25
 
35
26
namespace po= boost::program_options;
36
27
using namespace std;
45
36
  extern size_t my_thread_stack_size;
46
37
}
47
38
 
48
 
namespace multi_thread {
49
 
 
50
 
void MultiThreadScheduler::runSession(drizzled::session_id_t id)
 
39
void MultiThreadScheduler::runSession(drizzled::Session *session)
51
40
{
52
 
  char stack_dummy;
53
 
  boost::this_thread::disable_interruption disable_by_default;
54
 
 
55
 
  Session::shared_ptr session(session::Cache::singleton().find(id));
56
 
 
57
 
  try
 
41
  if (drizzled::internal::my_thread_init())
58
42
  {
59
 
 
60
 
    if (not session)
61
 
    {
62
 
      std::cerr << _("Session killed before thread could execute") << endl;
63
 
      return;
64
 
    }
65
 
    session->pushInterrupt(&disable_by_default);
66
 
 
67
 
    if (drizzled::internal::my_thread_init())
68
 
    {
69
 
      session->disconnect(drizzled::ER_OUT_OF_RESOURCES);
70
 
      session->status_var.aborted_connects++;
71
 
    }
72
 
    else
73
 
    {
74
 
      boost::this_thread::at_thread_exit(&internal::my_thread_end);
75
 
 
76
 
      session->thread_stack= (char*) &stack_dummy;
77
 
      session->run();
78
 
    }
79
 
 
 
43
    session->disconnect(drizzled::ER_OUT_OF_RESOURCES, true);
 
44
    session->status_var.aborted_connects++;
80
45
    killSessionNow(session);
81
46
  }
82
 
  catch (abort_exception& ex)
83
 
  {
84
 
    cout << _("Drizzle has receieved an abort event.") << endl;
85
 
    cout << _("In Function: ") << *::boost::get_error_info<boost::throw_function>(ex) << endl;
86
 
    cout << _("In File: ") << *::boost::get_error_info<boost::throw_file>(ex) << endl;
87
 
    cout << _("On Line: ") << *::boost::get_error_info<boost::throw_line>(ex) << endl;
 
47
  boost::this_thread::at_thread_exit(&internal::my_thread_end);
88
48
 
89
 
    TransactionServices::singleton().sendShutdownEvent(*session.get());
90
 
  }
91
 
  // @todo remove hard spin by disconnection the session first from the
92
 
  // thread.
93
 
  while (not session.unique()) {}
 
49
  session->thread_stack= (char*) &session;
 
50
  session->run();
 
51
  killSessionNow(session);
94
52
}
95
53
 
96
54
void MultiThreadScheduler::setStackSize()
106
64
 
107
65
  if (err != 0)
108
66
  {
109
 
    errmsg_printf(error::ERROR, _("Unable to get thread stack size"));
 
67
    errmsg_printf(ERRMSG_LVL_ERROR, _("Unable to get thread stack size\n"));
110
68
    my_thread_stack_size= 524288; // At the time of the writing of this code, this was OSX's
111
69
  }
112
70
 
131
89
#endif
132
90
}
133
91
 
134
 
bool MultiThreadScheduler::addSession(Session::shared_ptr &session)
 
92
bool MultiThreadScheduler::addSession(Session *session)
135
93
{
136
94
  if (thread_count >= max_threads)
137
95
    return true;
138
96
 
139
97
  thread_count.increment();
140
 
  try
141
 
  {
142
 
    session->getThread().reset(new boost::thread((boost::bind(&MultiThreadScheduler::runSession, this, session->getSessionId()))));
143
 
  }
144
 
  catch (std::exception&)
145
 
  {
146
 
    thread_count.decrement();
147
 
    return true;
148
 
  }
149
 
 
150
 
  if (not session->getThread())
151
 
  {
152
 
    thread_count.decrement();
153
 
    return true;
154
 
  }
155
 
 
156
 
  if (not session->getThread()->joinable())
 
98
 
 
99
  boost::thread new_thread(boost::bind(&MultiThreadScheduler::runSession, this, session));
 
100
 
 
101
  if (not new_thread.joinable())
157
102
  {
158
103
    thread_count.decrement();
159
104
    return true;
163
108
}
164
109
 
165
110
 
166
 
void MultiThreadScheduler::killSession(Session *session)
167
 
{
168
 
  boost_thread_shared_ptr thread(session->getThread());
169
 
 
170
 
  if (thread)
171
 
  {
172
 
    thread->interrupt();
173
 
  }
174
 
}
175
 
 
176
 
void MultiThreadScheduler::killSessionNow(Session::shared_ptr &session)
177
 
{
178
 
  killSession(session.get());
179
 
 
180
 
  session->disconnect();
181
 
 
 
111
void MultiThreadScheduler::killSessionNow(Session *session)
 
112
{
182
113
  /* Locks LOCK_thread_count and deletes session */
183
114
  Session::unlink(session);
184
115
  thread_count.decrement();
186
117
 
187
118
MultiThreadScheduler::~MultiThreadScheduler()
188
119
{
189
 
  boost::mutex::scoped_lock scopedLock(drizzled::session::Cache::singleton().mutex());
 
120
  boost::mutex::scoped_lock scopedLock(LOCK_thread_count);
190
121
  while (thread_count)
191
122
  {
192
123
    COND_thread_count.wait(scopedLock);
193
124
  }
194
125
}
195
126
 
196
 
} // multi_thread namespace
197
 
 
198
127
  
199
128
static int init(drizzled::module::Context &context)
200
129
{
201
130
  
202
 
  context.add(new multi_thread::MultiThreadScheduler("multi_thread"));
 
131
  context.add(new MultiThreadScheduler("multi_thread"));
203
132
 
204
133
  return 0;
205
134
}
208
137
{
209
138
  context("max-threads",
210
139
          po::value<max_threads_constraint>(&max_threads)->default_value(2048),
211
 
          _("Maximum number of user threads available."));
 
140
          N_("Maximum number of user threads available."));
212
141
}
213
142
 
214
143
DRIZZLE_DECLARE_PLUGIN
220
149
  "One Thread Per Session Scheduler",
221
150
  PLUGIN_LICENSE_GPL,
222
151
  init, /* Plugin Init */
223
 
  NULL,   /* depends */
 
152
  NULL,   /* system variables */
224
153
  init_options    /* config options */
225
154
}
226
155
DRIZZLE_DECLARE_PLUGIN_END;