0

用例

我正在尝试使用 micronaut 声明式客户端调用服务。该服务实际上是许多相同的服务,但针对我们系统中的每个原则托管在不同的主机上。例如

  • tenetA.example.com/api
  • tenetB.example.com/api

从 micronaut,我想使用请求标头X-tenetID,并基于它调用正确的服务。听起来很简单吧?

第一次尝试:使用过滤器

我尝试的第一件事是在客户端上使用过滤器

@FilterMatcher
@Documented
@Retention(RUNTIME)
@Target({TYPE, PARAMETER})
public @interface MyFilterAnnotation
{
}

@MyFilterAnnotation
@Client("http://replaceme.example.com/")
public interface MyClient
{
    @Post(uri = "some/endpoint")
    AuthResponse auth(@Body CustomCredentials credentials, @RequestAttribute(name = "tenet-id") String tenetID);
}

@MyFilterAnnotation
@Singleton
public class MyFilter implements HttpClientFilter
{
    @Override
    public int getOrder()
    {
        return -10; // Tried playing with the order here to no avail
    }

    @Override
    public Publisher<? extends HttpResponse<?>> doFilter(MutableHttpRequest<?> request, ClientFilterChain chain)
    {
        String tenetID = request.getAttribute("tenet-id", String.class).orElse(null);
        UriBuilder builder = UriBuilder.of(request.getUri());
        builder.host(tenetID + ".example.com");
        request.uri(builder.build());
        return chain.proceed(request);
    }
}

我已经确认我的过滤器被调用,请求 uri 被设置,但请求 uri 没有在链的下游得到尊重。覆盖订单似乎也没有任何效果。它仍在向replaceme.example.com. 我在这里做错了吗?

我发现一些论坛帖子说这种方法可能行不通,因为选择的主机是在流程的早期完成的,以说明 LoadBalancer 机制。这让我进行了第二次尝试:

第二次尝试:使用客户端 LoadBalancer

由于似乎无法在过滤器中更改 url 的主机,我尝试编写自己的DiscoveryClientLoadBalancerFactory

@Replaces(DiscoveryClientLoadBalancerFactory.class)
public class MyLoadBalancer extends DiscoveryClientLoadBalancerFactory
{

    /**
     * @param discoveryClient The discover client
     */
    public MyLoadBalancer(final DiscoveryClient discoveryClient)
    {
        super(discoveryClient);
    }


    @Override
    public LoadBalancer create(final String serviceID)
    {
        return discriminator ->
        {
            // discriminator always seems to be null here.
            return Publishers.just(ServiceInstance.of("myService", discriminator + "example.com",8080));
        };       
    }
}

不过我被困在这里,因为我不知道如何告诉声明式客户端使用我的 tenetID 作为鉴别器。我只见过@Inject注释创建 a DefaultHttpClient,它只会调用loadBalancer.select(getLoadBalancerDiscriminator())getLoadBalancerDiscriminator()总是返回null。有没有办法可以根据请求标头设置鉴别器?

什么不可行:应用程序配置

一些使用 bitbucket 的示例文档表明 url 来自配置。我们的原则来来去去,每项服务都必须使用存储在云库中的自己的凭据进行身份验证。因此,在 application.conf 或类似文件中存储某种映射在这里并不是很有用,因为我们不希望每次添加新原则或轮换密钥时都必须重新启动。

帮助

在这两种方法之间,LoadBalancer 路由似乎更骇人听闻,因为我正在做的是请求路由或 url 重写,而不是负载平衡。这些方法之一应该有效吗?有没有更好的方法来做我想做的事?

4

1 回答 1

0

我的最终解决方案是扩展DefaultHttpClient即使它被标记@Internal并覆盖该resolveRequestURI方法

public interface MyClient
{
    @Post(uri = "some/endpoint")
    AuthResponse auth(@Body CustomCredentials credentials, @Header("tenet-id") String tenetID);
}

@Singleton
public class MyHttpClient extends DefaultHttpClient
{
    // Takes the tenet-id header and looks up the url to send the request to.
    @Override
    protected <I> Publisher<URI> resolveRequestURI(final HttpRequest<I> request, final boolean includeContextPath)
    {
        var headers = (MutableHttpHeaders) request.getHeaders();
        var customerID = headers.get("tenet-id", String.class).orElseThrow(() -> new IllegalArgumentException("Request must have an tenet-id header"));
        var requestURI = request.getUri();
        var resolvedURI = secret.getUrl().resolve(includeContextPath ? prependContextPath(requestURI) : requestURI);
        return Publishers.just(resolvedURI);
    }
}

编辑

我了解到这个解决方案并不理想,因为它提供了一个具体的 bean 候选者并阻止了任何类型的配置,就像你在普通的 http 服务上所做的那样。租户传播之类的东西不再起作用。我不得不解决这个问题。

于 2021-04-02T23:56:13.063 回答