Kafka header mapper for (un)marshalling
Signed-off-by: Fabio José <fabiojose@gmail.com>
This commit is contained in:
parent
9b41ae001d
commit
c11ca15fc4
|
@ -0,0 +1,69 @@
|
||||||
|
package io.cloudevents.v03.kafka;
|
||||||
|
|
||||||
|
import static io.cloudevents.v03.kafka.AttributeMapper.HEADER_PREFIX;
|
||||||
|
|
||||||
|
import java.util.Locale;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Objects;
|
||||||
|
import java.util.AbstractMap.SimpleEntry;
|
||||||
|
import java.util.Map.Entry;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
import org.apache.kafka.common.serialization.Serdes;
|
||||||
|
import org.apache.kafka.common.serialization.Serializer;
|
||||||
|
|
||||||
|
import io.cloudevents.extensions.ExtensionFormat;
|
||||||
|
import io.cloudevents.fun.FormatHeaderMapper;
|
||||||
|
import io.cloudevents.v03.AttributesImpl;
|
||||||
|
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
* @author fabiojose
|
||||||
|
* @version 0.3
|
||||||
|
*/
|
||||||
|
public class HeaderMapper {
|
||||||
|
private HeaderMapper() {}
|
||||||
|
|
||||||
|
private static final Serializer<String> SERIALIZER =
|
||||||
|
Serdes.String().serializer();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Following the signature of {@link FormatHeaderMapper}
|
||||||
|
* @param attributes The map of attributes created by
|
||||||
|
* {@link AttributesImpl#marshal(AttributesImpl)}
|
||||||
|
* @param extensions The map of extensions created by
|
||||||
|
* {@link ExtensionFormat#marshal(java.util.Collection)}
|
||||||
|
* @return The map of Kafka Headers with values as {@code byte[]}
|
||||||
|
*/
|
||||||
|
public static Map<String, byte[]> map(Map<String, String> attributes,
|
||||||
|
Map<String, String> extensions) {
|
||||||
|
Objects.requireNonNull(attributes);
|
||||||
|
Objects.requireNonNull(extensions);
|
||||||
|
|
||||||
|
Map<String, byte[]> result = attributes.entrySet()
|
||||||
|
.stream()
|
||||||
|
.filter(attribute -> null!= attribute.getValue())
|
||||||
|
.map(attribute ->
|
||||||
|
new SimpleEntry<>(attribute.getKey()
|
||||||
|
.toLowerCase(Locale.US), attribute.getValue()))
|
||||||
|
.map(attribute ->
|
||||||
|
new SimpleEntry<>(HEADER_PREFIX+attribute.getKey(),
|
||||||
|
attribute.getValue()))
|
||||||
|
.map(attribute ->
|
||||||
|
new SimpleEntry<>(attribute.getKey(),
|
||||||
|
SERIALIZER.serialize(null, attribute.getValue())))
|
||||||
|
.collect(Collectors.toMap(Entry::getKey, Entry::getValue));
|
||||||
|
|
||||||
|
result.putAll(
|
||||||
|
extensions.entrySet()
|
||||||
|
.stream()
|
||||||
|
.filter(extension -> null!= extension.getValue())
|
||||||
|
.map(extension ->
|
||||||
|
new SimpleEntry<>(extension.getKey(),
|
||||||
|
SERIALIZER.serialize(null, extension.getValue())))
|
||||||
|
.collect(Collectors.toMap(Entry::getKey, Entry::getValue))
|
||||||
|
);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,148 @@
|
||||||
|
/**
|
||||||
|
* Copyright 2019 The CloudEvents Authors
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package io.cloudevents.v03.kafka;
|
||||||
|
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertFalse;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertThrows;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
import io.cloudevents.v03.kafka.HeaderMapper;
|
||||||
|
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
* @author fabiojose
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
public class HeaderMapperTest {
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void error_when_attributes_map_isnull() {
|
||||||
|
// setup
|
||||||
|
Map<String, String> extensions = new HashMap<>();
|
||||||
|
|
||||||
|
assertThrows(NullPointerException.class, () -> {
|
||||||
|
// act
|
||||||
|
HeaderMapper.map(null, extensions);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void error_when_extensions_map_isnull() {
|
||||||
|
// setup
|
||||||
|
Map<String, String> attributes = new HashMap<>();
|
||||||
|
|
||||||
|
assertThrows(NullPointerException.class, () -> {
|
||||||
|
// act
|
||||||
|
HeaderMapper.map(attributes, null);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void should_not_map_null_attribute_value() {
|
||||||
|
// setup
|
||||||
|
Map<String, String> attributes = new HashMap<>();
|
||||||
|
attributes.put("type", null);
|
||||||
|
attributes.put("specversion", "0.3");
|
||||||
|
|
||||||
|
Map<String, String> extensions = new HashMap<>();
|
||||||
|
|
||||||
|
// act
|
||||||
|
Map<String, byte[]> actual = HeaderMapper.map(attributes, extensions);
|
||||||
|
|
||||||
|
//assert
|
||||||
|
assertFalse(actual.containsKey("ce-type"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void should_not_map_null_extension_value() {
|
||||||
|
// setup
|
||||||
|
Map<String, String> attributes = new HashMap<>();
|
||||||
|
attributes.put("type", "mytype");
|
||||||
|
attributes.put("specversion", "0.3");
|
||||||
|
|
||||||
|
Map<String, String> extensions = new HashMap<>();
|
||||||
|
extensions.put("null-ext", null);
|
||||||
|
extensions.put("comexampleextension1", "value");
|
||||||
|
|
||||||
|
// act
|
||||||
|
Map<String, byte[]> actual = HeaderMapper.map(attributes, extensions);
|
||||||
|
|
||||||
|
//assert
|
||||||
|
assertFalse(actual.containsKey("null-ext"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void should_not_map_absent_contenttype() {
|
||||||
|
// setup
|
||||||
|
Map<String, String> attributes = new HashMap<>();
|
||||||
|
attributes.put("type", "mytype");
|
||||||
|
attributes.put("specversion", "0.3");
|
||||||
|
|
||||||
|
Map<String, String> extensions = new HashMap<>();
|
||||||
|
extensions.put("null-ext", "null-value");
|
||||||
|
extensions.put("comexampleextension1", "value");
|
||||||
|
|
||||||
|
// act
|
||||||
|
Map<String, byte[]> actual = HeaderMapper.map(attributes, extensions);
|
||||||
|
|
||||||
|
//assert
|
||||||
|
assertFalse(actual.containsKey("Content-Type"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void should_map_extension_without_prefix() {
|
||||||
|
// setup
|
||||||
|
Map<String, String> attributes = new HashMap<>();
|
||||||
|
attributes.put("type", "mytype");
|
||||||
|
attributes.put("specversion", "0.3");
|
||||||
|
|
||||||
|
Map<String, String> extensions = new HashMap<>();
|
||||||
|
extensions.put("null-ext", "null-value");
|
||||||
|
extensions.put("comexampleextension1", "value");
|
||||||
|
|
||||||
|
// act
|
||||||
|
Map<String, byte[]> actual = HeaderMapper.map(attributes, extensions);
|
||||||
|
|
||||||
|
//assert
|
||||||
|
assertTrue(actual.containsKey("comexampleextension1"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void should_all_values_as_byte_array() {
|
||||||
|
// setup
|
||||||
|
Map<String, String> attributes = new HashMap<>();
|
||||||
|
attributes.put("type", "mytype");
|
||||||
|
attributes.put("specversion", "0.3");
|
||||||
|
|
||||||
|
Map<String, String> extensions = new HashMap<>();
|
||||||
|
extensions.put("null-ext", "null-value");
|
||||||
|
extensions.put("comexampleextension1", "value");
|
||||||
|
|
||||||
|
// act
|
||||||
|
Map<String, byte[]> actuals = HeaderMapper.map(attributes, extensions);
|
||||||
|
|
||||||
|
// assert
|
||||||
|
actuals.values()
|
||||||
|
.forEach(actual -> {
|
||||||
|
assertTrue(actual instanceof byte[]);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue