io.rtr.alchemy.service.resources.ActiveTreatmentsResource 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

There is a newer version: 2.2.17

package io.rtr.alchemy.service.resources;

import com.codahale.metrics.annotation.Timed;
import com.google.common.collect.Maps;
import com.google.inject.Inject;
import io.rtr.alchemy.dto.identities.IdentityDto;
import io.rtr.alchemy.dto.models.TreatmentDto;
import io.rtr.alchemy.identities.Identity;
import io.rtr.alchemy.mapping.Mappers;
import io.rtr.alchemy.models.Experiment;
import io.rtr.alchemy.models.Experiments;
import io.rtr.alchemy.models.Treatment;

import javax.validation.Valid;
import javax.ws.rs.Consumes;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;
import java.util.Map;

/**
 * Resource for querying active experiments
 */
@Path("/active")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public class ActiveTreatmentsResource extends BaseResource {
    private final Experiments experiments;
    private final Mappers mapper;

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

    @POST
    @Timed
    @Path("/experiments/{experimentName}/treatment")
    public TreatmentDto getActiveTreatment(@PathParam("experimentName") String experimentName,
                                           @Valid IdentityDto identity) {
        return mapper.toDto(
            experiments.getActiveTreatment(
                experimentName,
                mapper.fromDto(identity, Identity.class)
            ),
            TreatmentDto.class
        );
    }

    @POST
    @Timed
    @Path("/treatments")
    public Map getActiveTreatments(@Valid IdentityDto identityDto) {
        final Map treatments = Maps.newHashMap();
        final Map activeTreatments = experiments.getActiveTreatments(
            mapper.fromDto(identityDto, Identity.class)
        );

        for (final Map.Entry entry : activeTreatments.entrySet()) {
            treatments.put(entry.getKey().getName(), mapper.toDto(entry.getValue(), TreatmentDto.class));
        }

        return treatments;
    }
}