Examples with DefaultTcpServerMetrics org.apache.servicecomb.foundation.vertx.metrics.DefaultTcpServerMetrics used on opensource projects

Example 1 with DefaultTcpServerMetrics

use of org.apache.servicecomb.foundation.vertx.metrics.DefaultTcpServerMetrics in project java-chassis by ServiceComb.

the class TcpServer method init.

public void init(Vertx vertx, String sslKey, AsyncResultCallback<InetSocketAddress> callback) {
    NetServer netServer;
    if (endpointObject.isSslEnabled()) {
        SSLOptionFactory factory = SSLOptionFactory.createSSLOptionFactory(sslKey, null);
        SSLOption sslOption;
        if (factory == null) {
            sslOption = SSLOption.buildFromYaml(sslKey);
        } else {
            sslOption = factory.createSSLOption();
        }
        SSLCustom sslCustom = SSLCustom.createSSLCustom(sslOption.getSslCustomClass());
        NetServerOptions serverOptions = new NetServerOptions();
        VertxTLSBuilder.buildNetServerOptions(sslOption, sslCustom, serverOptions);
        netServer = vertx.createNetServer(serverOptions);
    } else {
        netServer = vertx.createNetServer();
    }
    netServer.connectHandler(netSocket -> {
        DefaultTcpServerMetrics serverMetrics = (DefaultTcpServerMetrics) ((NetSocketImpl) netSocket).metrics();
        DefaultServerEndpointMetric endpointMetric = serverMetrics.getEndpointMetric();
        long connectedCount = endpointMetric.getCurrentConnectionCount();
        int connectionLimit = getConnectionLimit();
        if (connectedCount > connectionLimit) {
            netSocket.close();
            endpointMetric.onRejectByConnectionLimit();
            return;
        }
        TcpServerConnection connection = createTcpServerConnection();
        connection.init(netSocket);
    });
    netServer.exceptionHandler(e -> {
        LOGGER.error("Unexpected error in server.{}", ExceptionUtils.getExceptionMessageWithoutTrace(e));
    });
    InetSocketAddress socketAddress = endpointObject.getSocketAddress();
    netServer.listen(socketAddress.getPort(), socketAddress.getHostString(), ar -> {
        if (ar.succeeded()) {
            callback.success(socketAddress);
            return;
        }
        // 监听失败
        String msg = String.format("listen failed, address=%s", socketAddress.toString());
        callback.fail(new Exception(msg, ar.cause()));
    });
}

Example 2 with DefaultTcpServerMetrics

use of org.apache.servicecomb.foundation.vertx.metrics.DefaultTcpServerMetrics in project java-chassis by ServiceComb.

the class TestTcpServer method testConnectionLimit.

@SuppressWarnings({ "rawtypes", "unchecked" })
@Test
public void testConnectionLimit(@Mocked Vertx vertx, @Mocked AsyncResultCallback<InetSocketAddress> callback, @Mocked NetServer netServer, @Mocked NetSocketImpl netSocket) {
    DefaultServerEndpointMetric endpointMetric = new DefaultServerEndpointMetric(null);
    DefaultTcpServerMetrics tcpServerMetrics = new DefaultTcpServerMetrics(endpointMetric);
    new MockUp<NetServer>(netServer) {

        @Mock
        NetServer connectHandler(Handler<NetSocket> handler) {
            connectHandler = handler;
            return netServer;
        }
    };
    new MockUp<NetSocketImpl>(netSocket) {

        @Mock
        void close() {
            netSocketClosed = true;
        }
    };
    new Expectations() {

        {
            vertx.createNetServer((NetServerOptions) any);
            result = netServer;
            netServer.listen(anyInt, anyString, (Handler) any);
            netSocket.metrics();
            result = tcpServerMetrics;
        }
    };
    URIEndpointObject endpointObject = new URIEndpointObject("highway://127.0.0.1:6663?sslEnabled=true");
    TcpServer server = new TcpServerForTest(endpointObject) {

        @Override
        protected int getConnectionLimit() {
            return 2;
        }
    };
    // assert done in Expectations
    server.init(vertx, "", callback);
    // no problem
    endpointMetric.onConnect();
    endpointMetric.onConnect();
    connectHandler.handle(netSocket);
    // reject
    endpointMetric.onConnect();
    connectHandler.handle(netSocket);
    Assert.assertTrue(netSocketClosed);
    Assert.assertEquals(1, endpointMetric.getRejectByConnectionLimitCount());
}