← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/memcached_functions/memc_behavior_get.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_get.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_get.h"
 
38
 
 
39
#include <libmemcached/memcached.h>
 
40
 
 
41
#include <string>
 
42
#include <algorithm>
 
43
 
 
44
using namespace std;
 
45
 
 
46
void MemcachedBehaviorGet::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 *MemcachedBehaviorGet::val_str(String *str)
 
55
{
 
56
  memcached_behavior mbehavior;
 
57
  uint64_t isetting;
 
58
  map<const string, memcached_behavior>::iterator it;
 
59
  String *tmp_behavior;
 
60
 
 
61
  if (arg_count != 1 ||
 
62
      ! (tmp_behavior= args[0]->val_str(str)) ||
 
63
      ! memc)
 
64
  {
 
65
    setFailureString("USAGE: memc_behavior_get('<behavior type>')");
 
66
    return &failure_buff;
 
67
  }
 
68
 
 
69
  string behavior(tmp_behavior->c_ptr());
 
70
 
 
71
  /*
 
72
   * We don't want the user to have to type in all input in upper
 
73
   * case so we transform the input strings to upper case here.
 
74
   */
 
75
  std::transform(behavior.begin(), behavior.end(),
 
76
                 behavior.begin(), ::toupper);
 
77
 
 
78
  it = behavior_map.find(behavior);
 
79
  if (it == behavior_map.end()) 
 
80
  {
 
81
    setFailureString("UNKNOWN BEHAVIOR TYPE!");
 
82
    return &failure_buff;
 
83
  }
 
84
        
 
85
  mbehavior= behavior_map[behavior];
 
86
 
 
87
  isetting= memcached_behavior_get(memc, mbehavior);
 
88
 
 
89
  switch (mbehavior)
 
90
  {
 
91
  case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
 
92
  case MEMCACHED_BEHAVIOR_NO_BLOCK:
 
93
  case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
 
94
  case MEMCACHED_BEHAVIOR_USER_DATA:
 
95
  case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 
96
  case MEMCACHED_BEHAVIOR_VERIFY_KEY:
 
97
  case MEMCACHED_BEHAVIOR_TCP_NODELAY:
 
98
  case MEMCACHED_BEHAVIOR_KETAMA:
 
99
  case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
 
100
    if (isetting == 1)
 
101
      return_buff.append("1");
 
102
    else if (isetting == 0)
 
103
      return_buff.append("0");
 
104
    else
 
105
    {
 
106
      setFailureString("INVALID VALUE FOR BEHAVIOR - MUST be 1 OR 0!");
 
107
      return &failure_buff;
 
108
    }
 
109
    break;
 
110
  case MEMCACHED_BEHAVIOR_DISTRIBUTION:
 
111
    {
 
112
      string setting(dist_settings_reverse_map[isetting]);
 
113
      return_buff.append(setting.c_str());
 
114
    }
 
115
    break;
 
116
  case MEMCACHED_BEHAVIOR_HASH:
 
117
    {
 
118
      string setting(hash_settings_reverse_map[isetting]);
 
119
      return_buff.append(setting.c_str());
 
120
    }
 
121
    break;
 
122
  case MEMCACHED_BEHAVIOR_KETAMA_HASH:
 
123
    {
 
124
      string setting(ketama_hash_settings_reverse_map[isetting]);
 
125
      return_buff.append(setting.c_str());
 
126
    }
 
127
    break;
 
128
  case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 
129
  case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 
130
  case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 
131
  case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
 
132
  case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT:
 
133
  case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK:
 
134
  case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK:
 
135
    {
 
136
      size_t setting_len= 0;
 
137
      char tmp_buff[16];
 
138
 
 
139
      snprintf(tmp_buff, 16, "%"PRIu64, isetting);
 
140
      setting_len= strlen(tmp_buff);
 
141
      return_buff.realloc(setting_len);
 
142
      return_buff.length(setting_len);
 
143
      memcpy(return_buff.ptr(),tmp_buff, setting_len);
 
144
    }
 
145
    break;
 
146
  default:
 
147
    break;
 
148
  }
 
149
 
 
150
  return &return_buff;
 
151
}