Skip to content
Original file line number Diff line number Diff line change
Expand Up @@ -23,14 +23,17 @@
class DefaultNatsManualAck implements NatsManualAck {

private final Message message;
private final Runnable onAcknowledged;

DefaultNatsManualAck(Message message) {
DefaultNatsManualAck(Message message, Runnable onAcknowledged) {
this.message = message;
this.onAcknowledged = onAcknowledged;
}

@Override
public void ack() {
message.ack();
onAcknowledged.run();
}

@Override
Expand All @@ -46,6 +49,7 @@ public void nakWithDelay(Duration delay) {
@Override
public void term() {
message.term();
onAcknowledged.run();
}

@Override
Expand Down
Loading
Loading