1 files changed, 9 insertions, 4 deletions
diff --git a/src/KingSystem/Utils/Thread/MessageTransceiverBase.cpp b/src/KingSystem/Utils/Thread/MessageTransceiverBase.cpp
index 32b15361..d1c8a023 100644
--- a/src/KingSystem/Utils/Thread/MessageTransceiverBase.cpp
+++ b/src/KingSystem/Utils/Thread/MessageTransceiverBase.cpp
@@ -1,4 +1,5 @@
 #include "KingSystem/Utils/Thread/MessageTransceiverBase.h"
+#include "KingSystem/Utils/Thread/MessageBroker.h"
 #include "KingSystem/Utils/Thread/MessageReceiverEx.h"
 
 namespace ksys {
@@ -9,12 +10,12 @@ MessageTransceiverBase::MessageTransceiverBase() = default;
 
 MessageTransceiverBase::~MessageTransceiverBase() = default;
 
-bool MessageTransceiverBase::checkGeneratorFlag() const {
-    return getGenerator()->checkFlag();
+bool MessageTransceiverBase::checkReceiverFlag() const {
+    return getReceiver()->checkFlag();
 }
 
-bool MessageTransceiverBase::checkGeneratorCounter() const {
-    return getGenerator()->checkCounter();
+bool MessageTransceiverBase::checkReceiverCounter() const {
+    return getReceiver()->checkCounter();
 }
 
 bool MessageTransceiverBase::m2() {
@@ -41,4 +42,8 @@ void MessageTransceiverBase::setGlobalDispatcher(MessageDispatcherBase* dispatch
     sDispatcher = dispatcher;
 }
 
+IMessageBrokerRegister* MessageTransceiverBase::getRegister(IMessageBroker& broker) const {
+    return broker.getRegister();
+}
+
 }  // namespace ksys