org.apache.mina.filter.codec.demux.DemuxingProtocolEncoder Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.mina.filter.codec.demux;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import org.apache.mina.core.session.AttributeKey;
import org.apache.mina.core.session.IoSession;
import org.apache.mina.core.session.UnknownMessageTypeException;
import org.apache.mina.filter.codec.ProtocolEncoder;
import org.apache.mina.filter.codec.ProtocolEncoderOutput;
import org.apache.mina.util.CopyOnWriteMap;
import org.apache.mina.util.IdentityHashSet;
/**
* A composite {@link ProtocolEncoder} that demultiplexes incoming message
* encoding requests into an appropriate {@link MessageEncoder}.
*
* Disposing resources acquired by {@link MessageEncoder}
*
* Override {@link #dispose(IoSession)} method. Please don't forget to call
* super.dispose().
*
* @author Apache MINA Project
*
* @see MessageEncoderFactory
* @see MessageEncoder
*/
public class DemuxingProtocolEncoder implements ProtocolEncoder {
private static final AttributeKey STATE = new AttributeKey(DemuxingProtocolEncoder.class, "state");
@SuppressWarnings("rawtypes")
private final Map, MessageEncoderFactory> type2encoderFactory = new CopyOnWriteMap<>();
private static final Class[] EMPTY_PARAMS = new Class[0];
/**
* Add a new message encoder class for a given message type
*
* @param messageType The message type
* @param encoderClass The encoder class
*/
@SuppressWarnings({ "rawtypes", "unchecked" })
public void addMessageEncoder(Class messageType, Class encoderClass) {
if (encoderClass == null) {
throw new IllegalArgumentException("encoderClass");
}
try {
encoderClass.getConstructor(EMPTY_PARAMS);
} catch (NoSuchMethodException e) {
throw new IllegalArgumentException("The specified class doesn't have a public default constructor.");
}
boolean registered = false;
if (MessageEncoder.class.isAssignableFrom(encoderClass)) {
addMessageEncoder(messageType, new DefaultConstructorMessageEncoderFactory(encoderClass));
registered = true;
}
if (!registered) {
throw new IllegalArgumentException("Unregisterable type: " + encoderClass);
}
}
/**
* Add a new message encoder instance for a given message type
*
* @param The message type
* @param messageType The message type
* @param encoder The encoder instance
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
public void addMessageEncoder(Class messageType, MessageEncoder encoder) {
addMessageEncoder(messageType, new SingletonMessageEncoderFactory(encoder));
}
/**
* Add a new message encoder factory for a given message type
*
* @param The message type
* @param messageType The message type
* @param factory The encoder factory
*/
public void addMessageEncoder(Class messageType, MessageEncoderFactory factory) {
if (messageType == null) {
throw new IllegalArgumentException("messageType");
}
if (factory == null) {
throw new IllegalArgumentException("factory");
}
synchronized (type2encoderFactory) {
if (type2encoderFactory.containsKey(messageType)) {
throw new IllegalStateException("The specified message type (" + messageType.getName()
+ ") is registered already.");
}
type2encoderFactory.put(messageType, factory);
}
}
/**
* Add a new message encoder class for a list of message types
*
* @param messageTypes The message types
* @param encoderClass The encoder class
*/
@SuppressWarnings("rawtypes")
public void addMessageEncoder(Iterable> messageTypes, Class encoderClass) {
for (Class messageType : messageTypes) {
addMessageEncoder(messageType, encoderClass);
}
}
/**
* Add a new message instance class for a list of message types
*
* @param The message type
* @param messageTypes The message types
* @param encoder The encoder instance
*/
public void addMessageEncoder(Iterable> messageTypes, MessageEncoder encoder) {
for (Class messageType : messageTypes) {
addMessageEncoder(messageType, encoder);
}
}
/**
* Add a new message encoder factory for a list of message types
*
* @param The message type
* @param messageTypes The message types
* @param factory The encoder factory
*/
public void addMessageEncoder(Iterable> messageTypes,
MessageEncoderFactory factory) {
for (Class messageType : messageTypes) {
addMessageEncoder(messageType, factory);
}
}
/**
* {@inheritDoc}
*/
@Override
public void encode(IoSession session, Object message, ProtocolEncoderOutput out) throws Exception {
State state = getState(session);
MessageEncoder