LocalClientSessionFactory.java

/*
 * 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 com.ok2c.lightmtp.impl.protocol;

import org.apache.http.nio.reactor.IOSession;
import org.apache.http.util.Args;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.ok2c.lightmtp.protocol.DeliveryRequestHandler;
import com.ok2c.lightmtp.protocol.ProtocolCodecs;
import com.ok2c.lightmtp.protocol.SessionFactory;

public class LocalClientSessionFactory implements SessionFactory<ClientSession> {

    private final Logger iolog = LoggerFactory.getLogger(IOSession.class);
    private final Logger wirelog = LoggerFactory.getLogger(Wire.WIRELOG_CAT);

    private final DeliveryRequestHandler deliveryRequestHandler;
	private final String heloName;

    public LocalClientSessionFactory(
            final DeliveryRequestHandler deliveryRequestHandler, final String heloName) {
        super();
        Args.notNull(deliveryRequestHandler, "Delivery request handler");
        this.deliveryRequestHandler = deliveryRequestHandler;
        this.heloName = heloName;
    }
    public LocalClientSessionFactory(
            final DeliveryRequestHandler deliveryRequestHandler) {
        this(deliveryRequestHandler, null);
    }

    @Override
    public ClientSession create(final IOSession iosession) {
        SMTPBuffers iobuffers = new SMTPBuffers();
        ProtocolCodecs<ClientState> codecs = new ProtocolCodecRegistry<ClientState>();
        codecs.register(ProtocolState.HELO.name(),
                new SendLocalHeloCodec(iobuffers, heloName));
        codecs.register(ProtocolState.MAIL.name(),
                new PipeliningSendEnvelopCodec(iobuffers, true));
        codecs.register(ProtocolState.DATA.name(),
                new SendDataCodec(iobuffers, true, DataAckMode.PER_RECIPIENT));
        codecs.register(ProtocolState.QUIT.name(),
                new SendQuitCodec(iobuffers));
        codecs.register(ProtocolState.RSET.name(),
                new SendRsetCodec(iobuffers));
        final IOSession localIOSession;
        if (iolog.isDebugEnabled() || wirelog.isDebugEnabled()) {
            localIOSession = new LoggingIOSession(iosession, "LMTP client", iolog, wirelog);
        } else {
            localIOSession = iosession;
        }
        return new ClientSession(localIOSession, iobuffers, this.deliveryRequestHandler, codecs);
    }

}