io.tracee.binding.springws.TraceeEndpointInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tracee-springws Show documentation

Please refer to https://github.com/tracee/tracee.

The newest version!

package io.tracee.binding.springws;

import io.tracee.Tracee;
import io.tracee.TraceeBackend;
import io.tracee.Utilities;
import io.tracee.configuration.TraceeFilterConfiguration;
import io.tracee.configuration.TraceeFilterConfiguration.Profile;
import org.springframework.ws.context.MessageContext;
import org.springframework.ws.server.EndpointInterceptor;

import static io.tracee.configuration.TraceeFilterConfiguration.Channel.IncomingRequest;
import static io.tracee.configuration.TraceeFilterConfiguration.Channel.OutgoingResponse;

public final class TraceeEndpointInterceptor extends AbstractTraceeInterceptor implements EndpointInterceptor {

	public TraceeEndpointInterceptor() {
		this(Tracee.getBackend(), Profile.DEFAULT);
	}

	public TraceeEndpointInterceptor(final String profile) {
		this(Tracee.getBackend(), profile);
	}

	TraceeEndpointInterceptor(final TraceeBackend backend, final String profile) {
		super(backend, profile);
	}

	@Override
	public boolean handleRequest(MessageContext messageContext, Object o) {
		parseContextFromSoapHeader(messageContext.getRequest(), IncomingRequest);

		Utilities.generateInvocationIdIfNecessary(backend);
		return true;
	}

	@Override
	public boolean handleResponse(MessageContext messageContext, Object o) {
		serializeContextToSoapHeader(messageContext.getResponse(), OutgoingResponse);
		return true;
	}

	@Override
	public boolean handleFault(MessageContext messageContext, Object o) {
		return handleResponse(messageContext, o);
	}

	@Override
	public void afterCompletion(MessageContext messageContext, Object o, Exception e) {
		backend.clear();
	}
}