Drizzled Public API Documentation

memc_increment.cc

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