linux c++线程池实现 - tbsys实现详解

linux线程池 - tbsys实现详解

一年前刚刚毕业入职还会写几篇博客,后来项目取消开源要求员工删除博客,写博客这件事情就荒废了。一年后,回忆一下,如果不养成写东西的习惯可能很难积累下知识,成长会慢很多。现在还是继续写吧,也可以读读其它开源代码写一写。

tbsys是阿里的一份开源代码,在阿里集团广泛应用,由多隆大神创作,内部代码实现有很多精致巧妙的地方。

今天学习下tbsys的线程池实现。

实现框架

tbsys线程池类图

Runnable类

该类是一个抽象类,拥有run的纯虚方法。该类主要提供给CThread类使用。

CThread类

该类对Linux线程方法做了一个简单的封装,实现了方法:

  • start 起一个线程,开始运行
  • join 等待线程退出
  • getRunnable 得到Runnable对象
  • getArgs 得到回调参数
  • getpid 获得线程id,linux系统中的id号
  • hook 线程回调函数,静态方法

CDefaultRunnable类

该类继承自Runnable,是真正的thread_pool。该类实现了方法:

  • setThreadCount 设置线程数
  • start 启动thread_count个线程,开始运行
  • stop 设置类内_stop变量,在该类中没有实际意义
  • wait 等待子线程退出,防止主线程先结束

使用方法

使用tbsys线程池时,具体业务类继承CDefaultRunnable类,实现Runnable类R中的run方法,通过setThreadCount设置线程数目,调用start方法启动线程池,然后调用wait方法等待退出。

实现思路

1、在CDefaultRunnable类中,start方法会申请thread_count个CThread类,然后用指向自身的this指针和thread index作为参数调用这thread_count个CThread变量的start方法。

2、CThread类的start方法,将传入的Runnable(CDefaultRunnable)指针赋值给自身runnable成员。然后通过pthread_create启动线程,同时用自身的静态方法hook作为线程回调函数。又把指向自己的this指针作为参数。

3、CThread的hook函数从入参中获取CThread变量,又通过CThread变量获取runnable成员,调用runnable成员的run函数。

代码

/*
 * (C) 2007-2010 Taobao Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 *
 * Version: $Id$
 *
 * Authors:
 *   duolong <duolong@taobao.com>
 *
 */

#ifndef TBSYS_RUNNABLE_H_
#define TBSYS_RUNNABLE_H_

namespace tbsys {

/** 
 * @brief Runnable是一个抽象类,它拥有一个run纯虚方法
 * 主要用于Thread类  
 */
class Runnable {

public:
    /*
     * 析构
     */
    virtual ~Runnable() {
    }
    /**
     * 运行入口函数
     */
    virtual void run(CThread *thread, void *arg) = 0;
};

}

#endif /*RUNNABLE_H_*/
/*
 * (C) 2007-2010 Taobao Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 *
 * Version: $Id$
 *
 * Authors:
 *   duolong <duolong@taobao.com>
 *
 */

#ifndef TBSYS_THREAD_H_
#define TBSYS_THREAD_H_

#include <linux/unistd.h>

namespace tbsys {

/** 
 * @brief 对linux线程简单封装 
 */
class CThread {

public:
    /**
     * 构造函数
     */
    CThread() {
        tid = 0;
        pid = 0;
    }

    /**
     * 起一个线程,开始运行
     */
    bool start(Runnable *r, void *a) {
        runnable = r;
        args = a;
        return 0 == pthread_create(&tid, NULL, CThread::hook, this);
    }

    /**
     * 等待线程退出
     */
    void join() {
        if (tid) {
            pthread_join(tid, NULL);
            tid = 0;
            pid = 0;
        }
    }

    /**
     * 得到Runnable对象
     * 
     * @return Runnable
     */
    Runnable *getRunnable() {
        return runnable;
    }

    /**
     * 得到回调参数
     * 
     * @return args
     */
    void *getArgs() {
        return args;
    }

    /***
     * 得到线程的进程ID
     */
    int getpid() {
        return pid;
    }

