Fix missing methods

This commit is contained in:
Manfred Karrer 2017-05-11 18:32:00 +02:00
parent be73d4d83c
commit a7b7cbbbca

View file

@ -41,6 +41,8 @@ import io.bisq.core.trade.TradeManager;
import io.bisq.core.trade.messages.TradeMsg;
import io.bisq.core.user.User;
import io.bisq.generated.protobuffer.PB;
import io.bisq.network.p2p.DecryptedMsgWithPubKey;
import io.bisq.network.p2p.MailboxMsg;
import io.bisq.network.p2p.NodeAddress;
import io.bisq.network.p2p.P2PService;
import lombok.Getter;
@ -53,7 +55,6 @@ import org.bitcoinj.core.Transaction;
import javax.annotation.Nullable;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
@Getter
@ -89,6 +90,8 @@ public class ProcessModel implements Model, Persistable {
transient private Transaction takeOfferFeeTx;
@Setter
transient private TradeMsg tradeMessage;
@Setter
transient private DecryptedMsgWithPubKey decryptedMsgWithPubKey;
// Mutable
private String takeOfferFeeTxId;
@ -235,4 +238,15 @@ public class ProcessModel implements Model, Persistable {
.setTempTradingPeerNodeAddress(tempTradingPeerNodeAddress.toProto())
.build();
}
public void removeMailboxMessageAfterProcessing(Trade trade) {
if (tradeMessage instanceof MailboxMsg &&
decryptedMsgWithPubKey != null &&
decryptedMsgWithPubKey.msg.equals(tradeMessage)) {
log.debug("Remove decryptedMsgWithPubKey from P2P network. decryptedMsgWithPubKey = " + decryptedMsgWithPubKey);
p2PService.removeEntryFromMailbox(decryptedMsgWithPubKey);
trade.removeDecryptedMsgWithPubKey(decryptedMsgWithPubKey);
}
}
}