net.ymate.platform.serv.nio.support.NioEventGroup.java Source code

Introduction

Here is the source code for net.ymate.platform.serv.nio.support.NioEventGroup.java

Source

/*
 * Copyright 2007-2016 the original author or 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 net.ymate.platform.serv.nio.support;

import net.ymate.platform.serv.AbstractEventGroup;
import net.ymate.platform.serv.IClientCfg;
import net.ymate.platform.serv.IListener;
import net.ymate.platform.serv.nio.INioClientCfg;
import net.ymate.platform.serv.nio.INioCodec;
import net.ymate.platform.serv.nio.INioServerCfg;
import net.ymate.platform.serv.nio.INioSession;
import org.apache.commons.lang.StringUtils;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.SelectableChannel;
import java.nio.channels.SelectionKey;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author  (suninformation@163.com) on 15/11/15 ?6:54
 * @version 1.0
 */
public class NioEventGroup<LISTENER extends IListener<INioSession>>
        extends AbstractEventGroup<INioCodec, LISTENER, INioSession> {

    protected SelectableChannel __channel;
    protected int __selectorCount = 1;

    protected NioEventProcessor[] __processors;
    protected AtomicInteger __handlerIndex = new AtomicInteger(0);

    public NioEventGroup(INioServerCfg cfg, LISTENER listener, INioCodec codec) throws IOException {
        super(cfg, listener, codec);
        //
        __channel = __doChannelCreate(cfg);
        __selectorCount = cfg.getSelectorCount();
    }

    public NioEventGroup(INioClientCfg cfg, LISTENER listener, INioCodec codec) throws IOException {
        super(cfg, listener, codec);
    }

    protected SelectableChannel __doChannelCreate(INioServerCfg cfg) throws IOException {
        ServerSocketChannel _channel = ServerSocketChannel.open();
        _channel.configureBlocking(false);
        _channel.socket().bind(new InetSocketAddress(cfg.getServerHost(), cfg.getPort()));
        return _channel;
    }

    @SuppressWarnings("unchecked")
    protected INioSession __doSessionCreate(IClientCfg cfg) throws IOException {
        SocketChannel _channel = SocketChannel.open();
        _channel.configureBlocking(false);
        _channel.socket().setReuseAddress(true);
        _channel.connect(new InetSocketAddress(cfg.getRemoteHost(), cfg.getPort()));
        __channel = _channel;
        return new NioSession(this, _channel);
    }

    @Override
    public synchronized void start() throws IOException {
        super.start();
        //
        __doInitProcessors();
        __doRegisterEvent();
    }

    protected String __doBuildProcessorName() {
        return StringUtils.capitalize(name()).concat(isServer() ? "Server" : "Client")
                .concat("-NioEventProcessor-");
    }

    protected void __doInitProcessors() throws IOException {
        __processors = new NioEventProcessor[__selectorCount];
        for (int _idx = 0; _idx < __selectorCount; _idx++) {
            __processors[_idx] = new NioEventProcessor<LISTENER>(this, __doBuildProcessorName() + _idx);
            __processors[_idx].start();
        }
    }

    protected void __doRegisterEvent() throws IOException {
        if (isServer()) {
            processor().registerEvent(__channel, SelectionKey.OP_ACCEPT, null);
        } else {
            processor().registerEvent(__channel, SelectionKey.OP_CONNECT, session());
            if (connectionTimeout() > 0) {
                session().connectSync(connectionTimeout());
            }
        }
    }

    @Override
    public synchronized void stop() throws IOException {
        for (NioEventProcessor _processor : __processors) {
            _processor.interrupt();
        }
        __channel.close();
        __channel = null;
        //
        super.stop();
    }

    public NioEventProcessor processor(SelectionKey key) {
        for (NioEventProcessor _processor : __processors) {
            if (key.selector() == _processor.__selector) {
                return _processor;
            }
        }
        return null;
    }

    public NioEventProcessor processor() {
        int _nextIdx = __handlerIndex.getAndIncrement() % __selectorCount;
        if (_nextIdx < 0) {
            __handlerIndex.set(0);
            _nextIdx = 0;
        }
        return __processors[_nextIdx];
    }
}