← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/scheduling.cc

Moved the last of the libdrizzleclient calls into Protocol.

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
 
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
 
#include "drizzled/plugin/scheduler.h"
22
 
#include "drizzled/plugin/registry.h"
23
 
 
24
 
#include "drizzled/gettext.h"
25
 
 
26
 
using namespace std;
27
 
 
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())
57
 
  {
58
 
    errmsg_printf(ERRMSG_LVL_ERROR,
59
 
                  _("Attempted to register a scheduler %s, but a scheduler "
60
 
                    "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 */
 
20
#include <drizzled/server_includes.h>
 
21
#include <drizzled/scheduling.h>
 
22
#include <drizzled/gettext.h>
 
23
#include <drizzled/connect.h>
 
24
 
 
25
Scheduler *thread_scheduler= NULL;
 
26
 
 
27
static bool scheduler_inited= false; /* We must insist that only one of these plugins get loaded at a time */
 
28
 
 
29
 
 
30
extern char *opt_scheduler;
 
31
 
 
32
Scheduler &get_thread_scheduler()
 
33
{
 
34
  assert(thread_scheduler != NULL);
 
35
  return *thread_scheduler;
 
36
}
 
37
 
 
38
int scheduling_initializer(st_plugin_int *plugin)
 
39
{
 
40
  if (memcmp(plugin->plugin->name, opt_scheduler, strlen(opt_scheduler)))
 
41
    return 0;
 
42
 
 
43
  if (scheduler_inited)
 
44
  {
 
45
    fprintf(stderr, "You cannot load more then one scheduler plugin\n");
 
46
    exit(1);
 
47
  }
 
48
 
 
49
  assert(plugin->plugin->init); /* Find poorly designed plugins */
 
50
 
 
51
  if (plugin->plugin->init((void *)&thread_scheduler))
 
52
  {
 
53
    /* 
 
54
      TRANSLATORS> The leading word "scheduling" is the name
 
55
      of the plugin api, and so should not be translated. 
 
56
    */
 
57
    errmsg_printf(ERRMSG_LVL_ERROR, _("scheduling plugin '%s' init() failed"),
 
58
                        plugin->name.str);
 
59
      return 1;
 
60
  }
 
61
 
 
62
  scheduler_inited= true;
 
63
  /* We populate so we can find which plugin was initialized later on */
 
64
  plugin->data= (void *)thread_scheduler;
 
65
 
 
66
  return 0;
 
67
 
 
68
}
 
69
 
 
70
int scheduling_finalizer(st_plugin_int *plugin)
 
71
{
 
72
  /* We know which one we initialized since its data pointer is filled */
 
73
  if (plugin->plugin->deinit && plugin->data)
 
74
  {
 
75
    if (plugin->plugin->deinit((void *)thread_scheduler))
 
76
    {
 
77
      /* TRANSLATORS: The leading word "scheduling" is the name
 
78
         of the plugin api, and so should not be translated. */
 
79
      errmsg_printf(ERRMSG_LVL_ERROR,
 
80
                    _("scheduling plugin '%s' deinit() failed"),
 
81
                    plugin->name.str);
 
82
    }
 
83
  }
 
84
 
 
85
  return 0;
 
86
}