// Copyright (c) 2020 Lv Decai, Shenzhen, China.
// Filename: ThreadLocalStorage.h
// Version: 1.0
// Release date: 2020年10月11日
// License: MIT (http://opensource.org/licenses/MIT)

#pragma once

#include "../CommonCpp.h"

NAMESPACE_BEGIN;

// 线程数据存储模板类
template<typename DataType>
class CThreadLocalStorage
{
public:
	// 构造函数
	CThreadLocalStorage()
	{
		// 分配存储空间
		m_iTlsIndex = TlsAlloc();

		if (m_iTlsIndex == TLS_OUT_OF_INDEXES)
		{
			ASSERT(FALSE);
		}
	}

	// 析构函数
	~CThreadLocalStorage()
	{
		// 释放存储空间
		TlsFree(m_iTlsIndex);
	}

	// 取得存储在当前线程数据区的对象
	DataType* GetData()
	{
		return (DataType*) TlsGetValue(m_iTlsIndex);
	}

	// 存储对象到当前线程数据区
	void SetData(DataType* pData)
	{
		TlsSetValue(m_iTlsIndex, pData);
	}

private:
	// 存储空间索引
	DWORD m_iTlsIndex = 0;
};

NAMESPACE_END;