com.github.jguaneri.notifications.web.controller.NotificationController.java Source code

Introduction

Here is the source code for com.github.jguaneri.notifications.web.controller.NotificationController.java

Source

/**
 * The MIT License (MIT)
 *
 * Copyright (c) 2014 Joseph Guaneri
 * 
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 * 
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
package com.github.jguaneri.notifications.web.controller;

import java.security.Principal;
import java.util.concurrent.CountDownLatch;

import org.atmosphere.cpr.AtmosphereResource;
import org.atmosphere.cpr.AtmosphereResourceEvent;
import org.atmosphere.cpr.AtmosphereResourceEventListenerAdapter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;

import com.github.jguaneri.notifications.Client;
import com.github.jguaneri.notifications.service.ChannelService;
import com.github.jguaneri.notifications.service.NotificationService;

/**
 * @author Joseph Guaneri
 * @version 1.0
 * @since 1.0
 */
@Controller
public class NotificationController {

    private static final Logger LOGGER = LoggerFactory.getLogger(NotificationController.class);

    @Autowired
    private NotificationService notificationService;

    @Autowired
    private ChannelService channelService;

    /**
     * Default Constructor
     * 
     * @since 1.0
     */
    public NotificationController() {
        if (LOGGER.isDebugEnabled()) {
            LOGGER.trace("Instantiating NotificationController");
        }
    }

    /**
     * @since 1.0
     * @param channel
     * @param resource
     */
    @RequestMapping(value = "/subscribe/{channel}", method = RequestMethod.GET)
    @ResponseBody
    public void subscribe(@PathVariable("channel") final String channel, final AtmosphereResource resource,
            Principal principal) throws Exception {
        final CountDownLatch countDownLatch = new CountDownLatch(1);
        resource.addEventListener(new AtmosphereResourceEventListenerAdapter() {
            @Override
            public void onSuspend(AtmosphereResourceEvent event) {
                countDownLatch.countDown();
                LOGGER.info("Suspending Client..." + resource.uuid());
                resource.removeEventListener(this);
            }

            @Override
            public void onDisconnect(AtmosphereResourceEvent event) {
                LOGGER.info("Disconnecting Client..." + resource.uuid());
                super.onDisconnect(event);
            }

            @Override
            public void onBroadcast(AtmosphereResourceEvent event) {
                LOGGER.info("Client is broadcasting..." + resource.uuid());
                super.onBroadcast(event);
            }

        });

        if (AtmosphereResource.TRANSPORT.LONG_POLLING.equals(resource.transport())) {
            resource.resumeOnBroadcast(true).suspend(-1);
        } else {
            resource.suspend(-1);
        }

        try {
            countDownLatch.await();
        } catch (InterruptedException e) {
            LOGGER.error("Interrupted while trying to suspend resource {}", resource);
        }

        if (LOGGER.isDebugEnabled()) {
            LOGGER.debug("Subscribing to channel: {}", channel);
        }
        final Client client = new Client();
        client.setSubscriptionId(resource.uuid());
        channelService.subscribe(client, channel);
    }

    @RequestMapping(value = "/acknowledge/{notificationId}", method = RequestMethod.POST)
    public void acknowledge(@PathVariable("notificationId") String notificationId, Principal principal)
            throws Exception {
        notificationService.acknowledgeNotification(notificationId, principal.getName());
    }
}