Linux安全网 - Linux操作系统_Linux 命令_Linux教程_Linux黑客

会员投稿 投稿指南 本期推荐:
搜索:
您的位置: Linux安全网 > Linux编程 > » 正文

多线程的那点儿事(之多线程数据结构)

来源: feixiaoxing 分享至:
【 声明:版权所有,欢迎转载,请勿用于商业用途。  联系信箱:feixiaoxing @163.com】

    要想编写多线程,那就要使用锁。而在软件编写中,数据结构是少不了的。所以,我们在编写多线程的时候,就需要考虑一下如何在数据结构中插入锁。当然,有些数据结构是没有锁的,所以自然这个锁并不一定是必须的。

    比如说,我们编写一个多线程堆栈,应该怎么做呢,

typedef struct _STACK
{
    void* pData;
    int maxLen;
    int top;
    void* hLock;
    
    STATUS (*push)(struct _STACK* pStack, void* pData);
    STATUS (*pop)(struct _STACK* pStack, void* pData);   
}STACK;
    (1) 初始化操作

STACK* get_new_stack(void* pData, int len, void* pLock)
{
    STACK* pStack;
      
    if(NULL == pData || 0 == len)
        return NULL;

    pStack = (STACK*)malloc(sizeof(STACK));
    assert(NULL != pStack);

    memset(pStack, 0, sizeof(STACK));
    pStack->pData  = pData;
    pStack->maxLen = len;
    
    if(NULL != pLock)
        pStack->hLock = pLock;

    return pStack;
}  
    (2) 添加数据
STATUS push(struct _STACK* pStack, void* pData, int size)
{
    assert(NULL != pStack && NULL != pData);

    if(NULL != pStack->hLock)
        WaitForSingleObject((HANDLE)pStack->hLock, INFINITE);

    if(pStack->top == pStack->maxLen){
        if(NULL != pStack->hLock)
            ReleaseMutex((HANDLE)pStack->hLock);
        
        return ERROR;
    }

    memmove((char*)pStack->pData + size * pStack->top, (char*)pData, size);
    pStack->top ++;
    
    if(NULL != pStack->hLock)
        ReleaseMutex((HANDLE)pStack->hLock);        

    return OK;
}
    (3) 对2进行优化,因为判断的条件比较复杂

#define STACK_CHECK_LOCK(hLock) \
do{\
    if(hLock)\
        WaitForSingleObject((HANDLE)hLock, INFINITE);\
}while(0)

#define STACK_CHECK_UNLOCK(hLock) \
do{\
    if(hLock)\
        ReleaseMutex((HANDLE)hLock);\
}while(0)
    所以,2的代码可以修改为,

STATUS push(struct _STACK* pStack, void* pData, int size)
{
    assert(NULL != pStack && NULL != pData);

    STACK_CHECK_LOCK(pStack->hLock);

    if(pStack->top == pStack->maxLen){
        STACK_CHECK_UNLOCK(pStack->hLock);
        return ERROR;
    }

    memmove((char*)pStack->pData + size * pStack->top, (char*)pData, size);
    pStack->top ++;
    
    STACK_CHECK_UNLOCK(pStack->hLock);      
    return OK;
}

总结:
    (1)  一般来说,比较好的数据结构要兼有多线程和没有多线程两种情况
    (2)  如果需要用其他的锁代替mutex,直接换掉就可以,十分方便



Tags:
分享至:
最新图文资讯
1 2 3 4 5 6
验证码:点击我更换图片 理智评论文明上网,拒绝恶意谩骂 用户名:
关于我们 - 联系我们 - 广告服务 - 友情链接 - 网站地图 - 版权声明 - 发展历史