gearman/libgearman/result.cc

/*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 * 
 *  Gearmand client and server library.
 *
 *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
 *  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.
 *
 *      * The names of its contributors may not 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
 *  OWNER 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 <libgearman/common.h>

#include <cassert>
#include <cstdlib>
#include <limits>
#include <memory>

#include <libgearman/common.h>
#include <libgearman/result.hpp>

bool gearman_result_is_null(const gearman_result_st *self)
{
  return self->is_null();
}

int64_t gearman_result_integer(const gearman_result_st *self)
{
  if (not self)
    return false;

  switch (self->type)
  {
    case GEARMAN_RESULT_BINARY:
      return atoll(gearman_string_value(&self->value.string));
    case GEARMAN_RESULT_BOOLEAN:
      return self->value.boolean;
    case GEARMAN_RESULT_INTEGER:
      return self->value.integer;
  }

  return false;
}

bool gearman_result_boolean(const gearman_result_st *self)
{
  if (not self)
    return false;

  switch (self->type)
  {
    case GEARMAN_RESULT_BINARY:
      return gearman_string_length(&self->value.string);
    case GEARMAN_RESULT_BOOLEAN:
      return self->value.boolean;
    case GEARMAN_RESULT_INTEGER:
      return self->value.integer ? true : false;
  }

  return false;
}

size_t gearman_result_size(const gearman_result_st *self)
{
  if (not self or self->type != GEARMAN_RESULT_BINARY)
    return 0;

  return gearman_string_length(&self->value.string);
}

const char *gearman_result_value(const gearman_result_st *self)
{
  if (not self or self->type != GEARMAN_RESULT_BINARY)
    return NULL;

  gearman_string_t ret= gearman_string(&self->value.string);

  return gearman_c_str(ret);
}

gearman_string_t gearman_result_string(const gearman_result_st *self)
{
  if (not self or self->type != GEARMAN_RESULT_BINARY)
  {
    gearman_string_t ret= {0, 0};
    return ret;
  }

  return gearman_string(&self->value.string);
}

gearman_string_t gearman_result_take_string(gearman_result_st *self)
{
  assert(self); // Programming error
  if (self->type == GEARMAN_RESULT_BINARY and gearman_result_size(self))
  {
    return gearman_string_take_string(&self->value.string);
  }
  self->_is_null= false;

  static gearman_string_t ret= {0, 0};
  return ret;
}

gearman_return_t gearman_result_store_string(gearman_result_st *self, gearman_string_t arg)
{
  return gearman_result_store_value(self, gearman_string_param(arg));
}

gearman_return_t gearman_result_store_value(gearman_result_st *self, const void *value, size_t size)
{
  if (not self)
    return GEARMAN_INVALID_ARGUMENT;

  if (self->type == GEARMAN_RESULT_BINARY)
  {
    gearman_string_reset(&self->value.string);
  }
  else
  {
    self->type= GEARMAN_RESULT_BINARY;
    gearman_string_create(&self->value.string, size);
  }
  self->_is_null= false;

  return gearman_string_append(&self->value.string, static_cast<const char *>(value), size);
}

void gearman_result_store_integer(gearman_result_st *self, int64_t value)
{
  if (not self)
    return;

  if (self->type == GEARMAN_RESULT_BINARY)
  {
    gearman_string_free(&self->value.string);
  }

  self->type= GEARMAN_RESULT_INTEGER;
  self->value.integer= value;
  self->_is_null= false;
}