include/qore/QoreThreadLocalStorage.h (4311)

/* -*- mode: c++; indent-tabs-mode: nil -*- */
/*
  QoreThreadLocalStorage.h

  Qore Programming Language

  Copyright (C) 2003 - 2011 David Nichols, all rights reserved

  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License as published by the Free Software Foundation; either
  version 2.1 of the License, or (at your option) any later version.

  This library 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
  Lesser General Public License for more details.

  You should have received a copy of the GNU Lesser General Public
  License along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#ifndef _QORE_QORETHREADLOCALSTORAGE_H

#define _QORE_QORETHREADLOCALSTORAGE_H

#include <pthread.h>
#include <assert.h>


template<typename T>
class QoreThreadLocalStorage {
protected:
   pthread_key_t key;
   
   DLLLOCAL QoreThreadLocalStorage(const QoreThreadLocalStorage&);
   
   DLLLOCAL QoreThreadLocalStorage& operator=(const QoreThreadLocalStorage&);

public:
   DLLLOCAL QoreThreadLocalStorage() {
      create();
   }

   DLLLOCAL ~QoreThreadLocalStorage() {
      destroy();
   }

   DLLLOCAL void create() {
      pthread_key_create(&key, 0);
   }

   DLLLOCAL void destroy() {
      pthread_key_delete(key);
   }

   DLLLOCAL T *get() {
      return (T *)pthread_getspecific(key);
   }

   DLLLOCAL void set(T *ptr) {
#ifndef DEBUG
      pthread_setspecific(key, (void *)ptr);
#else
      int rc = pthread_setspecific(key, (void *)ptr);
      assert(!rc);
#endif
   }
};

#endif // _QORE_QORETHREADLOCALSTORAGE_H