gr.aueb.service.EmailService.java Source code

Introduction

Here is the source code for gr.aueb.service.EmailService.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package gr.aueb.service;

import gr.aueb.demo.PropertyRegistryBean;
import gr.aueb.structures.Email;
import java.util.ArrayList;
import java.util.List;
import java.util.ListIterator;

import org.apache.solr.client.solrj.SolrClient;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.impl.HttpSolrClient;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;

/**
 *
 * @author pmeladianos
 */
public class EmailService {

    public static List<Email> createEmails() {
        ArrayList<Email> list = new ArrayList<Email>();
        list.add(new Email("0", "-", "-", "sample", "mail"));
        return list;
    }

    public static List<Email> getEmails(List<String> keywords) {
        String query = getEmailServiceQuery(keywords);
        SolrQuery solrQuery;
        solrQuery = new SolrQuery();
        if (query.startsWith(" AND")) {
            query = query.substring(4);
        }
        if (query.startsWith("AND")) {
            query = query.substring(3);
        }
        solrQuery.setQuery(query);
        if (!query.equals("")) {
            solrQuery.setHighlight(true).setHighlightSnippets(1).setHighlightSimplePost("</strong>")
                    .setHighlightSimplePre("<strong>"); //set other params as needed
            solrQuery.setParam("hl.fl", "content_t");
            solrQuery.setParam("hl.requireFieldMatch", "true");

        }

        solrQuery.setRows(15);
        String urlString = "http://195.251.252.8:8983/solr/mail";
        //SolrClient server = new HttpSolrClient.Builder(urlString).build();
        SolrClient server = new HttpSolrClient(urlString);
        QueryResponse response = null;
        try {
            response = server.query(solrQuery);

        } catch (Exception e) {
            System.out.println(e);
        }
        SolrDocumentList rs = response.getResults();
        long numFound = rs.getNumFound();
        int numResultsDisplay = (int) numFound;
        ArrayList<Email> items = new ArrayList<Email>();

        ListIterator<SolrDocument> iter = rs.listIterator();

        while (iter.hasNext()) {
            SolrDocument doc = iter.next();
            String id = doc.get("messageId").toString();
            String from = doc.get("from").toString();
            String sentDate = doc.get("sentDate").toString();
            String subject = doc.get("subject").toString();
            String content = doc.get("content").toString();

            items.add(new Email(id, from, sentDate, subject, content));
        }
        return items;
    }

    private static String getEmailServiceQuery(List<String> keywords) {
        String query = "";
        for (Object key : keywords) {
            String s = key.toString();
            query += s + " OR ";
        }
        String query_0 = queryBuilder(query, "content");

        String query_1 = queryBuilder(query, "subject");
        query = query_0 + " OR " + query_1;
        return query;
    }

    private static String queryBuilder(String s, String field) {
        String res = "";
        if (s.length() == 0) {
            res = field + ":*";
        } else {
            res = field + ":" + "'" + s + "'";
        }
        return res;

    }

}