← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/plugin/scheduler.cc

  • Committer: Monty Taylor
  • Date: 2011-02-13 17:26:39 UTC
  • mfrom: (2157.2.2 give-in-to-pkg-config)
  • mto: This revision was merged to the branch mainline in revision 2166.
  • Revision ID: mordred@inaugust.com-20110213172639-nhy7i72sfhoq13ms
Merged in pkg-config fixes.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/plugin/scheduler.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
 
4
 *  Copyright (C) 2008 Sun Microsystems, Inc.
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 <drizzled/server_includes.h>
21
 
#include <drizzled/scheduling.h>
22
 
#include <drizzled/gettext.h>
23
 
#include "drizzled/plugin/registry.h"
24
 
#include "drizzled/registry.h"
25
 
 
26
 
using namespace std;
27
 
using namespace drizzled;
28
 
 
29
 
plugin::SchedulerFactory *scheduler_factory= NULL;
30
 
Registry<plugin::SchedulerFactory *> all_schedulers;
31
 
 
32
 
bool add_scheduler_factory(plugin::SchedulerFactory *factory)
33
 
{
34
 
  if (all_schedulers.count(factory->getName()) != 0)
35
 
  {
36
 
    errmsg_printf(ERRMSG_LVL_ERROR,
 
20
#include "config.h"
 
21
 
 
22
#include <algorithm>
 
23
 
 
24
#include "drizzled/plugin/scheduler.h"
 
25
 
 
26
#include "drizzled/gettext.h"
 
27
#include "drizzled/errmsg_print.h"
 
28
 
 
29
namespace drizzled
 
30
{
 
31
 
 
32
extern size_t my_thread_stack_size;
 
33
 
 
34
std::vector<plugin::Scheduler *> all_schedulers;
 
35
 
 
36
/* Globals (TBK) */
 
37
static plugin::Scheduler *scheduler= NULL;
 
38
 
 
39
 
 
40
class FindSchedulerByName : public std::unary_function<plugin::Scheduler *, bool>
 
41
{
 
42
  const std::string *name;
 
43
public:
 
44
  FindSchedulerByName(const std::string *name_arg)
 
45
    : name(name_arg) {}
 
46
  result_type operator() (argument_type sched)
 
47
  {
 
48
    return (bool)((name->compare(sched->getName()) == 0));
 
49
  }
 
50
};
 
51
 
 
52
 
 
53
bool plugin::Scheduler::addPlugin(plugin::Scheduler *sched)
 
54
{
 
55
  std::vector<plugin::Scheduler *>::iterator iter=
 
56
    std::find_if(all_schedulers.begin(), all_schedulers.end(), 
 
57
            FindSchedulerByName(&sched->getName()));
 
58
 
 
59
  if (iter != all_schedulers.end())
 
60
  {
 
61
    errmsg_printf(error::ERROR,
37
62
                  _("Attempted to register a scheduler %s, but a scheduler "
38
63
                    "has already been registered with that name.\n"),
39
 
                    factory->getName().c_str());
40
 
    return true;
41
 
  }
42
 
  all_schedulers.add(factory);
43
 
  return false;
44
 
}
45
 
 
46
 
 
47
 
bool remove_scheduler_factory(plugin::SchedulerFactory *factory)
48
 
{
49
 
  scheduler_factory= NULL;
50
 
  all_schedulers.remove(factory);
51
 
  return false;
52
 
}
53
 
 
54
 
 
55
 
bool set_scheduler_factory(const string& name)
56
 
{
57
 
   
58
 
  plugin::SchedulerFactory *factory= all_schedulers.find(name);
59
 
  if (factory == NULL)
60
 
  {
61
 
    errmsg_printf(ERRMSG_LVL_WARN,
62
 
                  _("Attempted to configure %s as the scheduler, which did "
63
 
                    "not exist.\n"), name.c_str());
64
 
    return true;
65
 
  }
66
 
  scheduler_factory= factory;
67
 
 
68
 
  return false;
69
 
}
70
 
 
71
 
plugin::Scheduler *get_thread_scheduler()
72
 
{
73
 
  assert(scheduler_factory != NULL);
74
 
  plugin::Scheduler *sched= (*scheduler_factory)();
75
 
  if (sched == NULL)
76
 
  {
77
 
    errmsg_printf(ERRMSG_LVL_ERROR, _("Scheduler initialization failed.\n"));
78
 
    exit(1);
79
 
  }
80
 
  return sched;
81
 
}
82
 
 
 
64
                    sched->getName().c_str());
 
65
    return true;
 
66
  }
 
67
 
 
68
  sched->deactivate();
 
69
  all_schedulers.push_back(sched);
 
70
 
 
71
  return false;
 
72
}
 
73
 
 
74
 
 
75
void plugin::Scheduler::removePlugin(plugin::Scheduler *sched)
 
76
{
 
77
  all_schedulers.erase(std::find(all_schedulers.begin(),
 
78
                            all_schedulers.end(),
 
79
                            sched));
 
80
}
 
81
 
 
82
 
 
83
bool plugin::Scheduler::setPlugin(const std::string& name)
 
84
{
 
85
  std::vector<plugin::Scheduler *>::iterator iter=
 
86
    std::find_if(all_schedulers.begin(), all_schedulers.end(), 
 
87
            FindSchedulerByName(&name));
 
88
 
 
89
  if (iter != all_schedulers.end())
 
90
  {
 
91
    if (scheduler != NULL)
 
92
      scheduler->deactivate();
 
93
    scheduler= *iter;
 
94
    scheduler->activate();
 
95
    return false;
 
96
  }
 
97
 
 
98
  errmsg_printf(error::WARN,
 
99
                _("Attempted to configure %s as the scheduler, which did "
 
100
                  "not exist.\n"), name.c_str());
 
101
  return true;
 
102
}
 
103
 
 
104
 
 
105
plugin::Scheduler *plugin::Scheduler::getScheduler()
 
106
{
 
107
  return scheduler;
 
108
}
 
109
 
 
110
} /* namespace drizzled */