~drizzle-trunk/drizzle/development

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

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

 *

 *  Copyright (C) 2008 Sun Microsystems

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; version 2 of the License.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 */

#ifndef DRIZZLED_FUNCTION_SET_USER_VAR_H

#define DRIZZLED_FUNCTION_SET_USER_VAR_H

#include <drizzled/function/func.h>

/* Handling of user definable variables */

class user_var_entry;

class Item_func_set_user_var :public Item_func

{

  enum Item_result cached_result_type;

  user_var_entry *entry;

  char buffer[MAX_FIELD_WIDTH];

  String value;

  my_decimal decimal_buff;

  bool null_item;

  union

  {

    int64_t vint;

    double vreal;

    String *vstr;

    my_decimal *vdec;

  } save_result;

public:

  LEX_STRING name; // keep it public

  Item_func_set_user_var(LEX_STRING a,Item *b)

    :Item_func(b), cached_result_type(INT_RESULT), name(a)

  {}

  enum Functype functype() const { return SUSERVAR_FUNC; }

  double val_real();

  int64_t val_int();

  String *val_str(String *str);

  my_decimal *val_decimal(my_decimal *);

  double val_result();

  int64_t val_int_result();

  String *str_result(String *str);

  my_decimal *val_decimal_result(my_decimal *);

  bool update_hash(void *ptr, uint32_t length, enum Item_result type,

  		   const CHARSET_INFO * const cs, Derivation dv, bool unsigned_arg);

  bool send(Protocol *protocol, String *str_arg);

  void make_field(Send_field *tmp_field);

  bool check(bool use_result_field);

  bool update();

  enum Item_result result_type () const { return cached_result_type; }

  bool fix_fields(Session *session, Item **ref);

  void fix_length_and_dec();

  virtual void print(String *str, enum_query_type query_type);

  void print_as_stmt(String *str, enum_query_type query_type);

  const char *func_name() const { return "set_user_var"; }

  int save_in_field(Field *field, bool no_conversions,

                    bool can_use_result_field);

  int save_in_field(Field *field, bool no_conversions)

  {

    return save_in_field(field, no_conversions, 1);

  }

  void save_org_in_field(Field *field) { (void)save_in_field(field, 1, 0); }

  bool register_field_in_read_map(unsigned char *arg);

};

#endif /* DRIZZLED_FUNCTION_SET_USER_VAR_H */