~drizzle-trunk/drizzle/development

/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-

 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:

 *

 * Copyright (c) 2009, Patrick "CaptTofu" Galbraith, Padraig O'Sullivan

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions are met:

 *

 *   * Redistributions of source code must retain the above copyright notice,

 *     this list of conditions and the following disclaimer.

 *   * Redistributions in binary form must reproduce the above copyright notice,

 *     this list of conditions and the following disclaimer in the documentation

 *     and/or other materials provided with the distribution.

 *   * Neither the name of Patrick Galbraith nor the names of its contributors

 *     may be used to endorse or promote products derived from this software

 *     without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE

 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF

 * THE POSSIBILITY OF SUCH DAMAGE.

 *

 */

#include "config.h"

#include <drizzled/function/str/strfunc.h>

#include "memcached_functions.h"

#include "memc_increment.h"

#include <libmemcached/memcached.h>

#include <string>

using namespace std;

using namespace drizzled;

String *MemcachedIncrement::val_str(String *str)

{

  memcached_return rc;

  uint32_t offset;

  uint64_t value;

  size_t val_len;

  char tmp_buff[32]= "";

  String *key;

  String *inc_str;

  if ((arg_count != 1 && arg_count != 2) ||

      ! (key= args[0]->val_str(str)) ||

      ! memc)

  {

    return &failure_buff;

  }

  if (arg_count == 2) {

    inc_str= args[1]->val_str(str);

    offset= static_cast<uint32_t>(atoi(inc_str->c_ptr()));

  }

  else

  {

    offset= 1;

  }

  rc= memcached_increment(memc,

                    key->c_ptr(),

                    key->length(),

                    offset,

                    &value);

  snprintf(tmp_buff, 32, "%"PRIu64, value);

  val_len= strlen(tmp_buff);

  if (rc != MEMCACHED_SUCCESS)

  {

    return &failure_buff;

  }

  buffer.realloc(val_len);

  buffer.length(val_len);

  memcpy(buffer.ptr(), tmp_buff, val_len);

  return &buffer;

}