/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.ambari.logsearch.manager;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import com.google.common.collect.Lists;
import org.apache.ambari.logsearch.common.LogSearchConstants;
import org.apache.ambari.logsearch.model.response.GroupListResponse;
import org.apache.ambari.logsearch.model.response.LogData;
import org.apache.ambari.logsearch.model.response.LogSearchResponse;
import org.apache.ambari.logsearch.dao.SolrDaoBase;
import org.apache.ambari.logsearch.util.SolrUtil;
import org.apache.commons.collections.CollectionUtils;
import org.apache.log4j.Logger;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.response.FacetField;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocumentList;
import org.springframework.data.solr.core.DefaultQueryParser;
import org.springframework.data.solr.core.query.SimpleQuery;
import org.springframework.data.solr.core.query.SolrDataQuery;
import static org.apache.ambari.logsearch.solr.SolrConstants.CommonLogConstants.CLUSTER;
public abstract class ManagerBase<LOG_DATA_TYPE extends LogData, SEARCH_RESPONSE extends LogSearchResponse> extends JsonManagerBase {
private static final Logger logger = Logger.getLogger(ManagerBase.class);
public ManagerBase() {
super();
}
protected SEARCH_RESPONSE getLastPage(SolrDaoBase solrDoaBase, SimpleQuery lastPageQuery, String event) {
int maxRows = lastPageQuery.getRows();
SEARCH_RESPONSE logResponse = getLogAsPaginationProvided(lastPageQuery, solrDoaBase, event);
Long totalLogs = logResponse.getTotalCount();
int startIndex = (int)(totalLogs - totalLogs % maxRows);
int numberOfLogsOnLastPage = (int)(totalLogs - startIndex);
logResponse.setStartIndex(startIndex);
logResponse.setTotalCount(totalLogs);
logResponse.setPageSize(maxRows);
List<LOG_DATA_TYPE> docList = logResponse.getLogList();
List<LOG_DATA_TYPE> lastPageDocList = new ArrayList<>();
logResponse.setLogList(lastPageDocList);
int cnt = 0;
for (LOG_DATA_TYPE doc : docList) {
if (cnt < numberOfLogsOnLastPage) {
lastPageDocList.add(doc);
}
cnt++;
}
Collections.reverse(lastPageDocList);
return logResponse;
}
protected SEARCH_RESPONSE getLogAsPaginationProvided(SolrDataQuery solrQuery, SolrDaoBase solrDaoBase, String event) {
SolrQuery query = new DefaultQueryParser().doConstructSolrQuery(solrQuery);
return getLogAsPaginationProvided(query, solrDaoBase, event);
}
protected SEARCH_RESPONSE getLogAsPaginationProvided(SolrQuery solrQuery, SolrDaoBase solrDaoBase, String event) {
QueryResponse response = solrDaoBase.process(solrQuery, event);
SEARCH_RESPONSE logResponse = createLogSearchResponse();
SolrDocumentList docList = response.getResults();
logResponse.setTotalCount(docList.getNumFound());
List<LOG_DATA_TYPE> serviceLogDataList = convertToSolrBeans(response);
if (!docList.isEmpty()) {
logResponse.setLogList(serviceLogDataList);
logResponse.setStartIndex((int) docList.getStart());
Integer rowNumber = solrQuery.getRows();
if (rowNumber == null) {
logger.error("No RowNumber was set in solrQuery");
return createLogSearchResponse();
}
logResponse.setPageSize(rowNumber);
}
return logResponse;
}
protected abstract List<LOG_DATA_TYPE> convertToSolrBeans(QueryResponse response);
protected abstract SEARCH_RESPONSE createLogSearchResponse();
protected List<String> getClusters(SolrDaoBase solrDaoBase, String clusterField, String event) {
List<String> clusterResponse = Lists.newArrayList();
SolrQuery solrQuery = new SolrQuery();
solrQuery.setQuery("*:*");
SolrUtil.setFacetField(solrQuery, clusterField);
SolrUtil.setFacetSort(solrQuery, LogSearchConstants.FACET_INDEX);
QueryResponse response = solrDaoBase.process(solrQuery, event);
if (response == null) {
return clusterResponse;
}
List<FacetField> clusterFields = response.getFacetFields();
if (CollectionUtils.isNotEmpty(clusterFields)) {
FacetField clusterFacets = clusterFields.get(0);
for (FacetField.Count clusterCount : clusterFacets.getValues()) {
clusterResponse.add(clusterCount.getName());
}
}
return clusterResponse;
}
}