    /**
     * 线程的回调函数
     * 
     */

    static void *hook(void *arg) {
        CThread *thread = (CThread*) arg;
        thread->pid = gettid();

        if (thread->getRunnable()) {
            thread->getRunnable()->run(thread, thread->getArgs());
        }

        return (void*) NULL;
    }

private:    
    /**
     * 得到tid号
     */
    #ifdef _syscall0
    static _syscall0(pid_t,gettid)
    #else
    static pid_t gettid() { return static_cast<pid_t>(syscall(__NR_gettid));}
    #endif

private:
    pthread_t tid;      // pthread_self() id
    int pid;            // 线程的进程ID
    Runnable *runnable;
    void *args;
};

}

#endif /*THREAD_H_*/
/*
 * (C) 2007-2010 Taobao Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 *
 * Version: $Id$
 *
 * Authors:
 *   duolong <duolong@taobao.com>
 *
 */

#ifndef TBSYS_DEFAULT_RUNNABLE_H_
#define TBSYS_DEFAULT_RUNNABLE_H_

namespace tbsys {

/** 
 * @brief 线程执行具体的业务的封装类,同时它维护了一个线程数据,也可以将此类
 * 看成一个线程池类
 */
class CDefaultRunnable : public Runnable {

public:
    /**
     * 构造
     */
    CDefaultRunnable(int threadCount = 1);

    /**
     * 析构
     */
    virtual ~CDefaultRunnable();

    /**
     * 设置线程数
     */
    void setThreadCount(int threadCount);

    /** 
     * create %_threadCount threads
     * @return started thread count;
     */
    int start();

    /**
     * stop
     */
    void stop();

    /**
     * wait
     */
    void wait();

protected:    
    CThread *_thread;
    int _threadCount;
    bool _stop;
};

}

#endif /*RUNNABLE_H_*/
/*
 * (C) 2007-2010 Taobao Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 *
 * Version: $Id$
 *
 * Authors:
 *   duolong <duolong@taobao.com>
 *
 */

#include "tbsys.h"

namespace tbsys {

/**
 * 构造
 */
CDefaultRunnable::CDefaultRunnable(int threadCount) {
    _stop = false;
    _threadCount = threadCount;
    _thread = NULL;
}
/*
 * 析构
 */
CDefaultRunnable::~CDefaultRunnable() {
    if (_thread) {
        delete[] _thread;
        _thread = NULL;
    }
}

/**
 * 设置线程数
 */
void CDefaultRunnable::setThreadCount(int threadCount)
{
    if (_thread != NULL) {
        TBSYS_LOG(ERROR, "已经在运行了不能设置线程数");
        return;
    }
    _threadCount = threadCount;
}

// start
int CDefaultRunnable::start() {
    if (_thread != NULL || _threadCount < 1) {
        TBSYS_LOG(ERROR, "start failure, _thread: %p, threadCount: %d", _thread, _threadCount);
        return 0;
    }
    _thread = new CThread[_threadCount];
    if (NULL == _thread)
    {
        TBSYS_LOG(ERROR, "create _thread object failed, threadCount: %d", _threadCount);
        return 0;
    }
    int i = 0;
    for (; i<_threadCount; i++)
    {
        if (!_thread[i].start(this, (void*)((long)i)))
        {
          return i;
        }
    }
    return i;
}

// stop
void CDefaultRunnable::stop() {
    _stop = true;
}

// wait
void CDefaultRunnable::wait() {
    if (_thread != NULL)
    {
        for (int i=0; i<_threadCount; i++)
        {
            _thread[i].join();
        }
    }
}

}

