← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/scheduling.cc

  • Committer: Monty Taylor
  • Date: 2009-03-08 23:45:12 UTC
  • mto: (923.2.1 mordred)
  • mto: This revision was merged to the branch mainline in revision 921.
  • Revision ID: mordred@inaugust.com-20090308234512-tqkygxtu1iaig23s
Removed C99 isnan() usage, which allows us to remove the util/math.{cc,h} workarounds. Yay for standards!

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/scheduling.cc
1
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
3
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems, Inc.
 
4
 *  Copyright (C) 2008 Mark Atwood
5
5
 *
6
6
 *  This program is free software; you can redistribute it and/or modify
7
7
 *  it under the terms of the GNU General Public License as published by
17
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
#include <config.h>
21
 
 
22
 
#include <algorithm>
23
 
 
24
 
#include <drizzled/errmsg_print.h>
 
20
#include <drizzled/server_includes.h>
 
21
#include <drizzled/scheduling.h>
25
22
#include <drizzled/gettext.h>
26
 
#include <drizzled/plugin/scheduler.h>
27
 
 
28
 
namespace drizzled
29
 
{
30
 
 
31
 
extern size_t my_thread_stack_size;
32
 
 
33
 
std::vector<plugin::Scheduler *> all_schedulers;
34
 
 
35
 
/* Globals (TBK) */
36
 
static plugin::Scheduler *scheduler= NULL;
37
 
 
38
 
 
39
 
class FindSchedulerByName : public std::unary_function<plugin::Scheduler *, bool>
40
 
{
41
 
  const std::string *name;
42
 
public:
43
 
  FindSchedulerByName(const std::string *name_arg)
44
 
    : name(name_arg) {}
45
 
  result_type operator() (argument_type sched)
46
 
  {
47
 
    return (bool)((name->compare(sched->getName()) == 0));
48
 
  }
49
 
};
50
 
 
51
 
 
52
 
bool plugin::Scheduler::addPlugin(plugin::Scheduler *sched)
53
 
{
54
 
  std::vector<plugin::Scheduler *>::iterator iter=
55
 
    std::find_if(all_schedulers.begin(), all_schedulers.end(), 
56
 
            FindSchedulerByName(&sched->getName()));
57
 
 
58
 
  if (iter != all_schedulers.end())
59
 
  {
60
 
    errmsg_printf(error::ERROR,
61
 
                  _("Attempted to register a scheduler %s, but a scheduler "
62
 
                    "has already been registered with that name.\n"),
63
 
                    sched->getName().c_str());
64
 
    return true;
65
 
  }
66
 
 
67
 
  sched->deactivate();
68
 
  all_schedulers.push_back(sched);
69
 
 
70
 
  return false;
71
 
}
72
 
 
73
 
 
74
 
void plugin::Scheduler::removePlugin(plugin::Scheduler *sched)
75
 
{
76
 
  all_schedulers.erase(std::find(all_schedulers.begin(),
77
 
                            all_schedulers.end(),
78
 
                            sched));
79
 
}
80
 
 
81
 
 
82
 
bool plugin::Scheduler::setPlugin(const std::string& name)
83
 
{
84
 
  std::vector<plugin::Scheduler *>::iterator iter=
85
 
    std::find_if(all_schedulers.begin(), all_schedulers.end(), 
86
 
            FindSchedulerByName(&name));
87
 
 
88
 
  if (iter != all_schedulers.end())
89
 
  {
90
 
    if (scheduler != NULL)
91
 
      scheduler->deactivate();
92
 
    scheduler= *iter;
93
 
    scheduler->activate();
94
 
    return false;
95
 
  }
96
 
 
97
 
  errmsg_printf(error::WARN,
98
 
                _("Attempted to configure %s as the scheduler, which did "
99
 
                  "not exist.\n"), name.c_str());
100
 
  return true;
101
 
}
102
 
 
103
 
 
104
 
plugin::Scheduler *plugin::Scheduler::getScheduler()
105
 
{
106
 
  return scheduler;
107
 
}
108
 
 
109
 
} /* namespace drizzled */
 
23
#include <drizzled/plugin_scheduling.h>
 
24
#include <drizzled/connect.h>
 
25
 
 
26
scheduling_st thread_scheduler;
 
27
static bool scheduler_inited= false; /* We must insist that only one of these plugins get loaded at a time */
 
28
 
 
29
static void post_kill_dummy(Session *) { return; }
 
30
static bool end_thread_dummy(Session *, bool) { return false; }
 
31
 
 
32
extern char *opt_scheduler;
 
33
 
 
34
int scheduling_initializer(st_plugin_int *plugin)
 
35
{
 
36
  if (memcmp(plugin->plugin->name, opt_scheduler, strlen(opt_scheduler)))
 
37
    return 0;
 
38
 
 
39
  if (scheduler_inited)
 
40
  {
 
41
    fprintf(stderr, "You cannot load more then one scheduler plugin\n");
 
42
    exit(1);
 
43
  }
 
44
 
 
45
  memset(&thread_scheduler, 0, sizeof(scheduling_st));
 
46
 
 
47
  assert(plugin->plugin->init); /* Find poorly designed plugins */
 
48
  if (plugin->plugin->init)
 
49
  {
 
50
 
 
51
    thread_scheduler.post_kill_notification= post_kill_dummy;
 
52
    thread_scheduler.end_thread= end_thread_dummy;
 
53
    thread_scheduler.init_new_connection_thread= init_new_connection_handler_thread;
 
54
 
 
55
    if (plugin->plugin->init((void *)&thread_scheduler))
 
56
    {
 
57
      /* 
 
58
        TRANSLATORS> The leading word "scheduling" is the name
 
59
        of the plugin api, and so should not be translated. 
 
60
      */
 
61
      errmsg_printf(ERRMSG_LVL_ERROR, _("scheduling plugin '%s' init() failed"),
 
62
                      plugin->name.str);
 
63
      goto err;
 
64
    }
 
65
  }
 
66
 
 
67
  /* We are going to assert() on any plugin that is not well written. */
 
68
  assert(thread_scheduler.max_threads);
 
69
  assert(thread_scheduler.init_new_connection_thread);
 
70
  assert(thread_scheduler.add_connection);
 
71
  assert(thread_scheduler.post_kill_notification);
 
72
  assert(thread_scheduler.end_thread);
 
73
 
 
74
  scheduler_inited= true;
 
75
  /* We populate so we can find which plugin was initialized later on */
 
76
  plugin->data= (void *)&thread_scheduler;
 
77
  plugin->state= PLUGIN_IS_READY;
 
78
 
 
79
  return 0;
 
80
 
 
81
err:
 
82
 
 
83
  return 1;
 
84
}
 
85
 
 
86
int scheduling_finalizer(st_plugin_int *plugin)
 
87
{
 
88
  /* We know which one we initialized since its data pointer is filled */
 
89
  if (plugin->plugin->deinit && plugin->data)
 
90
  {
 
91
    if (plugin->plugin->deinit((void *)&thread_scheduler))
 
92
    {
 
93
      /* TRANSLATORS: The leading word "scheduling" is the name
 
94
         of the plugin api, and so should not be translated. */
 
95
      errmsg_printf(ERRMSG_LVL_ERROR, _("scheduling plugin '%s' deinit() failed"),
 
96
                    plugin->name.str);
 
97
    }
 
98
  }
 
99
 
 
100
  return 0;
 
101
}