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

如何解决使用Spring Web Services为每个请求提供不同的ClientInterceptor

我通过扩展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进行任何锁定，以免导致性能下降。

解决方法

好的。我为我的案件找到了解决方案。 我创建了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;
}