← Back to branch summary

~mordred/blitzdb/out-of-tree

« back to all changes in this revision

Viewing changes to plugin/memcached_functions/memc_behavior_set.cc

  • Committer: Monty Taylor
  • Date: 2010-01-04 01:23:00 UTC
  • Revision ID: mordred@inaugust.com-20100104012300-k9etllo2yyt1flap
Split blitzdb into its own tree.

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 "config.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