com.ning.metrics.collector.jaxrs.BodyResource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metrics.collector Show documentation
Event Collector core
There is a newer version: 2.0.16
/*
 * Copyright 2010-2012 Ning, Inc.
 *
 * Ning licenses this file to you 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 com.ning.metrics.collector.jaxrs;

import com.ning.metrics.collector.endpoint.ParsedRequest;
import com.ning.metrics.collector.endpoint.extractors.DeserializationType;
import com.ning.metrics.serialization.event.Granularity;

import com.google.inject.Inject;
import com.yammer.metrics.annotation.Timed;
import org.joda.time.DateTime;

import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.Consumes;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.io.ByteArrayInputStream;
import java.io.InputStream;

/**
 * Internal event API
 */
@Path("/rest/1.0/event")
public class BodyResource
{
    private static final String APPLICATION_JSON_SMILE = "application/json+smile";
    private static final String THRIFT = "application/thrift";

    private final EventDeserializerRequestHandler requestHandler;

    @Inject
    public BodyResource(final EventDeserializerRequestHandler requestHandler)
    {
        this.requestHandler = requestHandler;
    }

    // legacy endpoint to fall back on, for receiving single ThriftEnvelopeEvents
    @POST
    @Produces(MediaType.TEXT_PLAIN)
    @Timed(name = "POST_ThriftLegacy_API")
    public Response postThriftLegacy(@QueryParam("name") final String eventName,
                                     @QueryParam("date") final String eventDateTimeString,
                                     @QueryParam(Granularity.GRANULARITY_QUERY_PARAM) final String eventGranularity,
                                     final byte[] content,
                                     @Context final HttpHeaders httpHeaders,
                                     @Context final HttpServletRequest request)
    {
        final DateTime eventDateTime = new DateTime(eventDateTimeString);
        final ParsedRequest parsedRequest = new ParsedRequest(eventName, httpHeaders, new ByteArrayInputStream(content),
                                                              eventDateTime, eventGranularity, request.getRemoteAddr(),
                                                              DeserializationType.DEFAULT);
        return requestHandler.handleEventRequest(parsedRequest);
    }

    @POST
    @Consumes(THRIFT)
    @Produces(MediaType.TEXT_PLAIN)
    @Timed(name = "POST_Thrift_API")
    public Response postThrift(final InputStream body,
                               @Context final HttpHeaders httpHeaders,
                               @Context final HttpServletRequest request)
    {
        final ParsedRequest parsedRequest = new ParsedRequest(null, httpHeaders, body, null, null,
                                                              request.getRemoteAddr(), DeserializationType.THRIFT);
        return requestHandler.handleEventRequest(parsedRequest);
    }

    @POST
    @Consumes(MediaType.APPLICATION_JSON)
    @Produces(MediaType.TEXT_PLAIN)
    @Timed(name = "POST_Json_API")
    public Response postJson(final InputStream body,
                             @Context final HttpHeaders httpHeaders,
                             @Context final HttpServletRequest request)
    {
        final ParsedRequest parsedRequest = new ParsedRequest(null, httpHeaders, body, null, null,
                                                              request.getRemoteAddr(), DeserializationType.JSON);
        return requestHandler.handleEventRequest(parsedRequest);
    }

    @POST
    @Consumes(APPLICATION_JSON_SMILE)
    @Produces(MediaType.TEXT_PLAIN)
    @Timed(name = "POST_Smile_API")
    public Response postSmile(final InputStream body,
                              @Context final HttpHeaders httpHeaders,
                              @Context final HttpServletRequest request)
    {
        final ParsedRequest parsedRequest = new ParsedRequest(null, httpHeaders, body, null, null,
                                                              request.getRemoteAddr(), DeserializationType.SMILE);
        return requestHandler.handleEventRequest(parsedRequest);
    }
}