ru.org.linux.spring.SearchController.java Source code

Introduction

Here is the source code for ru.org.linux.spring.SearchController.java

Source

/*
 * Copyright 1998-2010 Linux.org.ru
 *    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 ru.org.linux.spring;

import java.sql.Connection;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import ru.org.linux.site.LorDataSource;
import ru.org.linux.site.SearchItem;
import ru.org.linux.site.SearchViewer;

import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Required;
import org.springframework.jdbc.support.JdbcUtils;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.servlet.ModelAndView;

@Controller
public class SearchController {
    private SolrServer solrServer;

    @Autowired
    @Required
    public void setSolrServer(SolrServer solrServer) {
        this.solrServer = solrServer;
    }

    @RequestMapping(value = "/search.jsp", method = { RequestMethod.GET, RequestMethod.HEAD })
    public ModelAndView search(@RequestParam(value = "q", defaultValue = "") String q,
            @RequestParam(value = "include", required = false) String includeString,
            @RequestParam(value = "noinclude_title", defaultValue = "false") boolean noinclude_title,
            @RequestParam(value = "date", required = false) String dateString,
            @RequestParam(value = "section", required = false) Integer section,
            @RequestParam(value = "sort", required = false) Integer sort,
            @RequestParam(value = "username", required = false) String username,
            @RequestParam(value = "usertopic", defaultValue = "false") boolean usertopic) throws Exception {
        Map<String, Object> params = new HashMap<String, Object>();

        boolean initial = q.isEmpty();
        params.put("initial", initial);

        params.put("usertopic", usertopic);

        params.put("q", q);

        int include = parseInclude(includeString);

        params.put("include", include);
        params.put("noinclude_title", noinclude_title);

        SearchViewer.SearchInterval date = parseInterval(dateString);

        params.put("date", date);

        if (section == null) {
            section = 0;
        }

        params.put("section", section);

        if (sort == null) {
            sort = SearchViewer.SORT_R;
        }

        params.put("sort", sort);

        if (username == null) {
            username = "";
        }

        params.put("username", username);

        if (!initial) {
            SearchViewer sv = new SearchViewer(q);

            sv.setInterval(date);
            sv.setInclude(include, noinclude_title);
            sv.setSection(section);
            sv.setSort(sort);
            sv.setUser(username);
            sv.setUserTopic(usertopic);

            Connection db = null;
            try {
                long current = System.currentTimeMillis();
                db = LorDataSource.getConnection();
                QueryResponse response = sv.performSearch(solrServer, db);

                SolrDocumentList list = response.getResults();
                List<SearchItem> res = new ArrayList<SearchItem>(list.size());
                for (SolrDocument doc : list) {
                    res.add(new SearchItem(db, doc));
                }

                long time = System.currentTimeMillis() - current;

                params.put("result", res);
                params.put("searchTime", response.getElapsedTime());
                params.put("numFound", list.getNumFound());

                params.put("time", time);
            } finally {
                JdbcUtils.closeConnection(db);
            }
        }

        return new ModelAndView("search", params);
    }

    public static int parseInclude(String include) {
        if (include == null) {
            return SearchViewer.SEARCH_ALL;
        }

        if ("topics".equals(include)) {
            return SearchViewer.SEARCH_TOPICS;
        }

        return SearchViewer.SEARCH_ALL;
    }

    public static SearchViewer.SearchInterval parseInterval(String date) {
        if (date == null) {
            return SearchViewer.DEFAULT_INTERVAL;
        }

        if ("3month".equalsIgnoreCase(date)) {
            return SearchViewer.SearchInterval.THREE_MONTH; // support for old url's
        }

        return SearchViewer.SearchInterval.valueOf(date.toUpperCase());
    }
}