de.fatalix.bookery.bl.solr.SolrHandler.java Source code

Introduction

Here is the source code for de.fatalix.bookery.bl.solr.SolrHandler.java

Source

/*
 * Copyright (c) 2015 Felix Husse under MIT License
 * see LICENSE file
 */
package de.fatalix.bookery.bl.solr;

import de.fatalix.bookery.bl.BookeryService;
import de.fatalix.bookery.bl.dao.AppSettingDAO;
import de.fatalix.bookery.solr.model.BookEntry;
import java.io.IOException;
import java.util.List;
import javax.ejb.Stateless;
import javax.inject.Inject;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrServer;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.client.solrj.response.UpdateResponse;
import org.apache.solr.common.SolrInputDocument;

/**
 *
 * @author felix.husse
 */
@Stateless
public class SolrHandler {
    @Inject
    private BookeryService bookeryService;

    public void addBeans(List<BookEntry> bookEntries) throws SolrServerException, IOException {
        SolrServer solr = bookeryService.getSolrConnection();
        solr.addBeans(bookEntries);
        UpdateResponse response = solr.commit(true, false);
    }

    public void updateBean(BookEntry bookEntry) throws SolrServerException, IOException {
        SolrServer solr = bookeryService.getSolrConnection();
        solr.addBean(bookEntry);
        solr.commit();

    }

    public void updateDocument(List<SolrInputDocument> solrDocs) throws SolrServerException, IOException {
        SolrServer solr = bookeryService.getSolrConnection();
        solr.add(solrDocs);
        solr.commit();
    }

    public void resetSolrIndex() throws SolrServerException, IOException {
        SolrServer solr = bookeryService.getSolrConnection();
        solr.deleteByQuery("*:*");
        solr.commit();
    }

    public QueryResponse searchSolr(SolrQuery solrQuery) throws SolrServerException {
        try {
            SolrServer solr = bookeryService.getSolrConnection();
            return solr.query(solrQuery);
        } catch (IOException ex) {
            throw new SolrServerException(ex.getMessage());
        }
    }

    public QueryResponse searchSolrIndex(String queryString, String fields, int rows, int startOffset)
            throws SolrServerException {
        SolrServer solr = null;
        try {
            solr = bookeryService.getSolrConnection();
        } catch (IOException ex) {
            throw new SolrServerException(ex.getMessage());
        }
        SolrQuery query = new SolrQuery();
        query.setQuery(queryString);
        query.setRows(rows);
        query.setStart(startOffset);

        query.setFields(fields);
        QueryResponse rsp = solr.query(query);
        return rsp;
    }

    public QueryResponse searchSolrIndexSorted(String queryString, String fields, int rows, int startOffset,
            String sortField) throws SolrServerException {
        SolrServer solr = null;
        try {
            solr = bookeryService.getSolrConnection();
        } catch (IOException ex) {
            throw new SolrServerException(ex.getMessage());
        }
        SolrQuery query = new SolrQuery();
        query.setQuery(queryString);
        query.setRows(rows);
        query.setStart(startOffset);
        query.setSort(sortField, SolrQuery.ORDER.desc);
        query.setFields(fields);
        QueryResponse rsp = solr.query(query);
        return rsp;
    }

    public List<BookEntry> getBookDetail(String bookID) throws SolrServerException {
        SolrServer solr = null;
        try {
            solr = bookeryService.getSolrConnection();
        } catch (IOException ex) {
            throw new SolrServerException(ex.getMessage());
        }
        SolrQuery query = new SolrQuery();
        query.setQuery("id:" + bookID);
        query.setRows(1);
        query.setFields(
                "id,author,title,isbn,publisher,description,language,releaseDate,likes,downloadcount,uploader,viewed,shared,cover,thumbnail,thumbnailgenerated,likedby");
        QueryResponse rsp = solr.query(query);
        return rsp.getBeans(BookEntry.class);
    }

    public List<BookEntry> getEpubBook(String bookID) throws SolrServerException {
        SolrServer solr = null;
        try {
            solr = bookeryService.getSolrConnection();
        } catch (IOException ex) {
            throw new SolrServerException(ex.getMessage());
        }
        SolrQuery query = new SolrQuery();
        query.setQuery("id:" + bookID);
        query.setRows(1);
        query.setFields("id,epub");
        QueryResponse rsp = solr.query(query);
        return rsp.getBeans(BookEntry.class);
    }

    public List<BookEntry> getMobiFormat(String bookID) throws SolrServerException {
        SolrServer solr = null;
        try {
            solr = bookeryService.getSolrConnection();
        } catch (IOException ex) {
            throw new SolrServerException(ex.getMessage());
        }
        SolrQuery query = new SolrQuery();
        query.setQuery("id:" + bookID);
        query.setRows(1);
        query.setFields("id,mobi");
        QueryResponse rsp = solr.query(query);
        return rsp.getBeans(BookEntry.class);
    }

    public long checkSolr() throws SolrServerException, IOException {
        SolrServer solr = bookeryService.getSolrConnection();
        SolrQuery query = new SolrQuery();
        query.setQuery("*:*");
        query.setRows(1);
        QueryResponse rsp = solr.query(query);

        return rsp.getResults().getNumFound();
    }
}