io.trino.gateway.ha.resource.GatewayResource Maven / Gradle / Ivy

Go to download
/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.trino.gateway.ha.resource;

import com.google.inject.Inject;
import io.airlift.log.Logger;
import io.trino.gateway.ha.router.GatewayBackendManager;
import jakarta.annotation.security.RolesAllowed;
import jakarta.ws.rs.GET;
import jakarta.ws.rs.POST;
import jakarta.ws.rs.Path;
import jakarta.ws.rs.PathParam;
import jakarta.ws.rs.Produces;
import jakarta.ws.rs.core.Context;
import jakarta.ws.rs.core.MediaType;
import jakarta.ws.rs.core.Request;
import jakarta.ws.rs.core.Response;

import static java.util.Objects.requireNonNull;

@RolesAllowed("API")
@Path("/gateway")
@Produces(MediaType.APPLICATION_JSON)
public class GatewayResource
{
    private static final Logger log = Logger.get(GatewayResource.class);

    private final GatewayBackendManager gatewayBackendManager;

    @Inject
    public GatewayResource(GatewayBackendManager gatewayBackendManager)
    {
        this.gatewayBackendManager = requireNonNull(gatewayBackendManager, "gatewayBackendManager is null");
    }

    @GET
    public Response ok(@Context Request request)
    {
        return Response.ok("ok").build();
    }

    @GET
    @Path("/backend/all")
    public Response getAllBackends()
    {
        return Response.ok(this.gatewayBackendManager.getAllBackends()).build();
    }

    @GET
    @Path("/backend/active")
    public Response getActiveBackends()
    {
        return Response.ok(gatewayBackendManager.getAllActiveBackends()).build();
    }

    @POST
    @Path("/backend/deactivate/{name}")
    public Response deactivateBackend(@PathParam("name") String name)
    {
        try {
            this.gatewayBackendManager.deactivateBackend(name);
        }
        catch (Exception e) {
            log.error(e, e.getMessage());
            return throwError(e);
        }
        return Response.ok().build();
    }

    @POST
    @Path("/backend/activate/{name}")
    public Response activateBackend(@PathParam("name") String name)
    {
        try {
            this.gatewayBackendManager.activateBackend(name);
        }
        catch (Exception e) {
            log.error(e);
            return throwError(e);
        }
        return Response.ok().build();
    }

    private Response throwError(Exception e)
    {
        return Response.status(Response.Status.NOT_FOUND)
                .entity(e.getMessage())
                .type("text/plain")
                .build();
    }
}