~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

 */

#include <drizzled/global.h>

#include <string.h>

#include <mysys/hash.h>

#include <drizzled/xid.h>

XID::XID()

{}

bool XID::eq(XID *xid)

{

  return eq(xid->gtrid_length, xid->bqual_length, xid->data);

}

bool XID::eq(long g, long b, const char *d)

{

  return g == gtrid_length && b == bqual_length && !memcmp(d, data, g+b);

}

void XID::set(XID *xid)

{

  memcpy(this, xid, xid->length());

}

void XID::set(long f, const char *g, long gl, const char *b, long bl)

{

  formatID= f;

  memcpy(data, g, gtrid_length= gl);

  memcpy(data+gl, b, bqual_length= bl);

}

void XID::set(uint64_t xid)

{

  my_xid tmp;

  formatID= 1;

  set(DRIZZLE_XID_PREFIX_LEN, 0, DRIZZLE_XID_PREFIX);

  memcpy(data+DRIZZLE_XID_PREFIX_LEN, &server_id, sizeof(server_id));

  tmp= xid;

  memcpy(data+DRIZZLE_XID_OFFSET, &tmp, sizeof(tmp));

  gtrid_length=DRIZZLE_XID_GTRID_LEN;

}

void XID::set(long g, long b, const char *d)

{

  formatID= 1;

  gtrid_length= g;

  bqual_length= b;

  memcpy(data, d, g+b);

}

bool XID::is_null()

{

  return formatID == -1;

}

void XID::null()

{

  formatID= -1;

}

my_xid XID::quick_get_my_xid()

{

  my_xid tmp;

  memcpy(&tmp, data+DRIZZLE_XID_OFFSET, sizeof(tmp));

  return tmp;

}

my_xid XID::get_my_xid()

{

  return gtrid_length == DRIZZLE_XID_GTRID_LEN && bqual_length == 0 &&

    !memcmp(data+DRIZZLE_XID_PREFIX_LEN, &server_id, sizeof(server_id)) &&

    !memcmp(data, DRIZZLE_XID_PREFIX, DRIZZLE_XID_PREFIX_LEN) ?

    quick_get_my_xid() : 0;

}

uint32_t XID::length()

{

  return sizeof(formatID)+sizeof(gtrid_length)+sizeof(bqual_length)+

    gtrid_length+bqual_length;

}

unsigned char *XID::key()

{

  return (unsigned char *)&gtrid_length;

}

uint32_t XID::key_length()

{

  return sizeof(gtrid_length)+sizeof(bqual_length)+gtrid_length+bqual_length;

}

/***************************************************************************

  Handling of XA id cacheing

***************************************************************************/

pthread_mutex_t LOCK_xid_cache;

HASH xid_cache;

extern "C" unsigned char *xid_get_hash_key(const unsigned char *, size_t *, bool);

extern "C" void xid_free_hash(void *);

unsigned char *xid_get_hash_key(const unsigned char *ptr, size_t *length,

                        bool )

{

  *length=((XID_STATE*)ptr)->xid.key_length();

  return ((XID_STATE*)ptr)->xid.key();

}

void xid_free_hash(void *ptr)

{

  XID_STATE *state= (XID_STATE *)ptr;

  if (state->in_session == false)

    delete state;

}

bool xid_cache_init()

{

  pthread_mutex_init(&LOCK_xid_cache, MY_MUTEX_INIT_FAST);

  return hash_init(&xid_cache, &my_charset_bin, 100, 0, 0,

                   xid_get_hash_key, xid_free_hash, 0) != 0;

}

void xid_cache_free()

{

  if (hash_inited(&xid_cache))

  {

    hash_free(&xid_cache);

    pthread_mutex_destroy(&LOCK_xid_cache);

  }

}

XID_STATE *xid_cache_search(XID *xid)

{

  pthread_mutex_lock(&LOCK_xid_cache);

  XID_STATE *res=(XID_STATE *)hash_search(&xid_cache, xid->key(), xid->key_length());

  pthread_mutex_unlock(&LOCK_xid_cache);

  return res;

}

bool xid_cache_insert(XID *xid, enum xa_states xa_state)

{

  XID_STATE *xs;

  bool res;

  pthread_mutex_lock(&LOCK_xid_cache);

  if (hash_search(&xid_cache, xid->key(), xid->key_length()))

    res= false;

  else if ((xs = new XID_STATE) == NULL)

    res= true;

  else

  {

    xs->xa_state=xa_state;

    xs->xid.set(xid);

    xs->in_session=0;

    res= my_hash_insert(&xid_cache, (unsigned char*)xs);

  }

  pthread_mutex_unlock(&LOCK_xid_cache);

  return res;

}

bool xid_cache_insert(XID_STATE *xid_state)

{

  pthread_mutex_lock(&LOCK_xid_cache);

  assert(hash_search(&xid_cache, xid_state->xid.key(),

                          xid_state->xid.key_length())==0);

  bool res=my_hash_insert(&xid_cache, (unsigned char*)xid_state);

  pthread_mutex_unlock(&LOCK_xid_cache);

  return res;

}

void xid_cache_delete(XID_STATE *xid_state)

{

  pthread_mutex_lock(&LOCK_xid_cache);

  hash_delete(&xid_cache, (unsigned char *)xid_state);

  pthread_mutex_unlock(&LOCK_xid_cache);

}