git.m6w6.name Git - m6w6/libmemcached/blob - libmemcached/options/context.h

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  Libmemcached library
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *  All rights reserved.
   7  *
   8  *  Redistribution and use in source and binary forms, with or without
   9  *  modification, are permitted provided that the following conditions are
  10  *  met:
  11  *
  12  *      * Redistributions of source code must retain the above copyright
  13  *  notice, this list of conditions and the following disclaimer.
  14  *
  15  *      * Redistributions in binary form must reproduce the above
  16  *  copyright notice, this list of conditions and the following disclaimer
  17  *  in the documentation and/or other materials provided with the
  18  *  distribution.
  19  *
  20  *      * The names of its contributors may not be used to endorse or
  21  *  promote products derived from this software without specific prior
  22  *  written permission.
  23  *
  24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35  *
  36  */
  37
  38 #pragma once
  39
  40 #include <libmemcached/common.h>
  41
  42 class Context
  43 {
  44 public:
  45   Context(const char *option_string, size_t option_string_length, memcached_st *memc_arg,
  46           memcached_return_t &rc_arg) :
  47     scanner(NULL),
  48     begin(NULL),
  49     pos(0),
  50     memc(NULL),
  51     rc(rc_arg),
  52     _is_server(false),
  53     _end(false)
  54   {
  55     _hostname[0]= 0;
  56     buf= option_string;
  57     length= option_string_length;
  58     memc= memc_arg;
  59     init_scanner();
  60     rc= MEMCACHED_SUCCESS;
  61   }
  62
  63   bool end()
  64   {
  65     return _end;
  66   }
  67
  68   void start();
  69
  70   void set_end()
  71   {
  72     rc= MEMCACHED_SUCCESS;
  73     _end= true;
  74   }
  75
  76   void set_server()
  77   {
  78     _is_server= true;
  79   }
  80
  81   void unset_server()
  82   {
  83     _is_server= false;
  84   }
  85
  86   bool is_server()
  87   {
  88     return _is_server;
  89   }
  90
  91   const char *set_hostname(const char *str, size_t size)
  92   {
  93     size_t copy_length= (size_t)NI_MAXHOST > size ? size : (size_t)NI_MAXHOST;
  94     memcpy(_hostname, str, copy_length);
  95     _hostname[copy_length]= 0;
  96
  97     return _hostname;
  98   }
  99
 100   const char *hostname()
 101   {
 102     return _hostname;
 103   }
 104
 105   void abort(const char *error)
 106   {
 107     if (rc == MEMCACHED_SUCCESS)
 108       rc= MEMCACHED_PARSE_ERROR;
 109
 110     memcached_string_st *error_string= memcached_string_create(memc, NULL, 1024);
 111     memcached_string_append(error_string, memcached_literal_param("Error occured while parsing: "));
 112     memcached_string_append(error_string, memcached_string_make_from_cstr(begin));
 113     memcached_string_append(error_string, memcached_literal_param(" ("));
 114
 115     if (rc == MEMCACHED_PARSE_ERROR and error)
 116     {
 117       memcached_string_append(error_string, memcached_string_make_from_cstr(error));
 118     }
 119     else
 120     {
 121       memcached_string_append(error_string, memcached_string_make_from_cstr(memcached_strerror(NULL, rc)));
 122     }
 123     memcached_string_append(error_string, memcached_literal_param(")"));
 124
 125     memcached_set_error(*memc, rc, MEMCACHED_AT, memcached_string_value(error_string), memcached_string_length(error_string));
 126
 127     memcached_string_free(error_string);
 128   }
 129
 130   ~Context()
 131   {
 132     destroy_scanner();
 133   }
 134
 135   void *scanner;
 136   const char *buf;
 137   const char *begin;
 138   size_t pos;
 139   size_t length;
 140   memcached_st *memc;
 141   memcached_return_t &rc;
 142
 143 protected:
 144   void init_scanner();
 145   void destroy_scanner();
 146
 147 private:
 148   bool _is_server;
 149   bool _end;
 150   char _hostname[NI_MAXHOST];
 151 };