在研究writer类之前,我们需要看下WriterBase,因为class Writer : public WriterBase
此处为语雀内容卡片,点击链接查看:https://www.yuque.com/u28554159/onrdis/vkq1w4hg2ahl1q6f
writer.h里面定义了一个Transmit类的share_ptr,实现了write方法,讲消息通过tansmitter发出去。至于tansmitter我们后面再看。
bool Write(char* buf, uint64_t msg_size, uint64_t msg_info_size) {
RETURN_VAL_IF(!WriterBase::IsInit(), false);
return transmitter_->Transmit(buf, msg_size, msg_info_size);
}
然后看下Init函数干了啥
Init
template <typename MessageT>
bool Writer<MessageT>::Init() {
{
std::lock_guard<std::mutex> g(lock_);
if (init_) {
return true;
}
transmitter_ =
transport::Transport::Instance()->CreateTransmitter<MessageT>(
role_attr_);
if (transmitter_ == nullptr) {
return false;
}
init_ = true;
}
this->role_attr_.set_id(transmitter_->id().HashValue());
channel_manager_ =
service_discovery::TopologyManager::Instance()->channel_manager();
JoinTheTopology();
return true;
该函数中创建了一个transmitter_实例(暂时先不关注细节),然后讲transmitter_id的has值设置到role_attr_属性。并调用service_discovery::TopologyManager::Instance()创建了一个channel_manager_单例。再调用JoinTheTopology函数。
JoinTheTopology
template <typename MessageT>
void Writer<MessageT>::JoinTheTopology() {
// add listener
change_conn_ = channel_manager_->AddChangeListener(std::bind(
&Writer<MessageT>::OnChannelChange, this, std::placeholders::_1));
// get peer readers
const std::string& channel_name = this->role_attr_.channel_name();
std::vector<proto::RoleAttributes> readers;
channel_manager_->GetReadersOfChannel(channel_name, &readers);
for (auto& reader : readers) {
transmitter_->Enable(reader);
}
channel_manager_->Join(this->role_attr_, proto::RoleType::ROLE_WRITER,
message::HasSerializer<MessageT>::value);
}
该函数用于讲创建好的channel_manager_实例中的AddChangeListener绑定一个回调,用于信号触发时的执行。这里就回到之前讲的,TopologyManager::AddChangeListener函数,当有参与者加入,信号触发,这边就会执行,那么继续看OnChannelChange函数里面干了啥。
OnChannelChange
template <typename MessageT>
void Writer<MessageT>::OnChannelChange(const proto::ChangeMsg& change_msg) {
if (change_msg.role_type() != proto::RoleType::ROLE_READER) {
return;
}
auto& reader_attr = change_msg.role_attr();
if (reader_attr.channel_name() != this->role_attr_.channel_name()) {
return;
}
auto operate_type = change_msg.operate_type();
if (operate_type == proto::OperateType::OPT_JOIN) {
transmitter_->Enable(reader_attr);
} else {
transmitter_->Disable(reader_attr);
}
}
当有参与者加入,改槽函数会被触发,然后比对传入的属性是啥,对方加入后的属性是读取,并且名字一样的情况下,在判断操作属性是否为加入,如果是加入,就使能transmitter_,否则不使能。 transmitter_是用户数据分发的。