← Back to branch summary

~ubuntu-branches/ubuntu/trusty/drizzle/trusty

« back to all changes in this revision

Viewing changes to .pc/gearman-includes.patch/plugin/gearman_udf/function_map.cc

  • Committer: Package Import Robot
  • Author(s): Martin Pitt
  • Date: 2012-02-14 08:13:28 UTC
  • Revision ID: package-import@ubuntu.com-20120214081328-qwikj4mxbuvveo93
Tags: 2011.03.13-0ubuntu5
* Add gearman-includes.patch: Fix include path for current gearmand version.
* Rebuild against libgearman6.

Show diffs side-by-side

added added

removed removed

Lines of Context:
.pc/gearman-includes.patch/plugin/gearman_udf/function_map.cc
 
1
/* Copyright (C) 2009 Sun Microsystems, Inc.
 
2
 
 
3
   This program is free software; you can redistribute it and/or modify
 
4
   it under the terms of the GNU General Public License as published by
 
5
   the Free Software Foundation; version 2 of the License.
 
6
 
 
7
   This program is distributed in the hope that it will be useful,
 
8
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
   GNU General Public License for more details.
 
11
 
 
12
   You should have received a copy of the GNU General Public License
 
13
   along with this program; if not, write to the Free Software
 
14
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
 
15
 
 
16
#include "function_map.h"
 
17
#include <libgearman/client.h>
 
18
#include <string.h>
 
19
#include <stdlib.h>
 
20
 
 
21
using namespace std;
 
22
 
 
23
/* Constructor and destructor happen during module dlopen/dlclose. */
 
24
static GearmanFunctionMap _functionMap;
 
25
 
 
26
GearmanFunctionMap& GetFunctionMap(void)
 
27
{
 
28
  return _functionMap;
 
29
}
 
30
 
 
31
GearmanFunctionMap::GearmanFunctionMap()
 
32
{
 
33
  (void) pthread_mutex_init(&lock, NULL);
 
34
}
 
35
 
 
36
GearmanFunctionMap::~GearmanFunctionMap()
 
37
{
 
38
  map<string, gearman_client_st>::iterator x;
 
39
 
 
40
  for (x= functionMap.begin(); x != functionMap.end(); x++)
 
41
    gearman_client_free(&((*x).second));
 
42
 
 
43
  (void) pthread_mutex_destroy(&lock);
 
44
}
 
45
 
 
46
bool GearmanFunctionMap::add(string function, string servers)
 
47
{
 
48
  map<string, gearman_client_st>::iterator x;
 
49
  gearman_return_t ret;
 
50
 
 
51
  pthread_mutex_lock(&lock);
 
52
 
 
53
  x= functionMap.find(function);
 
54
  if (x == functionMap.end())
 
55
  {
 
56
    if (gearman_client_create(&(functionMap[function])) == NULL)
 
57
    {
 
58
      pthread_mutex_unlock(&lock);
 
59
      return false;
 
60
    }
 
61
  }
 
62
 
 
63
  gearman_client_remove_servers(&(functionMap[function]));
 
64
  ret= gearman_client_add_servers(&(functionMap[function]), servers.c_str());
 
65
  pthread_mutex_unlock(&lock);
 
66
  if (ret != GEARMAN_SUCCESS)
 
67
    return false;
 
68
 
 
69
  return true;
 
70
}
 
71
 
 
72
bool GearmanFunctionMap::get(string function, gearman_client_st *client)
 
73
{
 
74
  map<string, gearman_client_st>::iterator x;
 
75
 
 
76
  pthread_mutex_lock(&lock);
 
77
 
 
78
  x= functionMap.find(function);
 
79
  if (x == functionMap.end())
 
80
  {
 
81
    x= functionMap.find(string(""));
 
82
    if (x == functionMap.end())
 
83
    {
 
84
      pthread_mutex_unlock(&lock);
 
85
      return false;
 
86
    }
 
87
  }
 
88
 
 
89
  /* Clone the object, the list of host:port pairs get cloned with it. */
 
90
  if (gearman_client_clone(client, &((*x).second)) == NULL)
 
91
  {
 
92
    pthread_mutex_unlock(&lock);
 
93
    return false;
 
94
  }
 
95
 
 
96
  pthread_mutex_unlock(&lock);
 
97
  return true;
 
98
}