io.castled.resources.QueryModelResource Maven / Gradle / Ivy

Go to download
package io.castled.resources;

import com.google.inject.Inject;
import com.google.inject.Singleton;
import io.castled.dtos.EntityCreateResponse;
import io.castled.dtos.querymodel.QueryModelDTO;
import io.castled.dtos.querymodel.ModelInputDTO;
import io.castled.models.users.User;
import io.castled.resources.validators.ResourceAccessController;
import io.castled.services.QueryModelService;
import io.dropwizard.auth.Auth;
import lombok.extern.slf4j.Slf4j;

import javax.validation.Valid;
import javax.ws.rs.*;
import javax.ws.rs.core.MediaType;
import java.util.List;

@Path("/v1/models")
@Produces(MediaType.APPLICATION_JSON)
@Singleton
@Slf4j
public class QueryModelResource {

    private final QueryModelService queryModelService;

    @Inject
    public QueryModelResource(QueryModelService queryModelService) {
        this.queryModelService = queryModelService;
    }

    @POST
    public EntityCreateResponse createModel(@Valid ModelInputDTO modelInputDTO, @Auth User user) {
        return new EntityCreateResponse(this.queryModelService.createModel(modelInputDTO, user));
    }

    @GET
    @Path("/{id}")
    public QueryModelDTO getModel(@PathParam("id") Long id, @Auth User user) {
        return this.queryModelService.getQueryModel(id, user.getTeamId());
    }

    @DELETE
    @Path("/{id}")
    public void deleteModel(@PathParam("id") Long id, @Auth User user) {
        this.queryModelService.deleteModel(id, user.getTeamId());
    }

    @GET
    public List getAllModels(@QueryParam("warehouseId") Long warehouseId, @Auth User user) {
        return this.queryModelService.getAllModels(warehouseId, user.getTeamId());
    }
}