database/openGauss-server
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sync code

Browse Source
This commit is contained in:
LiHeng
2022-03-04 23:22:16 +08:00
parent d26ec83e7b
commit de223dd152
2618 changed files with 382415 additions and 163216 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/gausskernel/process/threadpool/threadpool_group.cpp
View File

@ -57,7 +57,7 @@
status == STATE_WAIT_XACTSYNC)
ThreadPoolGroup::ThreadPoolGroup(int maxWorkerNum, int expectWorkerNum, int maxStreamNum,
int groupId, int numaId, int cpuNum, int* cpuArr)
int groupId, int numaId, int cpuNum, int* cpuArr, bool enableBindCpuNuma)
: m_listener(NULL),
m_maxWorkerNum(maxWorkerNum),
m_maxStreamNum(maxStreamNum),
@ -78,11 +78,13 @@ ThreadPoolGroup::ThreadPoolGroup(int maxWorkerNum, int expectWorkerNum, int maxS
m_groupCpuNum(cpuNum),
m_groupCpuArr(cpuArr),
m_enableNumaDistribute(false),
m_enableBindCpuNuma(enableBindCpuNuma),
m_workers(NULL),
m_context(NULL)
{
pthread_mutex_init(&m_mutex, NULL);
CPU_ZERO(&m_nodeCpuSet);
CPU_ZERO(&m_CpuNumaSet);
m_streams = NULL;
m_freeStreamList = NULL;
@ -114,6 +116,12 @@ void ThreadPoolGroup::Init(bool enableNumaDistribute)
m_listener = New(CurrentMemoryContext) ThreadPoolListener(this);
m_listener->StartUp();
if (m_enableBindCpuNuma) {
for (int i = 0; i < m_groupCpuNum; i++) {
CPU_SET(m_groupCpuArr[i], &m_CpuNumaSet);
}
}
InitWorkerSentry();
InitStreamSentry();
@ -158,6 +166,8 @@ void ThreadPoolGroup::AddWorker(int i)
if (m_groupCpuArr) {
if (m_enableNumaDistribute) {
AttachThreadToNodeLevel(m_workers[i].worker->GetThreadId());
} else if (m_enableBindCpuNuma) {
AttachThreadToCpuNuma(m_workers[i].worker->GetThreadId());
} else {
AttachThreadToCPU(m_workers[i].worker->GetThreadId(), m_groupCpuArr[i % m_groupCpuNum]);
}
@ -402,6 +412,14 @@ void ThreadPoolGroup::AttachThreadToNodeLevel(ThreadId thread) const
ereport(WARNING, (errmsg("Fail to attach thread %lu to numa node %d", thread, m_numaId)));
}
void ThreadPoolGroup::AttachThreadToCpuNuma(ThreadId thread)
{
int ret = pthread_setaffinity_np(thread, sizeof(cpu_set_t), &m_CpuNumaSet);
if (ret != 0) {
ereport(WARNING, (errmsg("Fail to attach thread %lu to CPU NUMA", thread)));
}
}
void ThreadPoolGroup::InitStreamSentry()
{
m_streams = (ThreadStreamSentry*)palloc0_noexcept(sizeof(ThreadStreamSentry) * m_maxStreamNum);