#include "multiverso/actor.h"

#include <chrono>

#include <string>

#include <thread>

#include "multiverso/message.h"

#include "multiverso/util/log.h"

#include "multiverso/util/mt_queue.h"

#include "multiverso/zoo.h"

namespace multiverso {

Actor::Actor(const std::string& name) : name_(name) {

mailbox_.reset(new MtQueue<MessagePtr>());

Zoo::Get()->RegisterActor(name, this);

is_working_ = false;

}

Actor::~Actor() {}

void Actor::Start() {

thread_.reset(new std::thread(&Actor::Main, this));

while (!is_working_) {

std::this_thread::sleep_for(std::chrono::milliseconds(10));

}

}

void Actor::Stop() {

while (!mailbox_->Empty()) { ; }

is_working_ = false;

mailbox_->Exit();

thread_->join();

}

void Actor::Receive(MessagePtr& msg) { mailbox_->Push(msg); }

void Actor::Main() {

is_working_ = true;

MessagePtr msg;

while (mailbox_->Pop(msg)) {

if (handlers_.find(msg->type()) != handlers_.end()) {

handlers_[msg->type()](msg);

} else if (handlers_.find(MsgType::Default) != handlers_.end()) {

handlers_[MsgType::Default](msg);

} else {

Log::Fatal("Unexpected msg type\n");

}

}

}

void Actor::SendTo(const std::string& dst_name, MessagePtr& msg) {

Zoo::Get()->SendTo(dst_name, msg);

}

} // namespace multiverso