[并发并行]_[任务停止]_[使用Pthread的线程本地存储来停止任务执行]

场景

1.多线程应用里, 线程基本上都是出自线程池, 那么基本不能使用pthread_cancel这类API, 因为它会终止线程运行, 线程终止后不能再次循环使用, 这对性能要求高的软件时基本是不会用pthread_cancel来停止逻辑运行的.

2.自行设计的可预测伪线程停止,其实就是停止执行当前的业务逻辑,并不是终止线程, 让线程回归线程池. 目前我发现的方案基本上离不开对状态变量的判断. (比较忙, 没有参考那些其他大公司的开源项目, 如有不足,请批评指正), 这里只说我发现的一种方案, 使用specific data,线程本地存储.

说明

1.pthread也可以类似Win32一样使用线程本地存储, 使用方式是差不多的, 参考线程本地存储. 关键的地方就是创建一个默认key来存储任务对象, 通过判断任务对象的值来中断逻辑执行. 大部分情况下是线程A 正在执行逻辑A, 线程B接收到客户终止任务执行请求, 通过获取到这个任务对象A, 并修改任务对象A的状态， 这时线程A 判断到任务对象A的状态时Abort时, 退出执行循环.

例子

// test-pthread-stop.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <Windows.h>
#include <iostream>
#include <stdint.h>

#include "pthread.h"

pthread_key_t gTaskKey;

typedef enum kTaskStatus1
{
    kTaskStatusOk = 0,
    kTaskStatusDoing,
    kTaskStatusAbort,
    kTaskStatusFinish
}kTaskStatus;

struct IPCPacket
{
    uint64_t _id;               //Packet id
    uint8_t status;         //Packet 状态
    uint32_t command_type;      //Packet 命令类型

    void* data;                 //接收和发送数据.
    uint32_t data_length;       //数据长度.
};

static bool TestIsAbort(){
    void* data = pthread_getspecific(gTaskKey);

    // pthread specific data 默认是 NULL.
    IPCPacket* packet = (IPCPacket*)data;
    if(packet && packet->status == kTaskStatusAbort)
        return true;
    return false;
}

static void DoWork(void* data){
    for(int i = 0; ; ++i){
        Sleep(1000);
        if(TestIsAbort()){
            std::cout << "StartThread Abort: " << i << std::endl;
            break;
        }else{
            std::cout << "StartThread: " << i << std::endl;
        }
    }
}

void* StartThread(void* data){

    IPCPacket* task = (IPCPacket*)data;
    pthread_setspecific(gTaskKey,task);
    DoWork(data);

    return NULL;
}

void TestThreadAbort(){

    pthread_key_create(&gTaskKey,NULL);

    IPCPacket task;
    memset(&task,0,sizeof(task));

    int count = 0;
    pthread_t t1;
    pthread_create(&t1,NULL,StartThread,&task);

    Sleep(3000);
    task.status = kTaskStatusAbort;
    pthread_join(t1,NULL);
}

int _tmain(int argc, _TCHAR* argv[])
{

    std::cout << "TestThreadAbort Begin" << std::endl;
    TestThreadAbort();
    std::cout << "TestThreadAbort End" << std::endl;
    system("pause");
    return 0;
}

输出:

TestThreadAbort Begin
StartThread: 0
StartThread: 1
StartThread Abort: 2
TestThreadAbort End

参考

Programming With POSIX Threads