org.apache.mina.filter.codec.demux.DemuxingProtocolEncoder Maven / Gradle / Ivy
/**
* Copyright 2007-2015, Kaazing Corporation. All rights reserved.
*
* 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 org.apache.mina.filter.codec.demux;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
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 final AttributeKey STATE = new AttributeKey(getClass(), "state");
@SuppressWarnings("unchecked")
private final Map, MessageEncoderFactory> type2encoderFactory = new CopyOnWriteMap, MessageEncoderFactory>();
private static final Class[] EMPTY_PARAMS = new Class[0];
public DemuxingProtocolEncoder() {
// Do nothing
}
@SuppressWarnings("unchecked")
public void addMessageEncoder(Class messageType, Class encoderClass) {
if (encoderClass == null) {
throw new NullPointerException("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);
}
}
@SuppressWarnings("unchecked")
public void addMessageEncoder(Class messageType, MessageEncoder encoder) {
addMessageEncoder(messageType, new SingletonMessageEncoderFactory(encoder));
}
public void addMessageEncoder(Class messageType, MessageEncoderFactory factory) {
if (messageType == null) {
throw new NullPointerException("messageType");
}
if (factory == null) {
throw new NullPointerException("factory");
}
synchronized (type2encoderFactory) {
if (type2encoderFactory.containsKey(messageType)) {
throw new IllegalStateException(
"The specified message type (" + messageType.getName() + ") is registered already.");
}
type2encoderFactory.put(messageType, factory);
}
}
@SuppressWarnings("unchecked")
public void addMessageEncoder(Iterable> messageTypes, Class encoderClass) {
for (Class messageType : messageTypes) {
addMessageEncoder(messageType, encoderClass);
}
}
public void addMessageEncoder(Iterable> messageTypes, MessageEncoder encoder) {
for (Class messageType : messageTypes) {
addMessageEncoder(messageType, encoder);
}
}
public void addMessageEncoder(Iterable> messageTypes, MessageEncoderFactory factory) {
for (Class messageType : messageTypes) {
addMessageEncoder(messageType, factory);
}
}
public void encode(IoSession session, Object message,
ProtocolEncoderOutput out) throws Exception {
State state = getState(session);
MessageEncoder