haiku.git

// ThreadLocal.h

#ifndef THREAD_LOCAL_H
#define THREAD_LOCAL_H

#include <SupportDefs.h>

// ThreadLocalFreeHandler
class ThreadLocalFreeHandler {
public:
								ThreadLocalFreeHandler();
	virtual						~ThreadLocalFreeHandler();

	virtual	void				Free(void* data) = 0;
};

// ThreadLocal
class ThreadLocal {
public:
								ThreadLocal(
									ThreadLocalFreeHandler* freeHandler = NULL);
								~ThreadLocal();

			status_t			Set(void* data);
			void				Unset();
			void*				Get() const;

private:
			struct ThreadLocalMap;

			ThreadLocalMap*		fMap;
			ThreadLocalFreeHandler* fFreeHandler;
};

// ThreadLocalUnsetter
class ThreadLocalUnsetter {
public:
	ThreadLocalUnsetter(ThreadLocal* threadLocal)
		: fThreadLocal(threadLocal)
	{
	}

	ThreadLocalUnsetter(ThreadLocal& threadLocal)
		: fThreadLocal(&threadLocal)
	{
	}

	~ThreadLocalUnsetter()
	{
		if (fThreadLocal)
			fThreadLocal->Unset();
	}

private:
	ThreadLocal*	fThreadLocal;
};

#endif	// THREAD_LOCAL_H