使用Spring Web服务为每个请求提供不同的ClientInterceptor

Question

我已经通过扩展WebServiceGatewaySupport创建了一个定制的web服务客户端，还实现了定制的ClientInterceptor来记录一些请求/响应数据。我必须为每个调用创建新的拦截器，因为它必须存储有关请求的一些数据。

当我向我的客户发出两个或更多呼叫时，就会出现这个问题。第一个请求将自己的拦截器与其clientId一起应用。第二个应该做同样的事情。但是由于两个请求在我的客户端中使用相同的WebServicetemplate，所以第二个请求用它自己的clientId替换了拦截器。

因此，我应该将以下输出输出到控制台：

Request: clientId-1
Request: clientId-2
Response: clientId-1
Response: clientId-2

但我得到了这个：

Request: clientId-1
Request: clientId-2
Response: clientId-2
Response: clientId-2

下面是代码示例(只是为了理解它是如何工作的)：

@Data
class Response {
    private final String result;

    public Response(String result) {
        this.result = result;
    }
}

@Data
class Request {
    private final String firstName;
    private final String lastName;
}

@Data
class Context {
    private final String clientId;
}

@Data
class Client {
    private final String clientId;
    private final String firstName;
    private final String lastName;
}

class CustomInterceptor extends ClientInterceptorAdapter {
    private final String clientId;

    public CustomInterceptor(String clientId) {
        this.clientId = clientId;
    }

    @Override
    public boolean handleRequest(MessageContext messageContext) throws WebServiceClientException {
        System.out.println("Request: " + clientId);
        return true;
    }

    @Override
    public boolean handleResponse(MessageContext messageContext) throws WebServiceClientException {
        System.out.println("Response: " + clientId);
        return true;
    }

    @Override
    public boolean handleFault(MessageContext messageContext) throws WebServiceClientException {
        System.out.println("Error: " + clientId);
        return true;
    }
}

@Component
class CustomClient extends WebServiceGatewaySupport {

    public Response sendRequest(Request request, Context context) {
        CustomInterceptor[] interceptors = {new CustomInterceptor(context.getClientId())};
        setInterceptors(interceptors);
        return (Response) getWebServiceTemplate().marshalSendAndReceive(request);
    }
}

@Service
@RequiredArgsConstructor
class CustomService {

    private final CustomClient customClient;

    public String call(Request request, Context context) {
        Response response = customClient.sendRequest(request, context);
        return response.getResult();
    }
}

@RestController
@RequestMapping("/test")
@RequiredArgsConstructor
class CustomController {

    private final CustomService service;

    public CustomController(CustomService service) {
        this.service = service;
    }

    @PostMapping
    public String test(@RequestBody Client client) {
        Request request = new Request(client.getFirstName(), client.getLastName());
        Context context = new Context(client.getClientId());
        return service.call(request, context);
    }
}

是否可以为每个调用实现具有某种状态的自定义拦截器？优选地，在WebServicetemplate上没有任何锁定，以避免性能下降。

Answer 1

好吧。我已经为我的案例找到了解决方案。我已经创建了一个WebServiceMessageCallback的实现，并使用它将每个请求的数据保存在WebServiceMessage的mime头中，而不是拦截器中。

@Data
class CustomMessageCallback implements WebServiceMessageCallback {

    private final String clientId;
    
    @Override
    public void doWithMessage(WebServiceMessage message) throws IOException, TransformerException {
        MimeHeaders headers = ((SaajSoapMessage) message).getSaajMessage().getMimeHeaders();
        headers.addHeader("X-Client-Id", clientId);
    }
}

并在我的客户端实现中传递此回调：

@Component
class CustomClient extends WebServiceGatewaySupport {

    public Response sendRequest(Request request, Context context) {
        CustomInterceptor[] interceptors = {new CustomInterceptor()};
        setInterceptors(interceptors);
        return (Response) getWebServiceTemplate()
            .marshalSendAndReceive(request, new CustomMessageCallback(context.getClientId()));
    }
}

因此，现在我可以通过拦截器在处理请求/响应/错误时获得这些数据。

@Override
public boolean handleRequest(MessageContext messageContext) throws WebServiceClientException {
    String clientId = ((SaajSoapMessage) messageContext.getRequest())
        .getSaajMessage()
        .getMimeHeaders()
        .getHeader("X-Client-Id")[0];
    System.out.println("Request: " + clientId);
    return true;
}