com.palantir.atlasdb.timelock.lock.watch.ConjureLockWatchingServiceEndpoints Maven / Gradle / Ivy

Go to download
package com.palantir.atlasdb.timelock.lock.watch;

import com.google.common.collect.ImmutableList;
import com.google.common.util.concurrent.ListenableFuture;
import com.palantir.atlasdb.timelock.api.LockWatchRequest;
import com.palantir.conjure.java.undertow.lib.Deserializer;
import com.palantir.conjure.java.undertow.lib.Endpoint;
import com.palantir.conjure.java.undertow.lib.RequestContext;
import com.palantir.conjure.java.undertow.lib.ReturnValueWriter;
import com.palantir.conjure.java.undertow.lib.TypeMarker;
import com.palantir.conjure.java.undertow.lib.UndertowRuntime;
import com.palantir.conjure.java.undertow.lib.UndertowService;
import com.palantir.logsafe.SafeArg;
import com.palantir.tokens.auth.AuthHeader;
import io.undertow.server.HttpHandler;
import io.undertow.server.HttpServerExchange;
import io.undertow.util.HttpString;
import io.undertow.util.Methods;
import io.undertow.util.PathTemplateMatch;
import io.undertow.util.StatusCodes;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import javax.annotation.processing.Generated;

@Generated("com.palantir.conjure.java.services.UndertowServiceHandlerGenerator")
public final class ConjureLockWatchingServiceEndpoints implements UndertowService {
    private final UndertowConjureLockWatchingService delegate;

    private ConjureLockWatchingServiceEndpoints(UndertowConjureLockWatchingService delegate) {
        this.delegate = delegate;
    }

    public static UndertowService of(UndertowConjureLockWatchingService delegate) {
        return new ConjureLockWatchingServiceEndpoints(delegate);
    }

    @Override
    public List endpoints(UndertowRuntime runtime) {
        return ImmutableList.of(new StartWatchingEndpoint(runtime, delegate));
    }

    private static final class StartWatchingEndpoint implements HttpHandler, Endpoint, ReturnValueWriter {
        private final UndertowRuntime runtime;

        private final UndertowConjureLockWatchingService delegate;

        private final Deserializer deserializer;

        StartWatchingEndpoint(UndertowRuntime runtime, UndertowConjureLockWatchingService delegate) {
            this.runtime = runtime;
            this.delegate = delegate;
            this.deserializer = runtime.bodySerDe().deserializer(new TypeMarker() {}, this);
        }

        @Override
        public void handleRequest(HttpServerExchange exchange) throws IOException {
            AuthHeader authHeader = runtime.auth().header(exchange);
            RequestContext requestContext = runtime.contexts().createContext(exchange, this);
            LockWatchRequest request = deserializer.deserialize(exchange);
            Map pathParams =
                    exchange.getAttachment(PathTemplateMatch.ATTACHMENT_KEY).getParameters();
            String namespace = runtime.plainSerDe().deserializeString(pathParams.get("namespace"));
            requestContext.requestArg(SafeArg.of("namespace", namespace));
            ListenableFuture result = delegate.startWatching(authHeader, namespace, request);
            runtime.async().register(result, this, exchange);
        }

        @Override
        public void write(Void result, HttpServerExchange exchange) throws IOException {
            exchange.setStatusCode(StatusCodes.NO_CONTENT);
        }

        @Override
        public HttpString method() {
            return Methods.POST;
        }

        @Override
        public String template() {
            return "/lw/sw/{namespace}";
        }

        @Override
        public String serviceName() {
            return "ConjureLockWatchingService";
        }

        @Override
        public String name() {
            return "startWatching";
        }

        @Override
        public HttpHandler handler() {
            return this;
        }
    }
}