org.reaktivity.nukleus.tcp.internal.TcpNukleusFactorySpi Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2016-2017 The Reaktivity Project
 *
 * The Reaktivity Project 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.reaktivity.nukleus.tcp.internal;

import org.reaktivity.nukleus.Configuration;
import org.reaktivity.nukleus.Nukleus;
import org.reaktivity.nukleus.NukleusBuilder;
import org.reaktivity.nukleus.NukleusFactorySpi;
import org.reaktivity.nukleus.function.MessagePredicate;
import org.reaktivity.nukleus.route.RouteKind;
import org.reaktivity.nukleus.tcp.internal.poller.Poller;
import org.reaktivity.nukleus.tcp.internal.stream.Acceptor;
import org.reaktivity.nukleus.tcp.internal.stream.ClientStreamFactoryBuilder;
import org.reaktivity.nukleus.tcp.internal.stream.ServerStreamFactoryBuilder;

public final class TcpNukleusFactorySpi implements NukleusFactorySpi
{
    public static final String NAME = "tcp";

    @Override
    public String name()
    {
        return NAME;
    }

    @Override
    public Nukleus create(
        Configuration config,
        NukleusBuilder builder)
    {
        TcpConfiguration tcpConfig = new TcpConfiguration(config);
        Acceptor acceptor = new Acceptor(tcpConfig);
        Poller poller = new Poller();
        acceptor.setPoller(poller);

        ServerStreamFactoryBuilder serverStreamFactoryBuilder = new ServerStreamFactoryBuilder(tcpConfig, acceptor, poller);
        ClientStreamFactoryBuilder clientStreamFactoryBuilder = new ClientStreamFactoryBuilder(tcpConfig, poller);

        final MessagePredicate serverRouteHandler = (m, b, i, l) ->
            acceptor.handleRoute(m, b, i, l) &&
            serverStreamFactoryBuilder.handleRoute(m, b, i, l);

        return builder.streamFactory(RouteKind.CLIENT, clientStreamFactoryBuilder)
                      .streamFactory(RouteKind.SERVER, serverStreamFactoryBuilder)
                      .routeHandler(RouteKind.SERVER, serverRouteHandler)
                      .routeHandler(RouteKind.CLIENT, clientStreamFactoryBuilder::handleRoute)
                      .allowZeroRouteRef(RouteKind.SERVER::equals)
                      .layoutTarget(RouteKind.SERVER::equals)
                      .layoutSource(RouteKind.CLIENT::equals)
                      .inject(poller)
                      .build();

    }
}