Skip to content

Commit

Permalink
Compactede ws/hjttp gateways
Browse files Browse the repository at this point in the history
  • Loading branch information
artem-v committed Sep 29, 2024
1 parent 83c783b commit 5133b31
Show file tree
Hide file tree
Showing 25 changed files with 71 additions and 90 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
import java.net.InetSocketAddress;
import java.util.StringJoiner;
import java.util.function.Consumer;
import java.util.function.UnaryOperator;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.netty.DisposableServer;
Expand All @@ -35,10 +34,6 @@ private HttpGateway(Builder builder) {
this.corsConfigBuilder = builder.corsConfigBuilder;
}

public HttpGateway(UnaryOperator<Builder> operator) {
this(operator.apply(new Builder()));
}

@Override
public String id() {
return options.id();
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.exceptions;
package io.scalecube.services.gateway;

import io.scalecube.services.annotations.Service;
import io.scalecube.services.annotations.ServiceMethod;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.exceptions;
package io.scalecube.services.gateway;

import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.exceptions;
package io.scalecube.services.gateway;

import io.scalecube.services.api.ErrorData;
import io.scalecube.services.api.ServiceMessage;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.exceptions;
package io.scalecube.services.gateway;

import io.scalecube.services.exceptions.ServiceException;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -60,15 +60,13 @@ void testCrossOriginRequest() {
gatewayBuilder
.gateway(
opts ->
new HttpGateway(
builder ->
builder
.options(opts.id("http").port(HTTP_PORT))
.corsEnabled(true)
.corsConfigBuilder(
corsConfigBuilder ->
corsConfigBuilder.allowedRequestHeaders(
"Content-Type", "X-Correlation-ID"))))
new HttpGateway.Builder()
.options(opts.id("http").port(HTTP_PORT))
.corsEnabled(true)
.corsConfigBuilder(
builder ->
builder.allowedRequestHeaders("Content-Type", "X-Correlation-ID"))
.build())
.start()
.block(TIMEOUT);

Expand Down Expand Up @@ -118,8 +116,7 @@ void testOptionRequestCorsDisabled() {
gateway =
gatewayBuilder
.gateway(
opts ->
new HttpGateway(builder -> builder.options(opts.id("http").port(HTTP_PORT))))
opts -> new HttpGateway.Builder().options(opts.id("http").port(HTTP_PORT)).build())
.start()
.block(TIMEOUT);

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ void beforEach() {
.transport(cfg -> cfg.transportFactory(new WebsocketTransportFactory()))
.options(opts -> opts.metadata(serviceEndpoint)))
.transport(RSocketServiceTransport::new)
.gateway(options -> new HttpGateway(builder -> builder.options(options.id("HTTP"))))
.gateway(options -> new HttpGateway.Builder().options(options.id("HTTP")).build())
.startAwait();

gatewayAddress = gateway.gateway("HTTP").address();
Expand Down
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
package io.scalecube.services.gateway.http;

import static io.scalecube.services.gateway.GatewayErrorMapperImpl.ERROR_MAPPER;
import static io.scalecube.services.gateway.TestUtils.TIMEOUT;
import static io.scalecube.services.gateway.exceptions.GatewayErrorMapperImpl.ERROR_MAPPER;

import io.scalecube.services.ServiceInfo;
import io.scalecube.services.gateway.BaseTest;
import io.scalecube.services.gateway.exceptions.ErrorService;
import io.scalecube.services.gateway.exceptions.ErrorServiceImpl;
import io.scalecube.services.gateway.exceptions.SomeException;
import io.scalecube.services.gateway.ErrorService;
import io.scalecube.services.gateway.ErrorServiceImpl;
import io.scalecube.services.gateway.SomeException;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ class HttpGatewayExtension extends AbstractGatewayExtension {
HttpGatewayExtension(ServiceInfo serviceInfo) {
super(
serviceInfo,
opts -> new HttpGateway(builder -> builder.options(opts.id(GATEWAY_ALIAS_NAME))),
opts -> new HttpGateway.Builder().options(opts.id(GATEWAY_ALIAS_NAME)).build(),
GatewayClientTransports::httpGatewayClientTransport);
}
}
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
package io.scalecube.services.gateway.http;

import static io.scalecube.services.gateway.GatewayErrorMapperImpl.ERROR_MAPPER;
import static io.scalecube.services.gateway.TestUtils.TIMEOUT;
import static io.scalecube.services.gateway.exceptions.GatewayErrorMapperImpl.ERROR_MAPPER;

import io.scalecube.services.ServiceInfo;
import io.scalecube.services.gateway.BaseTest;
import io.scalecube.services.gateway.exceptions.ErrorService;
import io.scalecube.services.gateway.exceptions.ErrorServiceImpl;
import io.scalecube.services.gateway.exceptions.SomeException;
import io.scalecube.services.gateway.ErrorService;
import io.scalecube.services.gateway.ErrorServiceImpl;
import io.scalecube.services.gateway.SomeException;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
Expand All @@ -22,11 +22,10 @@ class HttpLocalGatewayErrorMapperTest extends BaseTest {
new HttpLocalGatewayExtension(
ServiceInfo.fromServiceInstance(new ErrorServiceImpl()).errorMapper(ERROR_MAPPER).build(),
opts ->
new HttpGateway(
builder ->
builder
.options(opts.call(opts.call().errorMapper(ERROR_MAPPER)))
.errorMapper(ERROR_MAPPER)));
new HttpGateway.Builder()
.options(opts.call(opts.call().errorMapper(ERROR_MAPPER)))
.errorMapper(ERROR_MAPPER)
.build());

private ErrorService service;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ class HttpLocalGatewayExtension extends AbstractLocalGatewayExtension {
}

HttpLocalGatewayExtension(ServiceInfo serviceInfo) {
this(serviceInfo, opts -> new HttpGateway(builder -> builder.options(opts)));
this(serviceInfo, opts -> new HttpGateway.Builder().options(opts).build());
}

HttpLocalGatewayExtension(
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import java.util.concurrent.atomic.AtomicLongFieldUpdater;
import java.util.concurrent.atomic.AtomicReferenceFieldUpdater;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import reactor.core.Disposable;

Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import static io.scalecube.services.gateway.TestUtils.TIMEOUT;
import static org.junit.jupiter.api.Assertions.assertEquals;
Expand All @@ -24,7 +24,6 @@
import io.scalecube.services.gateway.client.StaticAddressRouter;
import io.scalecube.services.gateway.client.websocket.WebsocketGatewayClient;
import io.scalecube.services.gateway.client.websocket.WebsocketGatewayClientSession;
import io.scalecube.services.gateway.ws.WebsocketGateway;
import io.scalecube.services.transport.rsocket.RSocketServiceTransport;
import io.scalecube.transport.netty.websocket.WebsocketTransportFactory;
import java.io.IOException;
Expand Down Expand Up @@ -70,9 +69,10 @@ void beforEach() {
.transport(RSocketServiceTransport::new)
.gateway(
options ->
new WebsocketGateway(
builder ->
builder.options(options.id("WS")).gatewayHandler(sessionEventHandler)))
new WebsocketGateway.Builder()
.options(options.id("WS"))
.gatewayHandler(sessionEventHandler)
.build())
.startAwait();

gatewayAddress = gateway.gateway("WS").address();
Expand Down
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import static io.scalecube.services.gateway.GatewayErrorMapperImpl.ERROR_MAPPER;
import static io.scalecube.services.gateway.TestUtils.TIMEOUT;
import static io.scalecube.services.gateway.exceptions.GatewayErrorMapperImpl.ERROR_MAPPER;

import io.scalecube.services.ServiceInfo;
import io.scalecube.services.gateway.BaseTest;
import io.scalecube.services.gateway.exceptions.ErrorService;
import io.scalecube.services.gateway.exceptions.ErrorServiceImpl;
import io.scalecube.services.gateway.exceptions.SomeException;
import io.scalecube.services.gateway.ErrorService;
import io.scalecube.services.gateway.ErrorServiceImpl;
import io.scalecube.services.gateway.SomeException;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import io.netty.buffer.ByteBuf;
import io.scalecube.services.Address;
Expand All @@ -16,7 +16,6 @@
import io.scalecube.services.gateway.client.GatewayClientTransports;
import io.scalecube.services.gateway.client.StaticAddressRouter;
import io.scalecube.services.gateway.client.websocket.WebsocketGatewayClient;
import io.scalecube.services.gateway.ws.WebsocketGateway;
import io.scalecube.services.transport.rsocket.RSocketServiceTransport;
import io.scalecube.transport.netty.websocket.WebsocketTransportFactory;
import java.time.Duration;
Expand Down Expand Up @@ -56,11 +55,10 @@ static void beforeAll() {
.transport(RSocketServiceTransport::new)
.gateway(
options ->
new WebsocketGateway(
builder ->
builder
.options(options.id("WS"))
.gatewayHandler(new TestGatewaySessionHandler())))
new WebsocketGateway.Builder()
.options(options.id("WS"))
.gatewayHandler(new TestGatewaySessionHandler())
.build())
.startAwait();
gatewayAddress = gateway.gateway("WS").address();

Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import io.scalecube.services.ServiceInfo;
import io.scalecube.services.gateway.AbstractGatewayExtension;
import io.scalecube.services.gateway.client.GatewayClientTransports;
import io.scalecube.services.gateway.ws.WebsocketGateway;

class WebsocketGatewayExtension extends AbstractGatewayExtension {

Expand All @@ -16,7 +15,7 @@ class WebsocketGatewayExtension extends AbstractGatewayExtension {
WebsocketGatewayExtension(ServiceInfo serviceInfo) {
super(
serviceInfo,
opts -> new WebsocketGateway(builder -> builder.options(opts.id(GATEWAY_ALIAS_NAME))),
opts -> new WebsocketGateway.Builder().options(opts.id(GATEWAY_ALIAS_NAME)).build(),
GatewayClientTransports::websocketGatewayClientTransport);
}
}
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import static org.junit.jupiter.api.Assertions.assertEquals;

Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
Expand Down
Original file line number Diff line number Diff line change
@@ -1,14 +1,13 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import static io.scalecube.services.gateway.GatewayErrorMapperImpl.ERROR_MAPPER;
import static io.scalecube.services.gateway.TestUtils.TIMEOUT;
import static io.scalecube.services.gateway.exceptions.GatewayErrorMapperImpl.ERROR_MAPPER;

import io.scalecube.services.ServiceInfo;
import io.scalecube.services.gateway.BaseTest;
import io.scalecube.services.gateway.exceptions.ErrorService;
import io.scalecube.services.gateway.exceptions.ErrorServiceImpl;
import io.scalecube.services.gateway.exceptions.SomeException;
import io.scalecube.services.gateway.ws.WebsocketGateway;
import io.scalecube.services.gateway.ErrorService;
import io.scalecube.services.gateway.ErrorServiceImpl;
import io.scalecube.services.gateway.SomeException;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.RegisterExtension;
Expand All @@ -21,11 +20,10 @@ class WebsocketLocalGatewayErrorMapperTest extends BaseTest {
new WebsocketLocalGatewayExtension(
ServiceInfo.fromServiceInstance(new ErrorServiceImpl()).errorMapper(ERROR_MAPPER).build(),
opts ->
new WebsocketGateway(
builder ->
builder
.options(opts.call(opts.call().errorMapper(ERROR_MAPPER)))
.errorMapper(ERROR_MAPPER)));
new WebsocketGateway.Builder()
.options(opts.call(opts.call().errorMapper(ERROR_MAPPER)))
.errorMapper(ERROR_MAPPER)
.build());

private ErrorService service;

Expand Down
Original file line number Diff line number Diff line change
@@ -1,10 +1,9 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import io.scalecube.services.ServiceInfo;
import io.scalecube.services.gateway.AbstractLocalGatewayExtension;
import io.scalecube.services.gateway.GatewayOptions;
import io.scalecube.services.gateway.client.GatewayClientTransports;
import io.scalecube.services.gateway.ws.WebsocketGateway;
import java.util.function.Function;

class WebsocketLocalGatewayExtension extends AbstractLocalGatewayExtension {
Expand All @@ -16,7 +15,7 @@ class WebsocketLocalGatewayExtension extends AbstractLocalGatewayExtension {
}

WebsocketLocalGatewayExtension(ServiceInfo serviceInfo) {
this(serviceInfo, opts -> new WebsocketGateway(builder -> builder.options(opts)));
this(serviceInfo, opts -> new WebsocketGateway.Builder().options(opts).build());
}

WebsocketLocalGatewayExtension(
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import static org.junit.jupiter.api.Assertions.assertEquals;

Expand Down
Original file line number Diff line number Diff line change
@@ -1,11 +1,10 @@
package io.scalecube.services.gateway.websocket;
package io.scalecube.services.gateway.ws;

import io.scalecube.services.ServiceInfo;
import io.scalecube.services.gateway.AbstractLocalGatewayExtension;
import io.scalecube.services.gateway.AuthRegistry;
import io.scalecube.services.gateway.GatewaySessionHandlerImpl;
import io.scalecube.services.gateway.client.GatewayClientTransports;
import io.scalecube.services.gateway.ws.WebsocketGateway;

public class WebsocketLocalWithAuthExtension extends AbstractLocalGatewayExtension {

Expand All @@ -19,11 +18,10 @@ public class WebsocketLocalWithAuthExtension extends AbstractLocalGatewayExtensi
super(
serviceInfo,
opts ->
new WebsocketGateway(
builder ->
builder
.options(opts.id(GATEWAY_ALIAS_NAME))
.gatewayHandler(new GatewaySessionHandlerImpl(authReg))),
new WebsocketGateway.Builder()
.options(opts.id(GATEWAY_ALIAS_NAME))
.gatewayHandler(new GatewaySessionHandlerImpl(authReg))
.build(),
GatewayClientTransports::websocketGatewayClientTransport);
}
}
Loading

0 comments on commit 5133b31

Please sign in to comment.