controllers.ProxyController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metrics-portal_2.11 Show documentation
metrics-portal
There is a newer version: 0.9.2
/**
 * Copyright 2015 Groupon.com
 *
 * 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 controllers;

import actors.ProxyConnection;
import akka.actor.ActorRef;
import akka.actor.ActorSystem;
import com.arpnetworking.metrics.MetricsFactory;
import com.fasterxml.jackson.databind.JsonNode;
import com.google.common.collect.Maps;
import com.google.inject.Inject;
import models.messages.ProxyConnectDestination;
import models.messages.ProxyConnectOriginator;
import play.mvc.Controller;
import play.mvc.WebSocket;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.Map;

/**
 * Metrics portal proxy controller. Exposes API to proxy streaming to application hosts.
 *
 * @author Ville Koskela (vkoskela at groupon dot com)
 */
public class ProxyController extends Controller {

    /**
     * Public constructor.
     *
     * @param metricsFactory The MetricsFactory instance.
     * @param system The ActorSystem instance.
     */
    @Inject
    public ProxyController(final MetricsFactory metricsFactory, final ActorSystem system) {
        _metricsFactory = metricsFactory;
        _system = system;
    }

    /**
     * Proxy the specified stream.
     *
     * @param uri The uri of the stream to proxy.
     * @return Proxied stream.
     * @throws URISyntaxException if supplied uri is invalid.
     */
    public WebSocket stream(final String uri) throws URISyntaxException {
        final ActorRef proxyActor = _system.actorOf(ProxyConnection.props(_metricsFactory));

        // Initiate web socket connection to proxy destination
        proxyActor.tell(new ProxyConnectDestination(new URI(uri)), ActorRef.noSender());

        // Accept web socket connection from proxy originator
        return new WebSocket() {
            // Called when the Websocket Handshake is done.
            @Override
            public void onReady(final WebSocket.In in, final WebSocket.Out out) {
                proxyActor.tell(new ProxyConnectOriginator(in, out), ActorRef.noSender());
            }
        };
    }

    private final MetricsFactory _metricsFactory;
    private final ActorSystem _system;
    private final Map, ActorRef> _connections = Maps.newHashMap();
}