org.zalando.riptide.idempotency.MethodOverrideIdempotencyDetector Maven / Gradle / Ivy

Go to download
package org.zalando.riptide.idempotency;

import lombok.extern.slf4j.Slf4j;
import org.apiguardian.api.API;
import org.springframework.http.HttpMethod;
import org.zalando.riptide.RequestArguments;

import javax.annotation.Nullable;
import java.util.List;
import java.util.Map;
import java.util.function.Predicate;

import static java.util.Collections.emptyList;
import static org.apiguardian.api.API.Status.EXPERIMENTAL;
import static org.springframework.http.HttpMethod.POST;

/**
 * @see OpenSocial Core API Server Specification 2.5.1, Section 2.1.1.1
 */
@API(status = EXPERIMENTAL)
@Slf4j
public final class MethodOverrideIdempotencyDetector implements IdempotencyDetector {

    @Override
    public boolean test(final RequestArguments arguments, final Predicate root) {
        if (arguments.getMethod() != POST) {
            return false;
        }

        @Nullable final HttpMethod method = getOverride(arguments);

        if (method == null) {
            return false;
        }

        return root.test(arguments.withMethod(method));
    }

    @Nullable
    private HttpMethod getOverride(final RequestArguments arguments) {
        final Map> headers = arguments.getHeaders();
        final String name = "X-HTTP-Method-Override";
        final List overrides = headers.getOrDefault(name, emptyList());

        @Nullable final String override = overrides.stream().findFirst().orElse(null);

        if (override == null) {
            return null;
        }

        try {
            return HttpMethod.valueOf(override);
        } catch (final IllegalArgumentException e) {
            log.warn("Received invalid method in {} header: \"{}\"", name, override);
            return null;
        }
    }

}