-
Notifications
You must be signed in to change notification settings - Fork 214
/
DittoMessageMapper.java
99 lines (85 loc) · 3.98 KB
/
DittoMessageMapper.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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*
* Copyright (c) 2017 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.mapping;
import static java.util.Collections.singletonList;
import java.util.List;
import org.eclipse.ditto.base.model.common.DittoConstants;
import org.eclipse.ditto.base.model.exceptions.DittoJsonException;
import org.eclipse.ditto.base.model.headers.DittoHeaders;
import org.eclipse.ditto.connectivity.api.ExternalMessage;
import org.eclipse.ditto.connectivity.api.ExternalMessageFactory;
import org.eclipse.ditto.connectivity.model.ConnectivityModelFactory;
import org.eclipse.ditto.connectivity.model.MappingContext;
import org.eclipse.ditto.json.JsonFactory;
import org.eclipse.ditto.json.JsonObject;
import org.eclipse.ditto.protocol.Adaptable;
import org.eclipse.ditto.protocol.JsonifiableAdaptable;
import org.eclipse.ditto.protocol.ProtocolFactory;
/**
* A message mapper implementation for the Ditto Protocol.
* Expects messages to contain a JSON serialized Ditto Protocol message.
*/
@PayloadMapper(
alias = {"Ditto",
// legacy full qualified name
"org.eclipse.ditto.connectivity.service.mapping.DittoMessageMapper"})
public final class DittoMessageMapper extends AbstractMessageMapper {
static final JsonObject DEFAULT_OPTIONS = JsonObject.newBuilder()
.set(MessageMapperConfiguration.CONTENT_TYPE_BLOCKLIST,
String.join(",", "application/vnd.eclipse-hono-empty-notification",
"application/vnd.eclipse-hono-device-provisioning-notification",
"application/vnd.eclipse-hono-dc-notification+json",
"application/vnd.eclipse-hono-delivery-failure-notification+json"
))
.build();
/**
* The context representing this mapper
*/
public static final MappingContext CONTEXT = ConnectivityModelFactory.newMappingContextBuilder(
DittoMessageMapper.class.getCanonicalName(),
DEFAULT_OPTIONS
).build();
@Override
public List<Adaptable> map(final ExternalMessage message) {
final String payload = extractPayloadAsString(message);
final JsonifiableAdaptable jsonifiableAdaptable = DittoJsonException.wrapJsonRuntimeException(() ->
ProtocolFactory.jsonifiableAdaptableFromJson(JsonFactory.newObject(payload))
);
final DittoHeaders mergedHeaders = jsonifiableAdaptable.getDittoHeaders();
return singletonList(
ProtocolFactory.newAdaptableBuilder(jsonifiableAdaptable).withHeaders(mergedHeaders).build());
}
@Override
public List<ExternalMessage> map(final Adaptable adaptable) {
return List.of(ExternalMessageFactory.newExternalMessageBuilder(getExternalDittoHeaders(adaptable))
.withTopicPath(adaptable.getTopicPath())
.withText(getJsonString(adaptable))
.asResponse(isResponse(adaptable))
.asError(isError(adaptable))
.build());
}
private static DittoHeaders getExternalDittoHeaders(final Adaptable adaptable) {
return DittoHeaders.newBuilder()
.contentType(DittoConstants.DITTO_PROTOCOL_CONTENT_TYPE)
.correlationId(adaptable.getDittoHeaders().getCorrelationId().orElse(null))
.build();
}
private static String getJsonString(final Adaptable adaptable) {
final var jsonifiableAdaptable = ProtocolFactory.wrapAsJsonifiableAdaptable(adaptable);
return jsonifiableAdaptable.toJsonString();
}
@Override
public JsonObject getDefaultOptions() {
return DEFAULT_OPTIONS;
}
}