-
Notifications
You must be signed in to change notification settings - Fork 214
/
KafkaCompletableMessage.java
82 lines (69 loc) · 2.83 KB
/
KafkaCompletableMessage.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/*
* Copyright (c) 2021 Contributors to the Eclipse Foundation
*
* See the NOTICE file(s) distributed with this work for additional
* information regarding copyright ownership.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0
*
* SPDX-License-Identifier: EPL-2.0
*/
package org.eclipse.ditto.connectivity.service.messaging.kafka;
import java.util.Objects;
import java.util.concurrent.CompletableFuture;
import javax.annotation.concurrent.Immutable;
import org.eclipse.ditto.connectivity.api.ExternalMessage;
import org.eclipse.ditto.connectivity.service.messaging.AcknowledgeableMessage;
import org.apache.pekko.Done;
/**
* Wraps an {@link org.eclipse.ditto.connectivity.service.messaging.AcknowledgeableMessage} and provides a future of {@link org.apache.pekko.Done}
* which is completed successfully in all cases:
* <ul>
* <li>once the requested acknowledgement was issued/settled</li>
* <li>once the requested acknowledgement was rejected but not with redelivery request</li>
* <li>once the requested acknowledgement was rejected with redelivery request</li>
* </ul>
* Its purpose is to know when a message has been fully processed.
*/
@Immutable
final class KafkaCompletableMessage {
private final AcknowledgeableMessage acknowledgeableMessage;
private final CompletableFuture<Done> acknowledgementFuture;
KafkaCompletableMessage(final ExternalMessage message) {
this.acknowledgementFuture = new CompletableFuture<>();
this.acknowledgeableMessage = AcknowledgeableMessage.of(message,
() -> acknowledgementFuture.complete(Done.getInstance()),
shouldRedeliver -> acknowledgementFuture.complete(Done.getInstance()));
}
AcknowledgeableMessage getAcknowledgeableMessage() {
return acknowledgeableMessage;
}
CompletableFuture<Done> getAcknowledgementFuture() {
return acknowledgementFuture;
}
@Override
public boolean equals(final Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
final KafkaCompletableMessage that = (KafkaCompletableMessage) o;
return Objects.equals(acknowledgeableMessage, that.acknowledgeableMessage) &&
Objects.equals(acknowledgementFuture, that.acknowledgementFuture);
}
@Override
public int hashCode() {
return Objects.hash(acknowledgeableMessage, acknowledgementFuture);
}
@Override
public String toString() {
return getClass().getSimpleName() + " [" +
"acknowledgeableMessage=" + acknowledgeableMessage +
", acknowledgementFuture=" + acknowledgementFuture +
"]";
}
}