//

// Created by cleon on 22-9-25.

//

#include "EventLoopThreadPool.h"

#include "EventLoopThread.h"

#include <memory>

EventLoopThreadPool::EventLoopThreadPool(EventLoop* baseLoop, const std::string& nameArg)

: baseLoop_(baseLoop)

, name_(nameArg)

, started_(false)

, numThreads_(0)

, next_(0) {

}

EventLoopThreadPool::~EventLoopThreadPool() {

}

void EventLoopThreadPool::start(const ThreadInitCallback& cb) {

started_ = true;

for (int i = 0; i < numThreads_; i++) {

char buf[name_.size() + 32];

snprintf(buf, sizeof buf, "%s%d", name_.c_str(), i);

EventLoopThread* t = new EventLoopThread(cb, buf);

threads_.push_back((std::unique_ptr<EventLoopThread>(t)));

loops_.push_back(t->startLoop());

}

if (numThreads_ == 0 && cb) {

cb(baseLoop_);

}

}

EventLoop* EventLoopThreadPool::getNextLoop() {

EventLoop* loop = nullptr;

if (!loops_.empty()) {

loop = loops_[next_];

++next_;

if (next_ >= loops_.size()) { next_ = 0; }

}

return loop;

}

std::vector<EventLoop*> EventLoopThreadPool::getAllLoops() {

if (loops_.empty()) {

return std::vector<EventLoop*>(1, baseLoop_);

} else {

return loops_;

}

}