io.reactivesocket.tckdrivers.client.JavaTCPClient.java Source code

Introduction

Here is the source code for io.reactivesocket.tckdrivers.client.JavaTCPClient.java

Source

/*
 * Copyright 2016 Facebook, Inc.
 * <p>
 *  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
 *  <p>
 *  http://www.apache.org/licenses/LICENSE-2.0
 *  <p>
 *  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 io.reactivesocket.tckdrivers.client;

import io.netty.buffer.ByteBuf;
import io.netty.handler.logging.LogLevel;
import io.reactivesocket.ConnectionSetupPayload;
import io.reactivesocket.ReactiveSocket;
import io.reactivesocket.transport.tcp.client.TcpReactiveSocketConnector;
import io.reactivex.netty.protocol.tcp.client.TcpClient;

import java.net.*;
import java.util.List;
import java.util.function.Function;

import static rx.RxReactiveStreams.toObservable;

/**
 * A client that implements a method to create ReactiveSockets, and runs the tests.
 */
public class JavaTCPClient {

    private static URI uri;
    private static boolean debug;

    public void run(String realfile, String host, int port, boolean debug2, List<String> tests)
            throws MalformedURLException, URISyntaxException {
        debug = debug2;
        // we pass in our reactive socket here to the test suite
        String file = "reactivesocket-tck-drivers/src/test/resources/client.txt";
        if (realfile != null)
            file = realfile;
        try {
            setURI(new URI("tcp://" + host + ":" + port + "/rs"));
            JavaClientDriver jd = new JavaClientDriver(file, JavaTCPClient::createClient, tests);
            jd.runTests();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void setURI(URI uri2) {
        uri = uri2;
    }

    /**
     * A function that creates a ReactiveSocket on a new TCP connection.
     * @return a ReactiveSocket
     */
    public static ReactiveSocket createClient() {
        ConnectionSetupPayload setupPayload = ConnectionSetupPayload.create("", "");

        if ("tcp".equals(uri.getScheme())) {
            Function<SocketAddress, TcpClient<ByteBuf, ByteBuf>> clientFactory = socketAddress -> TcpClient
                    .newClient(socketAddress);

            if (debug)
                clientFactory = socketAddress -> TcpClient.newClient(socketAddress).enableWireLogging("rs",
                        LogLevel.ERROR);

            return toObservable(
                    TcpReactiveSocketConnector.create(setupPayload, Throwable::printStackTrace, clientFactory)
                            .connect(new InetSocketAddress(uri.getHost(), uri.getPort()))).toSingle().toBlocking()
                                    .value();
        } else {
            throw new UnsupportedOperationException("uri unsupported: " + uri);
        }
    }

}