~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/memcached_functions/memc_decrement.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:
 
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/function/str/strfunc.h>
 
34
 
 
35
#include "memcached_functions.h"
 
36
#include "memc_decrement.h"
 
37
 
 
38
#include <libmemcached/memcached.h>
 
39
 
 
40
#include <string>
 
41
 
 
42
using namespace std;
 
43
 
 
44
String *MemcachedDecrement::val_str(String *str)
 
45
{
 
46
  memcached_return rc;
 
47
  int64_t offset;
 
48
  uint64_t value;
 
49
  size_t val_len;
 
50
  char tmp_buff[32]= "";
 
51
  String *key;
 
52
  String *dec_str;
 
53
  null_value= false;
 
54
 
 
55
  if ((arg_count != 1 && arg_count != 2) ||
 
56
      ! (key= args[0]->val_str(str)) ||
 
57
      ! memc)
 
58
  {
 
59
    return &failure_buff;
 
60
  }
 
61
 
 
62
  if (arg_count == 2) {
 
63
    dec_str= args[1]->val_str(str);
 
64
    offset= atoi(dec_str->c_ptr());
 
65
  }
 
66
  else
 
67
  {
 
68
    offset= 1;
 
69
  }
 
70
 
 
71
  rc= memcached_decrement(memc,
 
72
                    key->c_ptr(),
 
73
                    key->length(),
 
74
                    offset,
 
75
                    &value);
 
76
 
 
77
  snprintf(tmp_buff, 32, "%"PRIu64, value);
 
78
  val_len= strlen(tmp_buff);
 
79
 
 
80
  if (rc != MEMCACHED_SUCCESS)
 
81
  {
 
82
    return &failure_buff;
 
83
  }
 
84
 
 
85
  buffer.realloc(val_len);
 
86
  buffer.length(val_len);
 
87
  memcpy(buffer.ptr(), tmp_buff, val_len);
 
88
 
 
89
  return &buffer;
 
90
}
 
91