← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/memcached_functions/memc_behavior_set.cc

  • Committer: Eric Day
  • Date: 2009-10-31 21:53:33 UTC
  • mfrom: (1200 staging)
  • mto: This revision was merged to the branch mainline in revision 1202.
  • Revision ID: eday@oddments.org-20091031215333-j94bjoanwmi68p6f
Merged trunk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/memcached_functions/memc_behavior_set.cc
 
1
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
 
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
3
 *
 
4
 * Copyright (c) 2009, Patrick "CaptTofu" Galbraith, Padraig O'Sullivan
 
5
 * All rights reserved.
 
6
 *
 
7
 * Redistribution and use in source and binary forms, with or without
 
8
 * modification, are permitted provided that the following conditions are met:
 
9
 *
 
10
 *   * Redistributions of source code must retain the above copyright notice,
 
11
 *     this list of conditions and the following disclaimer.
 
12
 *   * Redistributions in binary form must reproduce the above copyright notice,
 
13
 *     this list of conditions and the following disclaimer in the documentation
 
14
 *     and/or other materials provided with the distribution.
 
15
 *   * Neither the name of Patrick Galbraith nor the names of its contributors
 
16
 *     may be used to endorse or promote products derived from this software
 
17
 *     without specific prior written permission.
 
18
 *
 
19
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 
20
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 
21
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 
22
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 
23
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 
24
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 
25
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 
26
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 
27
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 
28
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
 
29
 * THE POSSIBILITY OF SUCH DAMAGE.
 
30
 */
 
31
 
 
32
#include <drizzled/server_includes.h>
 
33
#include <drizzled/item/func.h>
 
34
#include <drizzled/function/str/strfunc.h>
 
35
 
 
36
#include "memcached_functions.h"
 
37
#include "memc_behavior_set.h"
 
38
 
 
39
#include <libmemcached/memcached.h>
 
40
 
 
41
#include <string>
 
42
#include <algorithm>
 
43
 
 
44
using namespace std;
 
45
 
 
46
void MemcachedBehaviorSet::setFailureString(const char *error)
 
47
{
 
48
  size_t size= strlen(error);
 
49
  failure_buff.realloc(size);
 
50
  failure_buff.length(size);
 
51
  memcpy(failure_buff.ptr(), error, size);
 
52
}
 
53
 
 
54
String *MemcachedBehaviorSet::val_str(String *str)
 
55
{
 
56
  memcached_return rc;
 
57
  memcached_behavior mbehavior;
 
58
  uint64_t isetting= 0;
 
59
  map<const string, memcached_behavior>::iterator it_behav;
 
60
  map<const string, uint64_t>::iterator it_hash;
 
61
  map<const string, uint64_t>::iterator it_dist;
 
62
  String *tmp_behavior;
 
63
  String *tmp_setting;
 
64
 
 
65
  if (arg_count != 2 ||
 
66
      ! (tmp_behavior= args[0]->val_str(str)) ||
 
67
      ! (tmp_setting= args[1]->val_str(str)) ||
 
68
      ! memc)
 
69
  {
 
70
    setFailureString("USAGE: memc_behavior_set('<behavior type>','<value>')");
 
71
    return &failure_buff;
 
72
  }
 
73
 
 
74
  string behavior(tmp_behavior->c_ptr());
 
75
  string setting(tmp_setting->c_ptr());
 
76
 
 
77
  /*
 
78
   * We don't want the user to have to type in all input in upper
 
79
   * case so we transform the input strings to upper case here.
 
80
   */
 
81
  std::transform(behavior.begin(), behavior.end(),
 
82
                 behavior.begin(), ::toupper);
 
83
  std::transform(setting.begin(), setting.end(),
 
84
                 setting.begin(), ::toupper);
 
85
 
 
86
  it_behav= behavior_map.find(behavior);
 
87
  if (it_behav == behavior_map.end())
 
88
  {
 
89
    setFailureString("UNKNOWN BEHAVIOR TYPE!");
 
90
    return &failure_buff;
 
91
  }
 
92
  mbehavior= behavior_map[behavior];
 
93
 
 
94
  switch (mbehavior)
 
95
  {
 
96
  case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
 
97
  case MEMCACHED_BEHAVIOR_NO_BLOCK:
 
98
  case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
 
99
  case MEMCACHED_BEHAVIOR_USER_DATA:
 
100
  case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 
101
  case MEMCACHED_BEHAVIOR_VERIFY_KEY:
 
102
  case MEMCACHED_BEHAVIOR_TCP_NODELAY:
 
103
  case MEMCACHED_BEHAVIOR_KETAMA:
 
104
  case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
 
105
    if (setting.compare("1") == 0)
 
106
    {
 
107
      isetting= 1;
 
108
    }
 
109
    else if (setting.compare("0") == 0)
 
110
    {
 
111
      isetting= 0;
 
112
    }
 
113
    else
 
114
    {
 
115
      setFailureString("INVALID VALUE FOR BEHAVIOR - MUST be 1 OR 0!");
 
116
      return &failure_buff;
 
117
    }
 
118
    break;
 
119
  case MEMCACHED_BEHAVIOR_DISTRIBUTION:
 
120
    it_dist= dist_settings_map.find(setting);
 
121
    if (it_dist == dist_settings_map.end())
 
122
    {
 
123
      setFailureString("INVALID VALUE FOR DISTRIBUTION!");
 
124
      return &failure_buff;
 
125
    }
 
126
    isetting= dist_settings_map[setting];
 
127
    break;
 
128
  case MEMCACHED_BEHAVIOR_HASH:
 
129
    it_hash= hash_settings_map.find(setting);
 
130
    if (it_hash == hash_settings_map.end())
 
131
    {
 
132
      setFailureString("INVALID VALUE FOR MEMCACHED HASH ALGORITHM!");
 
133
      return &failure_buff;
 
134
    }
 
135
    isetting= hash_settings_map[setting];
 
136
    break;
 
137
  case MEMCACHED_BEHAVIOR_KETAMA_HASH:
 
138
    isetting= ketama_hash_settings_map[setting];
 
139
    if (! isetting)
 
140
    {
 
141
      setFailureString("INVALID VALUE FOR KETAMA HASH ALGORITHM!");
 
142
      return &failure_buff;
 
143
    }
 
144
    break;
 
145
  case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 
146
  case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 
147
  case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 
148
  case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
 
149
  case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT:
 
150
  case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK:
 
151
  case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK:
 
152
    /*
 
153
      What type of check the values passed to these behaviors?
 
154
      Range?
 
155
    */
 
156
    break;
 
157
  default:
 
158
    break;
 
159
  }
 
160
 
 
161
  rc= memcached_behavior_set(memc, mbehavior, isetting);
 
162
 
 
163
  if (rc != MEMCACHED_SUCCESS)
 
164
  {
 
165
    return &failure_buff;
 
166
  }
 
167
 
 
168
  return &success_buff;
 
169
}
 
170