← 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-08-18 07:19:56 UTC
  • mfrom: (1116.1.3 stewart)
  • mto: This revision was merged to the branch mainline in revision 1118.
  • Revision ID: brian@gaz-20090818071956-nfpoe9rp3i7p50kx
Merge my branch from Stewart into one branch

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 Sun Microsystems
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,
 
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,
61
37
                  _("Attempted to register a scheduler %s, but a scheduler "
62
38
                    "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 */
 
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