← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/scheduling.cc

  • Committer: Brian Aker
  • Date: 2009-07-03 00:39:59 UTC
  • mfrom: (1081.3.3 my_dir_removal)
  • Revision ID: brian@gaz-20090703003959-7y7mwb8u7etz9jco
Merge David

Show diffs side-by-side

added added

removed removed

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