com.arangodb.shaded.vertx.core.eventbus.impl.CodecManager Maven / Gradle / Ivy
/*
* Copyright (c) 2011-2022 Contributors to the Eclipse Foundation
*
* 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, or the Apache License, Version 2.0
* which is available at https://www.apache.org/licenses/LICENSE-2.0.
*
* SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
*/
package com.arangodb.shaded.vertx.core.eventbus.impl;
import com.arangodb.shaded.vertx.core.buffer.Buffer;
import com.arangodb.shaded.vertx.core.eventbus.EventBus;
import com.arangodb.shaded.vertx.core.eventbus.MessageCodec;
import com.arangodb.shaded.vertx.core.eventbus.ReplyException;
import com.arangodb.shaded.vertx.core.eventbus.impl.codecs.*;
import com.arangodb.shaded.vertx.core.json.JsonArray;
import com.arangodb.shaded.vertx.core.json.JsonObject;
import com.arangodb.shaded.vertx.core.shareddata.impl.ClusterSerializable;
import java.io.Serializable;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.function.Function;
/**
* @author Tim Fox
*/
public class CodecManager {
// The standard message codecs
public static final MessageCodec PING_MESSAGE_CODEC = new PingMessageCodec();
public static final MessageCodec NULL_MESSAGE_CODEC = new NullMessageCodec();
public static final MessageCodec STRING_MESSAGE_CODEC = new StringMessageCodec();
public static final MessageCodec BUFFER_MESSAGE_CODEC = new BufferMessageCodec();
public static final MessageCodec JSON_OBJECT_MESSAGE_CODEC = new JsonObjectMessageCodec();
public static final MessageCodec JSON_ARRAY_MESSAGE_CODEC = new JsonArrayMessageCodec();
public static final MessageCodec BYTE_ARRAY_MESSAGE_CODEC = new ByteArrayMessageCodec();
public static final MessageCodec INT_MESSAGE_CODEC = new IntMessageCodec();
public static final MessageCodec LONG_MESSAGE_CODEC = new LongMessageCodec();
public static final MessageCodec FLOAT_MESSAGE_CODEC = new FloatMessageCodec();
public static final MessageCodec DOUBLE_MESSAGE_CODEC = new DoubleMessageCodec();
public static final MessageCodec BOOLEAN_MESSAGE_CODEC = new BooleanMessageCodec();
public static final MessageCodec SHORT_MESSAGE_CODEC = new ShortMessageCodec();
public static final MessageCodec CHAR_MESSAGE_CODEC = new CharMessageCodec();
public static final MessageCodec BYTE_MESSAGE_CODEC = new ByteMessageCodec();
public static final MessageCodec REPLY_EXCEPTION_MESSAGE_CODEC = new ReplyExceptionMessageCodec();
private final MessageCodec[] systemCodecs;
private final ConcurrentMap userCodecMap = new ConcurrentHashMap<>();
private final ConcurrentMap defaultCodecMap = new ConcurrentHashMap<>();
private final ClusterSerializableCodec clusterSerializableCodec = new ClusterSerializableCodec(this);
private final SerializableCodec serializableCodec = new SerializableCodec(this);
private volatile Function clusterSerializableCheck = s -> Boolean.FALSE;
private volatile Function serializableCheck = EventBus.DEFAULT_SERIALIZABLE_CHECKER;
private volatile Function codecSelector = o -> null;
public CodecManager() {
this.systemCodecs = codecs(NULL_MESSAGE_CODEC, PING_MESSAGE_CODEC, STRING_MESSAGE_CODEC, BUFFER_MESSAGE_CODEC, JSON_OBJECT_MESSAGE_CODEC, JSON_ARRAY_MESSAGE_CODEC,
BYTE_ARRAY_MESSAGE_CODEC, INT_MESSAGE_CODEC, LONG_MESSAGE_CODEC, FLOAT_MESSAGE_CODEC, DOUBLE_MESSAGE_CODEC,
BOOLEAN_MESSAGE_CODEC, SHORT_MESSAGE_CODEC, CHAR_MESSAGE_CODEC, BYTE_MESSAGE_CODEC, REPLY_EXCEPTION_MESSAGE_CODEC,
clusterSerializableCodec, serializableCodec);
}
public MessageCodec lookupCodec(Object body, String codecName, boolean local) {
MessageCodec codec;
if (codecName != null) {
codec = getCodec(codecName);
} else if (body == null) {
codec = NULL_MESSAGE_CODEC;
} else if (body instanceof String) {
codec = STRING_MESSAGE_CODEC;
} else if (body instanceof Buffer) {
codec = BUFFER_MESSAGE_CODEC;
} else if (body instanceof JsonObject) {
codec = JSON_OBJECT_MESSAGE_CODEC;
} else if (body instanceof JsonArray) {
codec = JSON_ARRAY_MESSAGE_CODEC;
} else if (body instanceof byte[]) {
codec = BYTE_ARRAY_MESSAGE_CODEC;
} else if (body instanceof Integer) {
codec = INT_MESSAGE_CODEC;
} else if (body instanceof Long) {
codec = LONG_MESSAGE_CODEC;
} else if (body instanceof Float) {
codec = FLOAT_MESSAGE_CODEC;
} else if (body instanceof Double) {
codec = DOUBLE_MESSAGE_CODEC;
} else if (body instanceof Boolean) {
codec = BOOLEAN_MESSAGE_CODEC;
} else if (body instanceof Short) {
codec = SHORT_MESSAGE_CODEC;
} else if (body instanceof Character) {
codec = CHAR_MESSAGE_CODEC;
} else if (body instanceof Byte) {
codec = BYTE_MESSAGE_CODEC;
} else if (body instanceof ReplyException) {
codec = defaultCodecMap.get(body.getClass());
if (codec == null) {
codec = REPLY_EXCEPTION_MESSAGE_CODEC;
}
} else {
codec = defaultCodecMap.get(body.getClass());
if (codec == null) {
if ((codecName = codecSelector.apply(body)) != null) {
codec = getCodec(codecName);
} else if (body instanceof ClusterSerializable && (local || acceptClusterSerializable(body.getClass().getName()))) {
codec = clusterSerializableCodec;
} else if (body instanceof Serializable && (local || acceptSerializable(body.getClass().getName()))) {
codec = serializableCodec;
}
}
}
if (codec == null) {
throw new IllegalArgumentException("No message codec for type: " + body.getClass());
}
return codec;
}
public MessageCodec getCodec(String codecName) {
return userCodecMap.get(codecName);
}
public void registerCodec(MessageCodec codec) {
Objects.requireNonNull(codec, "codec");
Objects.requireNonNull(codec.name(), "code.name()");
checkSystemCodec(codec);
if (userCodecMap.containsKey(codec.name())) {
throw new IllegalStateException("Already a codec registered with name " + codec.name());
}
userCodecMap.put(codec.name(), codec);
}
public void unregisterCodec(String name) {
Objects.requireNonNull(name);
userCodecMap.remove(name);
}
public void registerDefaultCodec(Class clazz, MessageCodec codec) {
Objects.requireNonNull(clazz);
Objects.requireNonNull(codec, "codec");
Objects.requireNonNull(codec.name(), "code.name()");
checkSystemCodec(codec);
if (defaultCodecMap.containsKey(clazz)) {
throw new IllegalStateException("Already a default codec registered for class " + clazz);
}
if (userCodecMap.containsKey(codec.name())) {
throw new IllegalStateException("Already a codec registered with name " + codec.name());
}
defaultCodecMap.put(clazz, codec);
userCodecMap.put(codec.name(), codec);
}
public void unregisterDefaultCodec(Class clazz) {
Objects.requireNonNull(clazz);
MessageCodec codec = defaultCodecMap.remove(clazz);
if (codec != null) {
userCodecMap.remove(codec.name());
}
}
public MessageCodec[] systemCodecs() {
return systemCodecs;
}
private void checkSystemCodec(MessageCodec codec) {
if (codec.systemCodecID() != -1) {
throw new IllegalArgumentException("Can't register a system codec");
}
}
private MessageCodec[] codecs(MessageCodec... codecs) {
MessageCodec[] arr = new MessageCodec[codecs.length];
for (MessageCodec codec : codecs) {
arr[codec.systemCodecID()] = codec;
}
return arr;
}
public void clusterSerializableCheck(Function classNamePredicate) {
this.clusterSerializableCheck = Objects.requireNonNull(classNamePredicate);
}
public boolean acceptClusterSerializable(String className) {
return clusterSerializableCheck.apply(className);
}
public void serializableCheck(Function classNamePredicate) {
this.serializableCheck = Objects.requireNonNull(classNamePredicate);
}
public boolean acceptSerializable(String className) {
return serializableCheck.apply(className);
}
public void codecSelector(Function selector) {
this.codecSelector = Objects.requireNonNull(selector);
}
}