← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/plugin/registry.h

  • Committer: brian
  • Date: 2008-06-25 05:29:13 UTC
  • Revision ID: brian@localhost.localdomain-20080625052913-6upwo0jsrl4lnapl
clean slate

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/plugin/registry.h
1
 
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems
5
 
 *
6
 
 *  This program is free software; you can redistribute it and/or modify
7
 
 *  it under the terms of the GNU General Public License as published by
8
 
 *  the Free Software Foundation; version 2 of the License.
9
 
 *
10
 
 *  This program is distributed in the hope that it will be useful,
11
 
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
 *  GNU General Public License for more details.
14
 
 *
15
 
 *  You should have received a copy of the GNU General Public License
16
 
 *  along with this program; if not, write to the Free Software
17
 
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
 
 */
19
 
 
20
 
#ifndef DRIZZLED_PLUGIN_REGISTRY_H
21
 
#define DRIZZLED_PLUGIN_REGISTRY_H
22
 
 
23
 
#include <string>
24
 
#include <vector>
25
 
#include <map>
26
 
#include <algorithm>
27
 
 
28
 
#include "drizzled/gettext.h"
29
 
#include "drizzled/unireg.h"
30
 
#include "drizzled/errmsg_print.h"
31
 
 
32
 
namespace drizzled
33
 
{
34
 
namespace plugin
35
 
{
36
 
class Module;
37
 
class Plugin;
38
 
class Library;
39
 
 
40
 
class Registry
41
 
{
42
 
private:
43
 
  std::map<std::string, Library *> library_map;
44
 
  std::map<std::string, Module *> module_map;
45
 
  std::map<std::string, const Plugin *> plugin_registry;
46
 
 
47
 
  Module *current_module;
48
 
 
49
 
  Registry()
50
 
   : module_map(),
51
 
     plugin_registry(),
52
 
     current_module(NULL)
53
 
  { }
54
 
 
55
 
  Registry(const Registry&);
56
 
  Registry& operator=(const Registry&);
57
 
  ~Registry();
58
 
public:
59
 
 
60
 
  static plugin::Registry& singleton()
61
 
  {
62
 
    static plugin::Registry *registry= new plugin::Registry();
63
 
    return *registry;
64
 
  }
65
 
 
66
 
  static void shutdown();
67
 
 
68
 
  Module *find(std::string name);
69
 
 
70
 
  void add(Module *module);
71
 
 
72
 
  void setCurrentModule(Module *module)
73
 
  {
74
 
    current_module= module;
75
 
  }
76
 
 
77
 
  void clearCurrentModule()
78
 
  {
79
 
    current_module= NULL;
80
 
  }
81
 
 
82
 
  std::vector<Module *> getList(bool active);
83
 
 
84
 
  const std::map<std::string, const Plugin *> &getPluginsMap() const
85
 
  {
86
 
    return plugin_registry;
87
 
  }
88
 
 
89
 
  const std::map<std::string, Module *> &getModulesMap() const
90
 
  {
91
 
    return module_map;
92
 
  }
93
 
 
94
 
  Library *addLibrary(const std::string &plugin_name);
95
 
  void removeLibrary(const std::string &plugin_name);
96
 
  Library *findLibrary(const std::string &plugin_name) const;
97
 
 
98
 
  template<class T>
99
 
  void add(T *plugin)
100
 
  {
101
 
    plugin->setModule(current_module);
102
 
    bool failed= false;
103
 
    std::string plugin_name(plugin->getName());
104
 
    std::transform(plugin_name.begin(), plugin_name.end(),
105
 
                   plugin_name.begin(), ::tolower);
106
 
    if (plugin_registry.find(plugin_name) != plugin_registry.end())
107
 
    {
108
 
      errmsg_printf(ERRMSG_LVL_ERROR,
109
 
                    _("Loading plugin %s failed: a plugin by that name already "
110
 
                      "exists.\n"), plugin->getName().c_str());
111
 
      failed= true;
112
 
    }
113
 
    if (T::addPlugin(plugin))
114
 
      failed= true;
115
 
    if (failed)
116
 
    {
117
 
      errmsg_printf(ERRMSG_LVL_ERROR,
118
 
                    _("Fatal error: Failed initializing %s plugin.\n"),
119
 
                    plugin->getName().c_str());
120
 
      unireg_abort(1);
121
 
    }
122
 
    plugin_registry.insert(std::pair<std::string, const Plugin *>(plugin_name, plugin));
123
 
  }
124
 
 
125
 
  template<class T>
126
 
  void remove(T *plugin)
127
 
  {
128
 
    std::string plugin_name(plugin->getName());
129
 
    std::transform(plugin_name.begin(), plugin_name.end(),
130
 
                   plugin_name.begin(), ::tolower);
131
 
    T::removePlugin(plugin);
132
 
    plugin_registry.erase(plugin_name);
133
 
  }
134
 
 
135
 
 
136
 
};
137
 
 
138
 
} /* end namespace plugin */
139
 
} /* end namespace drizzled */
140
 
#endif /* DRIZZLED_PLUGIN_REGISTRY_H */