did/did-sdk/src/main/java/io/github/ehlxr/did/client/SdkClient.java

package io.github.ehlxr.did.client;

import io.github.ehlxr.did.client.handler.SdkClientDecoder;
import io.github.ehlxr.did.client.handler.SdkClientEncoder;
import io.github.ehlxr.did.client.handler.SdkClientHandler;
import io.github.ehlxr.did.common.Constants;
import io.netty.channel.ChannelFutureListener;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.socket.SocketChannel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.InetSocketAddress;
import java.util.Objects;

/**
 * @author ehlxr
 */
public class SdkClient extends AbstractClient {
    private final Logger logger = LoggerFactory.getLogger(SdkClient.class);

    public SdkClient(String host, int port) {
        this(host, port, 0);
    }

    public SdkClient(String host, int port, int timeoutMillis) {
        this.host = Objects.isNull(host) ?
                "".equals(Constants.getEnv("SDK_HOST")) ? Constants.SERVER_HOST : Constants.getEnv("HTTP_PORT") :
                host;

        this.port = port <= 0 ?
                "".equals(Constants.getEnv("SDK_PORT")) ? Constants.SDK_PORT : Integer.parseInt(Constants.getEnv("SDK_PORT")) :
                port;

        this.timeoutMillis = timeoutMillis <= 0 ? Constants.SDK_CLIENT_TIMEOUTMILLIS : timeoutMillis;
    }

    public SdkClient() {
        this(null, 0);
    }

    public static SdkClientBuilder newBuilder() {
        return new SdkClientBuilder();
    }

    @Override
    public void start() {
        init();

        bootstrap.handler(new ChannelInitializer<SocketChannel>() {
            @Override
            protected void initChannel(SocketChannel socketChannel) {
                socketChannel.pipeline().addLast("SdkServerDecoder", new SdkClientDecoder(12))
                        .addLast("SdkServerEncoder", new SdkClientEncoder())
                        .addLast("SdkClientHandler", new SdkClientHandler());
            }
        });

        try {
            channelFuture = bootstrap.connect(host, port)
                    .sync()
                    .channel()
                    .closeFuture()
                    .addListener((ChannelFutureListener) channelFuture ->
                            logger.warn("client channel close.", channelFuture.cause()));

            InetSocketAddress address = (InetSocketAddress) channelFuture.channel().remoteAddress();
            logger.info("SdkClient start success, host is {}, port is {}", address.getHostName(), address.getPort());
        } catch (InterruptedException e) {
            logger.error("SdkClient start error", e);
            shutdown();
        }
    }

    public static final class SdkClientBuilder {
        int timeoutMillis;
        String host;
        int port;

        private SdkClientBuilder() {
        }

        public static SdkClientBuilder aSdkClient() {
            return new SdkClientBuilder();
        }

        public SdkClientBuilder timeoutMillis(int timeoutMillis) {
            this.timeoutMillis = timeoutMillis;
            return this;
        }

        public SdkClientBuilder host(String host) {
            this.host = host;
            return this;
        }

        public SdkClientBuilder port(int port) {
            this.port = port;
            return this;
        }

        public SdkClient build() {
            return new SdkClient(host, port, timeoutMillis);
        }
    }
}