com.venilnoronha.dzone.feed.web.ArticlesRSSController.java Source code

Introduction

Here is the source code for com.venilnoronha.dzone.feed.web.ArticlesRSSController.java

Source

/*
 * Copyright 2015 Venil Noronha
 * 
 * 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 com.venilnoronha.dzone.feed.web;

import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.Iterator;
import java.util.List;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort.Direction;
import org.springframework.http.HttpEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.servlet.ModelAndView;

import com.venilnoronha.dzone.feed.model.Article;
import com.venilnoronha.dzone.feed.mongo.ArticlesRepository;
import com.venilnoronha.dzone.feed.util.HttpUtils;

/**
 * 
 * @author Venil Noronha
 */
@Controller
public class ArticlesRSSController {

    private static final DateFormat HTTP_FORMAT = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss 'GMT'");

    @Value("${articles.feedSize}")
    private int feedSize;

    @Autowired
    private ArticlesRepository articlesRepository;

    @RequestMapping(value = "/rss/articles", method = RequestMethod.GET)
    public ModelAndView articlesRssByLastModified(HttpEntity<byte[]> requestEntity, HttpServletRequest request,
            HttpServletResponse response) throws ParseException {
        HttpUtils.logUserIP("ArticlesRSS", request);
        String lastModified = requestEntity.getHeaders().getFirst("If-Modified-Since");
        if (lastModified == null) {
            lastModified = requestEntity.getHeaders().getFirst("If-None-Match");
        }
        PageRequest page = new PageRequest(0, feedSize, Direction.DESC, "articleDate");
        Page<Article> articlePage;
        if (lastModified != null) {
            Date lastModifiedDt = HTTP_FORMAT.parse(lastModified);
            articlePage = articlesRepository.findByArticleDateBetween(lastModifiedDt, new Date(), page);
        } else {
            articlePage = articlesRepository.findAll(page);
        }
        Iterator<Article> articlesIter = articlePage.iterator();
        List<Article> articles = new ArrayList<>();
        Date newLastModified = null;
        while (articlesIter.hasNext()) {
            Article article = articlesIter.next();
            Date articleDate = article.getArticleDate();
            if (newLastModified == null || newLastModified.before(articleDate)) {
                newLastModified = articleDate;
            }
            articles.add(article);
        }
        if (articles.isEmpty() && lastModified != null) {
            response.setStatus(HttpServletResponse.SC_NOT_MODIFIED);
            return null;
        } else {
            response.setHeader("ETag", HTTP_FORMAT.format(newLastModified));
            response.setHeader("Last-Modified", HTTP_FORMAT.format(newLastModified));
            ModelAndView mav = new ModelAndView();
            mav.setView(new ArticlesRSSView(articles));
            return mav;
        }
    }

}