实现一个行为类似于 nginx 的 Undertow 反向代理

Question

出于开发目的，并不是每个人都可以在他们的机器上安装 nginx（就像我们的开发人员在 Windows 环境中一样），但我们希望能够做一个行为类似于 nginx 的反向代理。

这是我们非常具体的案例：

我们在 http://0.0.0.0:8081
我们在 http://0.0.0.0:8082

我们希望从 http://0.0.0.0:8080

提供这两项服务

所以我们想这样映射它：

对 http://0.0.0.0:8080/ 的请求被代理到 http://0.0.0.0:8082
对 http://0.0.0.0:8080/api 的请求被代理到 http://0.0.0.0:8081

这样它的工作方式就像 nginx url 重写反向代理。

我查看了 Undertow 源代码和示例，甚至还有这个具体示例：Reverse Proxy Example，但这是一个负载均衡器示例，我还没有找到任何示例可以满足我的需要。

此外，我知道 Undertow 能够做到这一点，因为我们知道我们可以通过 Undertow 组件配置配置 WildFly 来毫无问题地涵盖这种特定情况，但我们希望自己将其实现为本地开发的轻量级解决方案.

有人知道这样做的例子吗？或者任何有足够信息来实现这个的文档？因为我还阅读了 Undertow 关于反向代理的文档，它根本没有帮助。

谢谢

Answer 1

根据 M. Deinum 的评论建议，我将使用 Zuul Spring Boot 组件而不是尝试使用 Undertow 来执行此操作，因为它更适合此任务。

这里有一个 link 教程：

https://spring.io/guides/gs/routing-and-filtering/

希望这对其他人有帮助，因为这是一个很常见的情况，我不知道 Spring Boot 上的 Zuul。

Answer 2

这应该可以完成工作。

它是 Java8，因此某些部分可能无法在您的设置上运行。

您可以按照您在问题中提到的 the example 类似的方式启动它。

package com.company

import com.google.common.collect.ImmutableMap;
import io.undertow.client.ClientCallback;
import io.undertow.client.ClientConnection;
import io.undertow.client.UndertowClient;
import io.undertow.server.HttpServerExchange;
import io.undertow.server.ServerConnection;
import io.undertow.server.handlers.proxy.ProxyCallback;
import io.undertow.server.handlers.proxy.ProxyClient;
import io.undertow.server.handlers.proxy.ProxyConnection;
import org.xnio.IoUtils;
import org.xnio.OptionMap;

import java.io.IOException;
import java.net.URI;
import java.util.concurrent.TimeUnit;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Start the ReverseProxy with an ImmutableMap of matching endpoints and a default
 * 
 * Example:
 * mapping: ImmutableMap("api" -> "http://some-domain.com")
 * default: "http://default-domain.com"
 * 
 * Request 1: localhost:8080/foo -> http://default-domain.com/foo
 * Request 2: localhost:8080/api/bar -> http://some-domain.com/bar
 */

public class ReverseProxyClient implements ProxyClient {
    private static final ProxyTarget TARGET = new ProxyTarget() {};

    private final UndertowClient client;
    private final ImmutableMap<String, URI> mapping;
    private final URI defaultTarget;

    public ReverseProxyClient(ImmutableMap<String, URI> mapping, URI defaultTarget) {
        this.client = UndertowClient.getInstance();
        this.mapping = mapping;
        this.defaultTarget = defaultTarget;
    }

    @Override
    public ProxyTarget findTarget(HttpServerExchange exchange) {
        return TARGET;
    }

    @Override
    public void getConnection(ProxyTarget target, HttpServerExchange exchange, ProxyCallback<ProxyConnection> callback, long timeout, TimeUnit timeUnit) {
        URI targetUri = defaultTarget;

        Matcher matcher = Pattern.compile("^/(\w+)(/.*)").matcher(exchange.getRequestURI());
        if (matcher.find()) {
            String firstUriSegment = matcher.group(1);
            String remaininguri = matcher.group(2);
            if (mapping.containsKey(firstUriSegment)) {
                // If the first uri segment is in the mapping, update the targetUri
                targetUri = mapping.get(firstUriSegment);
                // Strip the request uri from the part that is used to map upon.
                exchange.setRequestURI(remaininguri);
            }
        }

        client.connect(
            new ConnectNotifier(callback, exchange),
            targetUri,
            exchange.getIoThread(),
            exchange.getConnection().getByteBufferPool(),
            OptionMap.EMPTY);
    }

    private final class ConnectNotifier implements ClientCallback<ClientConnection> {
        private final ProxyCallback<ProxyConnection> callback;
        private final HttpServerExchange exchange;

        private ConnectNotifier(ProxyCallback<ProxyConnection> callback, HttpServerExchange exchange) {
            this.callback = callback;
            this.exchange = exchange;
        }

        @Override
        public void completed(final ClientConnection connection) {
            final ServerConnection serverConnection = exchange.getConnection();
            serverConnection.addCloseListener(serverConnection1 -> IoUtils.safeClose(connection));
            callback.completed(exchange, new ProxyConnection(connection, "/"));
        }

        @Override
        public void failed(IOException e) {
            callback.failed(exchange);
        }
    }
}

实现一个行为类似于 nginx 的 Undertow 反向代理

Implement an Undertow reverse proxy that behaves like nginx

java

spring

reverse-proxy

undertow