00001 /* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*- 00002 * vim:expandtab:shiftwidth=2:tabstop=2:smarttab: 00003 * 00004 * Copyright (C) 2009, Patrick "CaptTofu" Galbraith, Padraig O'Sullivan 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions are met: 00009 * 00010 * * Redistributions of source code must retain the above copyright notice, 00011 * this list of conditions and the following disclaimer. 00012 * * Redistributions in binary form must reproduce the above copyright notice, 00013 * this list of conditions and the following disclaimer in the documentation 00014 * and/or other materials provided with the distribution. 00015 * * Neither the name of Patrick Galbraith nor the names of its contributors 00016 * may be used to endorse or promote products derived from this software 00017 * without specific prior written permission. 00018 * 00019 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 00020 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 00021 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 00022 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE 00023 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 00024 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 00025 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 00026 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 00027 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 00028 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 00029 * THE POSSIBILITY OF SUCH DAMAGE. 00030 */ 00031 00032 #include <config.h> 00033 #include <drizzled/function/str/strfunc.h> 00034 00035 #include "memcached_functions.h" 00036 #include "memc_decrement.h" 00037 00038 #include <libmemcached/memcached.h> 00039 00040 #include <string> 00041 00042 using namespace std; 00043 using namespace drizzled; 00044 00045 String *MemcachedDecrement::val_str(String *str) 00046 { 00047 memcached_return rc; 00048 uint32_t offset; 00049 uint64_t value; 00050 size_t val_len; 00051 char tmp_buff[32]= ""; 00052 String *key; 00053 String *dec_str; 00054 null_value= false; 00055 00056 if ((arg_count != 1 && arg_count != 2) || 00057 ! (key= args[0]->val_str(str)) || 00058 ! memc) 00059 { 00060 return &failure_buff; 00061 } 00062 00063 if (arg_count == 2) { 00064 dec_str= args[1]->val_str(str); 00065 offset= static_cast<uint32_t>(atoi(dec_str->c_ptr())); 00066 } 00067 else 00068 { 00069 offset= 1; 00070 } 00071 00072 rc= memcached_decrement(memc, 00073 key->c_ptr(), 00074 key->length(), 00075 offset, 00076 &value); 00077 00078 snprintf(tmp_buff, 32, "%"PRIu64, value); 00079 val_len= strlen(tmp_buff); 00080 00081 if (rc != MEMCACHED_SUCCESS) 00082 { 00083 return &failure_buff; 00084 } 00085 00086 buffer.realloc(val_len); 00087 buffer.length(val_len); 00088 memcpy(buffer.ptr(), tmp_buff, val_len); 00089 00090 return &buffer; 00091 } 00092