← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/gearman_udf/gman_do.cc

  • Committer: Brian Aker
  • Date: 2009-09-26 04:00:11 UTC
  • mfrom: (1126.12.1 trunk-nodebug)
  • Revision ID: brian@gaz-20090926040011-2qzxdcbpm1ibpkhl
Merge Lee

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/gearman_udf/gman_do.cc
 
1
/* Copyright (C) 2009 Sun Microsystems
 
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 "gman_do.h"
 
17
#include "function_map.h"
 
18
 
 
19
using namespace std;
 
20
 
 
21
extern "C"
 
22
{
 
23
  static void *_do_malloc(size_t size, void *arg)
 
24
  {
 
25
    Item_func_gman_do *item= (Item_func_gman_do *)arg;
 
26
    return item->realloc(size);
 
27
  }
 
28
}
 
29
 
 
30
Item_func_gman_do::~Item_func_gman_do()
 
31
{
 
32
  if (options & GMAN_DO_OPTIONS_CLIENT)
 
33
    gearman_client_free(&client);
 
34
}
 
35
 
 
36
String *Item_func_gman_do::val_str(String *str)
 
37
{
 
38
  String *function;
 
39
  String *res;
 
40
  const char *unique;
 
41
  const char *workload;
 
42
  size_t workload_size;
 
43
  size_t result_size;
 
44
  gearman_return_t ret;
 
45
  char job_handle[GEARMAN_JOB_HANDLE_SIZE];
 
46
 
 
47
  if (arg_count < 1 || arg_count > 3 || !(function= args[0]->val_str(str)))
 
48
  {
 
49
    null_value= 1;
 
50
    return NULL;
 
51
  }
 
52
 
 
53
  if (arg_count > 1 && (res= args[1]->val_str(str)) != NULL)
 
54
  {
 
55
    workload= res->ptr();
 
56
    workload_size= res->length();
 
57
  }
 
58
  else
 
59
  {
 
60
    workload= NULL;
 
61
    workload_size= 0;
 
62
  }
 
63
 
 
64
  if (arg_count == 3 && (res= args[2]->val_str(str)) != NULL)
 
65
    unique= res->ptr();
 
66
  else
 
67
    unique= NULL;
 
68
 
 
69
  if (!(options & GMAN_DO_OPTIONS_CLIENT))
 
70
  {
 
71
    if (!GetFunctionMap().get(string(function->ptr()), &client))
 
72
    {
 
73
      null_value= 1;
 
74
      return NULL;
 
75
    }
 
76
 
 
77
    gearman_client_set_workload_malloc(&client, _do_malloc, this);
 
78
    options= (gman_do_options_t)(options | GMAN_DO_OPTIONS_CLIENT);
 
79
  }
 
80
 
 
81
  if (options & GMAN_DO_OPTIONS_BACKGROUND)
 
82
  {
 
83
    if (options & GMAN_DO_OPTIONS_HIGH)
 
84
    {
 
85
      ret= gearman_client_do_high_background(&client, function->ptr(), unique,
 
86
                                             workload, workload_size,
 
87
                                             job_handle);
 
88
    }
 
89
    else if (options & GMAN_DO_OPTIONS_LOW)
 
90
    {
 
91
      ret= gearman_client_do_low_background(&client, function->ptr(), unique,
 
92
                                            workload, workload_size,
 
93
                                            job_handle);
 
94
    }
 
95
    else
 
96
    {
 
97
      ret= gearman_client_do_background(&client, function->ptr(), unique,
 
98
                                        workload, workload_size, job_handle);
 
99
    }
 
100
 
 
101
    if (ret == GEARMAN_SUCCESS)
 
102
    {
 
103
      result_size= strlen(job_handle);
 
104
      buffer.realloc(result_size);
 
105
      buffer.length(result_size);
 
106
      memcpy(buffer.ptr(), job_handle, result_size);
 
107
    }
 
108
  }
 
109
  else
 
110
  {
 
111
    if (options & GMAN_DO_OPTIONS_HIGH)
 
112
    {
 
113
      (void) gearman_client_do_high(&client, function->ptr(), unique, workload,
 
114
                                    workload_size, &result_size, &ret);
 
115
    }
 
116
    else if (options & GMAN_DO_OPTIONS_LOW)
 
117
    {
 
118
      (void) gearman_client_do_low(&client, function->ptr(), unique, workload,
 
119
                                   workload_size, &result_size, &ret);
 
120
    }
 
121
    else
 
122
    {
 
123
      (void) gearman_client_do(&client, function->ptr(), unique, workload,
 
124
                               workload_size, &result_size, &ret);
 
125
    }
 
126
  }
 
127
 
 
128
  if (ret != GEARMAN_SUCCESS)
 
129
  {
 
130
    null_value= 1;
 
131
    return NULL;
 
132
  }
 
133
 
 
134
  null_value= 0;
 
135
  return &buffer;
 
136
}
 
137
 
 
138
void *Item_func_gman_do::realloc(size_t size)
 
139
{
 
140
  buffer.realloc(size);
 
141
  buffer.length(size);
 
142
  return buffer.ptr();
 
143
}