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 00033 #include <config.h> 00034 #include <drizzled/function/str/strfunc.h> 00035 00036 #include "memcached_functions.h" 00037 #include "memc_increment.h" 00038 00039 #include <libmemcached/memcached.h> 00040 00041 #include <string> 00042 00043 using namespace std; 00044 using namespace drizzled; 00045 00046 String *MemcachedIncrement::val_str(String *str) 00047 { 00048 memcached_return rc; 00049 uint32_t offset; 00050 uint64_t value; 00051 size_t val_len; 00052 char tmp_buff[32]= ""; 00053 String *key; 00054 String *inc_str; 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 inc_str= args[1]->val_str(str); 00065 offset= static_cast<uint32_t>(atoi(inc_str->c_ptr())); 00066 } 00067 else 00068 { 00069 offset= 1; 00070 } 00071 00072 00073 rc= memcached_increment(memc, 00074 key->c_ptr(), 00075 key->length(), 00076 offset, 00077 &value); 00078 00079 snprintf(tmp_buff, 32, "%"PRIu64, value); 00080 val_len= strlen(tmp_buff); 00081 00082 if (rc != MEMCACHED_SUCCESS) 00083 { 00084 return &failure_buff; 00085 } 00086 00087 buffer.realloc(val_len); 00088 buffer.length(val_len); 00089 memcpy(buffer.ptr(), tmp_buff, val_len); 00090 00091 return &buffer; 00092 } 00093