~drizzle-trunk/drizzle/development

/* Copyright (C) 2000 MySQL AB

   This program is free software; you can redistribute it and/or modify

   it under the terms of the GNU General Public License as published by

   the Free Software Foundation; version 2 of the License.

   This program is distributed in the hope that it will be useful,

   but WITHOUT ANY WARRANTY; without even the implied warranty of

   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License

   along with this program; if not, write to the Free Software

   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

/* This implements 'user defined functions' */

#include <drizzled/server_includes.h>

#include <drizzled/gettext.h>

#include <drizzled/sql_udf.h>

#include <map>

#include <string>

using namespace std;

static bool udf_startup= false; /* We do not lock because startup is single threaded */

static map<string, Function_builder *> udf_map;

/* This is only called if using_udf_functions != 0 */

Function_builder *find_udf(const char *name, uint32_t length)

{

  /**

   * @todo: check without transform, then check with transform if needed

   */

  Function_builder *udf= NULL;

  if (udf_startup == false)

    return NULL;

  string find_str(name, length);

  transform(find_str.begin(), find_str.end(),

            find_str.begin(), ::tolower);

  map<string, Function_builder *>::iterator find_iter;

  find_iter=  udf_map.find(find_str);

  if (find_iter != udf_map.end())

    udf= (*find_iter).second;

  return (udf);

}

static bool add_udf(Function_builder *udf)

{

  /**

   * @todo: add all lower and all upper version

   */

  string add_str= udf->get_name();

  transform(add_str.begin(), add_str.end(),

            add_str.begin(), ::tolower);

  udf_map[add_str]= udf;

  using_udf_functions= 1;

  return true;

}

static bool remove_udf(Function_builder *udf)

{

  return udf_map.erase(udf->get_name());

}

int initialize_udf(st_plugin_int *plugin)

{

  Function_builder *f;

  udf_startup= true;

  if (plugin->plugin->init)

  {

    int r;

    if ((r= plugin->plugin->init((void *)&f)))

    {

      errmsg_printf(ERRMSG_LVL_ERROR,

                    _("Plugin '%s' init function returned error %d."),

                    plugin->name.str, r);

      return r;

    }

  }

  else

    return 1;

  if (!add_udf(f))

    return 1;

  plugin->data= f;

  return 0;

}

int finalize_udf(st_plugin_int *plugin)

{

  Function_builder *udf = static_cast<Function_builder *>(plugin->data);

  if (udf != NULL)

  {

    remove_udf(udf);

    if (plugin->plugin->deinit)

    {

      if (plugin->plugin->deinit((void *)udf))

      {

        /* TRANSLATORS: The leading word "udf" is the name

           of the plugin api, and so should not be translated. */

        errmsg_printf(ERRMSG_LVL_ERROR, _("udf plugin '%s' deinit() failed"),

  		      plugin->name.str);

      }

    }

  }

  return 0;

}