commit 915d8279cc50a758a504d3002cc00a188b08c627
author dinglei <libya_003@163.com> Mon Feb 24 15:32:08 2020 +0800
committer GitHub <noreply@github.com> Mon Feb 24 15:32:08 2020 +0800
tree 3cd3bdd35fbd277a3a25f99d5bb59cf22a5ecae4
parent 38010bfc5027b440321a80c4daf088712818f492

refactor(transaction): use userdata to cache the local checker callback (#252)

src/extern/CProducer.cpp

diff --git a/src/extern/CProducer.cpp b/src/extern/CProducer.cpp
index d025a21..a0c3698 100644
--- a/src/extern/CProducer.cpp
+++ b/src/extern/CProducer.cpp
@@ -39,7 +39,15 @@
 #endif
 using namespace rocketmq;
 using namespace std;
-
+class MyLocalTransactionExecuterInner {
+ public:
+  MyLocalTransactionExecuterInner(CLocalTransactionExecutorCallback executor, CMessage* msg, void* userData)
+      : m_ExcutorCallback(executor), message(msg), data(userData) {}
+  ~MyLocalTransactionExecuterInner() {}
+  CLocalTransactionExecutorCallback m_ExcutorCallback;
+  CMessage* message;
+  void* data;
+};
 class LocalTransactionListenerInner : public TransactionListener {
  public:
   LocalTransactionListenerInner() {}
@@ -56,8 +64,10 @@
     if (m_CheckerCallback == NULL) {
       return LocalTransactionState::UNKNOWN;
     }
-    CMessage* msg = (CMessage*)(&message);
-    CTransactionStatus status = m_ExcutorCallback(m_producer, msg, arg);
+    (void)(message);
+    MyLocalTransactionExecuterInner* executerInner = (MyLocalTransactionExecuterInner*)arg;
+    CTransactionStatus status =
+        executerInner->m_ExcutorCallback(m_producer, executerInner->message, executerInner->data);
     switch (status) {
       case E_COMMIT_TRANSACTION:
         return LocalTransactionState::COMMIT_MESSAGE;
@@ -91,15 +101,8 @@
 
  private:
   CLocalTransactionCheckerCallback m_CheckerCallback;
-  CLocalTransactionExecutorCallback m_ExcutorCallback;
-
   CProducer* m_producer;
   void* m_data;
-
- public:
-  void setM_m_ExcutorCallback(CLocalTransactionExecutorCallback excutorcallback) {
-    m_ExcutorCallback = excutorcallback;
-  }
 };
 
 class SelectMessageQueueInner : public MessageQueueSelector {
@@ -584,8 +587,10 @@
   try {
     DefaultProducer* defaultMQProducer = (DefaultProducer*)producer;
     MQMessage* message = (MQMessage*)msg;
-    defaultMQProducer->listenerInner->setM_m_ExcutorCallback(callback);
-    SendResult sendResult = defaultMQProducer->innerTransactionProducer->sendMessageInTransaction(*message, userData);
+    MyLocalTransactionExecuterInner executerInner(callback, msg, userData);
+    // defaultMQProducer->listenerInner->setM_m_ExcutorCallback(callback);
+    SendResult sendResult =
+        defaultMQProducer->innerTransactionProducer->sendMessageInTransaction(*message, &executerInner);
     result->sendStatus = CSendStatus((int)sendResult.getSendStatus());
     result->offset = sendResult.getQueueOffset();
     strncpy(result->msgId, sendResult.getMsgId().c_str(), MAX_MESSAGE_ID_LENGTH - 1);