END
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 线程池是一种常用的多线程处理技术,可以用来实现TCP服务器。在TCP服务器中,当有客户端请求连接时,服务器需要为每个客户端请求创建一个新的线程来处理。 使用线程池的好处是可以减少线程创建和销毁的开销,并可以控制线程的数量,避免因过多线程而导致系统资源不足。 首先,创建一个线程池,可以使用Java中的ExecutorService类来实现。通过ExecutorService可以创建一个固定数量的线程池,并提供提交任务的方法。 接下来,初始化一个ServerSocket对象来监听客户端的连接请求。当有客户端连接请求到达时,服务器调用线程池的submit方法提交一个任务,用于处理客户端的请求。 在任务的处理过程中,服务器通过accept方法接受客户端的连接,并创建一个新的Socket对象用于与客户端进行通信。然后,将Socket对象传递给任务的处理函数,用于处理客户端的请求。 任务的处理函数可以自定义,可以根据具体的需求来编写。在任务处理完成后,服务器关闭与客户端的连接,并将线程重新放回线程池中。 使用线程池实现TCP服务器可以提高服务器的效率和性能,因为线程池可以重复利用线程,避免了频繁创建和销毁线程的开销。同时,通过控制线程的数量,可以避免因过多线程而导致系统资源不足的问题。 ### 回答2: 线程池是一种用于管理和复用线程的机制,通过线程池可以提高线程的利用率,降低线程创建和销毁的开销。使用线程池可以实现TCP服务器,下面是一个基本的实现思路: 1. 创建一个固定大小的线程池,可以通过ThreadPoolExecutor类来实现线程池的大小可以根据服务器的实际需求进行调整,一般根据服务器的处理能力和并发连接数来确定。 2. 创建一个ServerSocket对象,用于监听客户端的连接请求。 3. 在服务器的主线程中,通过一个循环不断地接受客户端的连接请求。当有新的连接请求时,将其封装为一个任务,并提交给线程池进行处理。可以使用execute()或submit()方法来提交任务。 4. 在任务中,首先需要通过Socket对象获取客户端的输入流和输出流,用于与客户端进行通信。可以使用BufferedReader和PrintWriter等类来简化输入输出操作。 5. 在任务中,可以根据具体的业务需求来处理客户端的请求。可以使用循环来不断接收客户端发送的数据。处理完客户端的请求后,通过输出流将响应结果返回给客户端。 6. 当任务处理完毕后,需要关闭输入流、输出流和Socket对象,释放资源。 7. 在循环中,如果达到了服务器的最大连接数或者其他限制条件,可以暂停接受新的连接请求,直到有已连接的客户端断开连接。 8. 当服务器不再需要监听新的连接请求时,需要关闭ServerSocket对象,停止监听。 通过以上步骤,可以使用线程池实现TCP服务器。线程池可以提高服务器的并发处理能力,避免因为大量的连接请求导致线程创建过多而耗尽系统资源。同时,线程池还可以复用线程,避免频繁创建和销毁线程的开销,提高系统的性能和响应速度。 ### 回答3: 线程池是一种用于管理和复用线程的机制,可以提高并发性和线程资源的利用率。在实现TCP服务器时,使用线程池可以有效地处理多客户端的请求。 首先,我们需要创建一个线程池来管理服务器的线程资源。线程池里的线程数根据服务器的需求来确定,可以根据实际情况调整线程数量。 接下来,我们需要创建一个服务器套接字,并绑定服务器的IP地址和端口号。然后,我们可以开始监听客户端的连接请求。 当有客户端连接请求到达时,服务器会接受连接并创建一个新的线程处理该客户端的请求。线程池会从线程池中取出一个空闲线程来处理请求,如果线程池中没有空闲线程,则任务将排队等待。 在线程池中处理客户端请求时,可以利用多线程的优势同时处理多个请求,提高服务器的并发性能。线程池会根据任务的处理时间和线程资源的可用性动态地分配线程。 在处理客户端请求的线程中,首先需要接收客户端发送的数据。然后,根据客户端的请求进行相应的处理,可以是发送数据给客户端,或者执行相应的操作。 处理完客户端请求后,线程将返回到线程池中,等待下一个任务的到来。这样可以避免频繁地创建和销毁线程,提高线程资源的利用率。 总结起来,通过使用线程池实现TCP服务器可以提高服务器的并发性和线程资源的利用率,更好地满足多客户端的请求。线程池可以动态地管理和复用线程,减少线程的创建和销毁开销,提高服务器的性能和可伸缩性。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值