com.elefana.search.SearchHitsQueryExecutor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elefana-core Show documentation
null
The newest version!
/*******************************************************************************
 * Copyright 2018 Viridian Software Limited
 *
 * 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.elefana.search;

import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;

import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.support.rowset.SqlRowSet;

import com.codahale.metrics.Histogram;
import com.elefana.api.search.SearchHit;
import com.elefana.api.search.SearchResponse;
import com.elefana.util.IndexUtils;
import com.jsoniter.JsonIterator;
import com.jsoniter.spi.TypeLiteral;

public abstract class SearchHitsQueryExecutor {
	protected final JdbcTemplate jdbcTemplate;
	protected final Histogram searchHitsTime, searchHits;
	
	public SearchHitsQueryExecutor(JdbcTemplate jdbcTemplate, Histogram searchHitsTime, Histogram searchHits) {
		super();
		this.jdbcTemplate = jdbcTemplate;
		this.searchHitsTime = searchHitsTime;
		this.searchHits = searchHits;
	}
	
	public abstract SqlRowSet queryHitsCount(PsqlQueryComponents queryComponents, long startTime, int from, int size);

	public abstract SqlRowSet queryHits(PsqlQueryComponents queryComponents, long startTime, int from, int size);
	
	public Callable executeCountQuery(final SearchResponse searchResponse, PsqlQueryComponents queryComponents, long startTime, int from, int size) {
		return new Callable() {
			@Override
			public SearchResponse call() throws Exception {
				SqlRowSet countRowSet = null;
				try {
					countRowSet = queryHitsCount(queryComponents, startTime, from, size);
				} catch (Exception e) {
					e.printStackTrace();
					if (!e.getMessage().contains("No results")) {
						throw e;
					}
					countRowSet = null;
				}
				if (countRowSet == null) {
					searchResponse.getHits().setTotal(0);
				} else {
					searchResponse.getHits().setTotal(getTotalHits(countRowSet));
				}
				searchHits.update(searchResponse.getHits().getTotal());
				return searchResponse;
			}
		};
	}

	public Callable executeHitsQuery(final SearchResponse searchResponse, PsqlQueryComponents queryComponents, long startTime, int from, int size) {
		return new Callable() {
			@Override
			public SearchResponse call() throws Exception {
				if(size > 0) {
					SqlRowSet hitsRowSet = null;
					try {
						hitsRowSet = queryHits(queryComponents, startTime, from, size);
					} catch (Exception e) {
						e.printStackTrace();
						if (!e.getMessage().contains("No results")) {
							throw e;
						}
						hitsRowSet = null;
					}
	
					if (hitsRowSet != null) {
						populateHits(searchResponse.getHits().getHits(), hitsRowSet);
					}
					searchResponse.getHits().setMaxScore(1.0);
	
					searchHitsTime.update(System.currentTimeMillis() - startTime);
				}
				return searchResponse;
			}
		};
	}
	
	private int getTotalHits(SqlRowSet countsRowSet) {
		int totalHits = 0;
		boolean hasCountColumn = true;
		while (countsRowSet.next()) {
			if (hasCountColumn) {
				try {
					totalHits += countsRowSet.getInt("count");
				} catch (Exception e) {
					hasCountColumn = false;
				}
			}
			if (!hasCountColumn) {
				totalHits++;
			}
		}
		return totalHits;
	}
	
	private void populateHits(List results, SqlRowSet hitsRowSet) {
		while (hitsRowSet.next()) {
			SearchHit searchHit = new SearchHit();
			searchHit._index = hitsRowSet.getString("_index");
			searchHit._type = hitsRowSet.getString("_type");
			searchHit._id = hitsRowSet.getString("_id");
			searchHit._score = 1.0;
			searchHit._source = JsonIterator.deserialize(IndexUtils.psqlUnescapeString(hitsRowSet.getString("_source")),
					new TypeLiteral>() {
					});
			results.add(searchHit);
		}
	}
}