mdf trx msg

Author: xxo1shine

src/main/java/org/tron/core/net/message/TransactionMessage.java

@@ -1,21 +1,25 @@
 package org.tron.core.net.message;
 
+import org.tron.common.utils.Sha256Hash;
 import org.tron.core.capsule.TransactionCapsule;
 import org.tron.protos.Protocol;
 import org.tron.protos.Protocol.Transaction;
 
 public class TransactionMessage extends TronMessage {
 
   private Transaction trx;
+  private TransactionCapsule transactionCapsule;
 
   public TransactionMessage(byte[] data) throws Exception {
     this.type = MessageTypes.TRX.asByte();
     this.data = data;
     this.trx = Protocol.Transaction.parseFrom(data);
+    this.transactionCapsule = new TransactionCapsule(this.trx);
   }
 
   public TransactionMessage(Transaction trx) {
     this.trx = trx;
+    this.transactionCapsule = new TransactionCapsule(this.trx);
     this.type = MessageTypes.TRX.asByte();
     this.data = trx.toByteArray();
   }
@@ -26,6 +30,11 @@ public String toString() {
         .append("messageId: ").append(super.getMessageId()).toString();
   }
 
+  @Override
+  public Sha256Hash getMessageId() {
+    return this.transactionCapsule.getTransactionId();
+  }
+
   @Override
   public Class<?> getAnswerMessage() {
     return null;
@@ -36,7 +45,7 @@ public Transaction getTransaction() {
   }
 
   public TransactionCapsule getTransactionCapsule() {
-    return new TransactionCapsule(getTransaction());
+    return this.transactionCapsule;
   }
 
 }