///*
|
// * Copyright 2014-2016 the original author or authors.
|
// *
|
// * 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.walker.support.es.impl;
|
//
|
//
|
//import com.fasterxml.jackson.core.JsonEncoding;
|
//import com.fasterxml.jackson.core.JsonFactory;
|
//import com.fasterxml.jackson.core.JsonGenerator;
|
//import com.walker.infrastructure.utils.StringUtils;
|
//import com.xinyuan.core.persistence.search.so.BaseSearchObject;
|
//import org.apache.commons.lang.StringUtils;
|
//import org.elasticsearch.action.get.GetResponse;
|
//import org.elasticsearch.action.get.MultiGetItemResponse;
|
//import org.elasticsearch.action.get.MultiGetResponse;
|
//import org.elasticsearch.action.search.SearchResponse;
|
//import org.elasticsearch.common.document.DocumentField;
|
//import org.elasticsearch.common.text.Text;
|
//import org.elasticsearch.search.SearchHit;
|
//import org.elasticsearch.search.fetch.subphase.highlight.HighlightField;
|
//import org.slf4j.Logger;
|
//import org.slf4j.LoggerFactory;
|
//import org.springframework.data.domain.Pageable;
|
//import org.springframework.data.elasticsearch.ElasticsearchException;
|
//import org.springframework.data.elasticsearch.annotations.Document;
|
//import org.springframework.data.elasticsearch.annotations.ScriptedField;
|
//import org.springframework.data.elasticsearch.core.AbstractResultMapper;
|
//import org.springframework.data.elasticsearch.core.DefaultEntityMapper;
|
//import org.springframework.data.elasticsearch.core.EntityMapper;
|
//import org.springframework.data.elasticsearch.core.aggregation.AggregatedPage;
|
//import org.springframework.data.elasticsearch.core.aggregation.impl.AggregatedPageImpl;
|
//import org.springframework.data.elasticsearch.core.mapping.ElasticsearchPersistentEntity;
|
//import org.springframework.data.elasticsearch.core.mapping.ElasticsearchPersistentProperty;
|
//import org.springframework.data.elasticsearch.core.mapping.SimpleElasticsearchMappingContext;
|
//import org.springframework.data.mapping.PersistentProperty;
|
//import org.springframework.data.mapping.context.MappingContext;
|
//import org.springframework.stereotype.Component;
|
//
|
//import java.io.ByteArrayOutputStream;
|
//import java.io.IOException;
|
//import java.nio.charset.Charset;
|
//import java.util.*;
|
//
|
///**
|
// * 扩展结果转换对象，增加高亮摘要字段内容处理
|
// *
|
// * @author Artur Konczak
|
// * @author Petar Tahchiev
|
// * @author Young Gu
|
// * @author Oliver Gierke
|
// */
|
//@Component
|
//public class DefaultResultMapper extends AbstractResultMapper {
|
// private static final Logger logger = LoggerFactory.getLogger(DefaultResultMapper.class);
|
//
|
// private MappingContext<? extends ElasticsearchPersistentEntity<?>, ElasticsearchPersistentProperty> mappingContext;
|
//
|
// public DefaultResultMapper(MappingContext<? extends ElasticsearchPersistentEntity<?>, ElasticsearchPersistentProperty> mappingContext) {
|
// super(new DefaultEntityMapper(mappingContext));
|
// this.mappingContext = mappingContext;
|
// }
|
//
|
// public DefaultResultMapper(EntityMapper entityMapper) {
|
// super(entityMapper);
|
// }
|
// public DefaultResultMapper() {
|
// this(new SimpleElasticsearchMappingContext());
|
// }
|
//
|
// public DefaultResultMapper(
|
// MappingContext<? extends ElasticsearchPersistentEntity<?>, ElasticsearchPersistentProperty> mappingContext,
|
// EntityMapper entityMapper) {
|
// super(entityMapper);
|
// this.mappingContext = mappingContext;
|
// }
|
//
|
// @Override
|
// public <T> AggregatedPage<T> mapResults(SearchResponse response, Class<T> clazz, Pageable pageable) {
|
// //logger.info("highlight -> {}", ToStringBuilder.reflectionToString(response, ToStringStyle.MULTI_LINE_STYLE));
|
// long totalHits = response.getHits().getTotalHits();
|
// List<T> results = new ArrayList<>();
|
// for (SearchHit hit : response.getHits()) {
|
// if (hit != null) {
|
// T result;
|
// if (StringUtils.isNotBlank(hit.getSourceAsString())) {
|
// result = mapEntity(hit.getSourceAsString(), clazz);
|
// } else {
|
// result = mapEntity(hit.getFields().values(), clazz);
|
// }
|
// setPersistentEntityId(result, hit.getId(), clazz);
|
// populateScriptFields(result, hit);
|
//
|
// // 处理高亮内容
|
// this.handleHighlight(hit, result);
|
// results.add(result);
|
// }
|
// }
|
//
|
// return new AggregatedPageImpl<>(results, pageable, totalHits, response.getAggregations());
|
// }
|
//
|
// private <T> void populateScriptFields(T result, SearchHit hit) {
|
// if (hit.getFields() != null && !hit.getFields().isEmpty() && result != null) {
|
// for (java.lang.reflect.Field field : result.getClass().getDeclaredFields()) {
|
// ScriptedField scriptedField = field.getAnnotation(ScriptedField.class);
|
// if (scriptedField != null) {
|
// String name = scriptedField.name().isEmpty() ? field.getName() : scriptedField.name();
|
// DocumentField searchHitField = hit.getFields().get(name);
|
// if (searchHitField != null) {
|
// field.setAccessible(true);
|
// try {
|
// field.set(result, searchHitField.getValue());
|
// } catch (IllegalArgumentException e) {
|
// throw new ElasticsearchException("failed to set scripted field: " + name + " with value: " + searchHitField.getValue(), e);
|
// } catch (IllegalAccessException e) {
|
// throw new ElasticsearchException("failed to access scripted field: " + name, e);
|
// }
|
// }
|
// }
|
// }
|
// }
|
// }
|
//
|
//
|
// private <T> T mapEntity(Collection<DocumentField> values, Class<T> clazz) {
|
// return mapEntity(buildJSONFromFields(values), clazz);
|
// }
|
//
|
// private String buildJSONFromFields(Collection<DocumentField> values) {
|
// JsonFactory nodeFactory = new JsonFactory();
|
// try {
|
// ByteArrayOutputStream stream = new ByteArrayOutputStream();
|
// JsonGenerator generator = nodeFactory.createGenerator(stream, JsonEncoding.UTF8);
|
// generator.writeStartObject();
|
// for (DocumentField value : values) {
|
// if (value.getValues().size() > 1) {
|
// generator.writeArrayFieldStart(value.getName());
|
// for (Object val : value.getValues()) {
|
// generator.writeObject(val);
|
// }
|
// generator.writeEndArray();
|
// } else {
|
// generator.writeObjectField(value.getName(), value.getValue());
|
// }
|
// }
|
// generator.writeEndObject();
|
// generator.flush();
|
// return new String(stream.toByteArray(), Charset.forName("UTF-8"));
|
// } catch (IOException e) {
|
// logger.error(e.getMessage(), e);
|
// return null;
|
// }
|
// }
|
//
|
// @Override
|
// public <T> T mapResult(GetResponse response, Class<T> clazz) {
|
// T result = mapEntity(response.getSourceAsString(), clazz);
|
// if (result != null) {
|
// setPersistentEntityId(result, response.getId(), clazz);
|
// }
|
// return result;
|
// }
|
//
|
// @Override
|
// public <T> LinkedList<T> mapResults(MultiGetResponse responses, Class<T> clazz) {
|
// LinkedList<T> list = new LinkedList<>();
|
// for (MultiGetItemResponse response : responses.getResponses()) {
|
// if (!response.isFailed() && response.getResponse().isExists()) {
|
// T result = mapEntity(response.getResponse().getSourceAsString(), clazz);
|
// setPersistentEntityId(result, response.getResponse().getId(), clazz);
|
// list.add(result);
|
// }
|
// }
|
// return list;
|
// }
|
//
|
// private <T> void setPersistentEntityId(T result, String id, Class<T> clazz) {
|
//
|
// if (mappingContext != null && clazz.isAnnotationPresent(Document.class)) {
|
//
|
// ElasticsearchPersistentEntity<?> persistentEntity = mappingContext.getPersistentEntity(clazz);
|
// PersistentProperty<?> idProperty = persistentEntity.getIdProperty();
|
//
|
// // Only deal with String because ES generated Ids are strings !
|
// if (idProperty != null && idProperty.getType().isAssignableFrom(String.class)) {
|
// persistentEntity.getPropertyAccessor(result).setProperty(idProperty, id);
|
// }
|
// }
|
// }
|
//
|
// /**
|
// * 处理高亮显示内容，将高亮数据放入最外层的highlightMap对象
|
// *
|
// * @param hit 命中的查询结果
|
// * @param obj 待返回的查询结果对象
|
// */
|
// private void handleHighlight(SearchHit hit, Object obj) {
|
// //logger.info("highlight -> {}", ToStringBuilder.reflectionToString(hit.highlightFields().keySet(), ToStringStyle.MULTI_LINE_STYLE));
|
// if (obj instanceof BaseSearchObject) {
|
// BaseSearchObject searchPo = (BaseSearchObject) obj;
|
// if (hit != null) {
|
// Map<String, HighlightField> highlightFieldMap = hit.getHighlightFields();
|
// if (highlightFieldMap != null) {
|
// for (Map.Entry<String, HighlightField> entry : highlightFieldMap.entrySet()) {
|
// HighlightField highlightField = entry.getValue();
|
// Text[] texts = highlightField.fragments();
|
// if (texts != null) {
|
// for (Text text : texts) {
|
// if (text != null) {
|
// searchPo.putHighlight(entry.getKey(), text.toString().replaceAll("\\s+", " ").trim());
|
// }
|
// }
|
// }
|
// }
|
// }
|
// }
|
// }
|
// }
|
//}
|
