io.rtr.alchemy.service.resources.TreatmentOverridesResource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of alchemy-service Show documentation
REST service for hosting Alchemy as a service
The newest version!
package io.rtr.alchemy.service.resources;

import com.google.inject.Inject;

import io.rtr.alchemy.dto.models.TreatmentOverrideDto;
import io.rtr.alchemy.dto.requests.TreatmentOverrideRequest;
import io.rtr.alchemy.mapping.Mappers;
import io.rtr.alchemy.models.Experiment;
import io.rtr.alchemy.models.Experiments;

import javax.validation.Valid;
import javax.ws.rs.Consumes;
import javax.ws.rs.DELETE;
import javax.ws.rs.GET;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;

/** Resource for interacting with treatment overrides */
@Path("/experiments/{experimentName}/overrides")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public class TreatmentOverridesResource extends BaseResource {
    private final Experiments experiments;
    private final Mappers mapper;

    @Inject
    public TreatmentOverridesResource(Experiments experiments, Mappers mapper) {
        this.experiments = experiments;
        this.mapper = mapper;
    }

    @GET
    @Path("/{overrideName}")
    public TreatmentOverrideDto getOverride(
            @PathParam("experimentName") String experimentName,
            @PathParam("overrideName") String overrideName) {
        return mapper.toDto(
                ensureExists(
                        ensureExists(experiments.get(experimentName)).getOverride(overrideName)),
                TreatmentOverrideDto.class);
    }

    @GET
    public Iterable getOverrides(
            @PathParam("experimentName") String experimentName) {
        return mapper.toDto(
                ensureExists(experiments.get(experimentName)).getOverrides(),
                TreatmentOverrideDto.class);
    }

    @PUT
    public Response addOverride(
            @PathParam("experimentName") String experimentName,
            @Valid TreatmentOverrideRequest request) {
        ensureExists(experiments.get(experimentName))
                .addOverride(request.getName(), request.getTreatment(), request.getFilter())
                .save();

        return created();
    }

    @DELETE
    public void clearOverrides(@PathParam("experimentName") String experimentName) {
        ensureExists(experiments.get(experimentName)).clearOverrides().save();
    }

    @DELETE
    @Path("/{overrideName}")
    public void removeOverride(
            @PathParam("experimentName") String experimentName,
            @PathParam("overrideName") String overrideName) {
        final Experiment experiment = ensureExists(experiments.get(experimentName));
        ensureExists(experiment.getOverride(overrideName));
        experiment.removeOverride(overrideName).save();
    }
}