Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.janusgraph.diskstorage.es.compat.AbstractESCompat Maven / Gradle / Ivy
// Copyright 2017 JanusGraph 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 org.janusgraph.diskstorage.es.compat;
import static org.janusgraph.diskstorage.es.ElasticSearchConstants.ES_ANALYZER;
import static org.janusgraph.diskstorage.es.ElasticSearchConstants.ES_INLINE_KEY;
import static org.janusgraph.diskstorage.es.ElasticSearchConstants.ES_LANG_KEY;
import static org.janusgraph.diskstorage.es.ElasticSearchConstants.ES_SCRIPT_KEY;
import static org.janusgraph.diskstorage.es.ElasticSearchConstants.ES_TYPE_KEY;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;
import org.janusgraph.core.Cardinality;
import org.janusgraph.core.attribute.Geo;
import org.janusgraph.core.schema.Mapping;
import org.janusgraph.core.schema.Parameter;
import org.janusgraph.diskstorage.es.ElasticSearchRequest;
import org.janusgraph.diskstorage.indexing.IndexFeatures;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
/**
* Base class for building Elasticsearch mapping and query objects.
*/
public abstract class AbstractESCompat {
static final Map MATCH_ALL = ImmutableMap.of("match_all", Collections.EMPTY_MAP);
static IndexFeatures.Builder coreFeatures() {
return new IndexFeatures.Builder()
.setDefaultStringMapping(Mapping.TEXT)
.supportedStringMappings(Mapping.TEXT, Mapping.TEXTSTRING, Mapping.STRING)
.setWildcardField("_all")
.supportsCardinality(Cardinality.SINGLE)
.supportsCardinality(Cardinality.LIST)
.supportsCardinality(Cardinality.SET)
.supportsNanoseconds()
.supportsCustomAnalyzer()
.supportNotQueryNormalForm()
;
}
public abstract IndexFeatures getIndexFeatures();
public Map createKeywordMapping() {
return ImmutableMap.of(ES_TYPE_KEY, "keyword");
}
public Map createTextMapping(String textAnalyzer) {
final ImmutableMap.Builder builder = ImmutableMap.builder().put(ES_TYPE_KEY, "text");
return (textAnalyzer != null ? builder.put(ES_ANALYZER, textAnalyzer) : builder).build();
}
public String scriptLang() {
return "painless";
}
public ImmutableMap.Builder prepareScript(String inline) {
final Map script = ImmutableMap.of(ES_INLINE_KEY, inline, ES_LANG_KEY, scriptLang());
return ImmutableMap.builder().put(ES_SCRIPT_KEY, script);
}
public Map prepareQuery(Map query) {
return query;
}
public Map term(String key, Object value) {
return ImmutableMap.of("term", ImmutableMap.of(key, value));
}
public Map contains(String key, List terms) {
return boolMust(terms.stream().map(term -> term(key, term)).collect(Collectors.toList()));
}
public Map boolMust(List> queries) {
return queries.size() > 1 ? ImmutableMap.of("bool", ImmutableMap.of("must", queries)) : queries.get(0);
}
public Map boolMustNot(Map query) {
return ImmutableMap.of("bool", ImmutableMap.of("must_not", query));
}
public Map boolShould(List> queries) {
return ImmutableMap.of("bool", ImmutableMap.of("should", queries));
}
public Map boolFilter(Map query) {
return ImmutableMap.of("bool", ImmutableMap.of("must", MATCH_ALL, "filter", query));
}
public Map lt(String key, Object value) {
return ImmutableMap.of("range", ImmutableMap.of(key, ImmutableMap.of("lt", value)));
}
public Map lte(String key, Object value) {
return ImmutableMap.of("range", ImmutableMap.of(key, ImmutableMap.of("lte", value)));
}
public Map gt(String key, Object value) {
return ImmutableMap.of("range", ImmutableMap.of(key, ImmutableMap.of("gt", value)));
}
public Map gte(String key, Object value) {
return ImmutableMap.of("range", ImmutableMap.of(key, ImmutableMap.of("gte", value)));
}
public Map prefix(String key, Object value) {
return ImmutableMap.of("prefix", ImmutableMap.of(key, value));
}
public Map regexp(String key, Object value) {
return ImmutableMap.of("regexp", ImmutableMap.of(key, value));
}
public Map match(String key, Object value) {
return match(key, value, null);
}
public Map fuzzyMatch(String key, Object value) {
return match(key, value, "AUTO");
}
public Map match(String key, Object value, String fuzziness) {
final ImmutableMap.Builder builder = ImmutableMap.builder().put("query", value);
builder.put("operator", "and");
if (fuzziness != null) builder.put("fuzziness", fuzziness);
return ImmutableMap.of("match", ImmutableMap.of(key, builder.build()));
}
public Map queryString(Object query) {
return ImmutableMap.of("query_string", ImmutableMap.of("query", query));
}
public Map geoDistance(String key, double lat, double lon, double radius) {
return filter(ImmutableMap.of("geo_distance", ImmutableMap.of("distance", radius + "km", key, ImmutableList.of(lon, lat))));
}
public Map geoBoundingBox(String key, double minLat, double minLon, double maxLat, double maxLon) {
return filter(ImmutableMap.of("geo_bounding_box", ImmutableMap.of(key,ImmutableMap.of(
"top_left", ImmutableList.of(minLon, maxLat),"bottom_right", ImmutableList.of(maxLon, minLat)))));
}
public Map geoPolygon(String key, List> points) {
return filter(ImmutableMap.of("geo_polygon", ImmutableMap.of(key, ImmutableMap.of("points", points))));
}
public Map geoShape(String key, Map geoShape, Geo predicate) {
final String relation = predicate == Geo.INTERSECT ? "intersects" : predicate.name().toLowerCase();
return filter(ImmutableMap.of("geo_shape", ImmutableMap.of(key, ImmutableMap.of("shape", geoShape, "relation", relation))));
}
public Map filter(Map query) {
return boolFilter(query);
}
public Map createRequestBody(ElasticSearchRequest request, Parameter[] parameters) {
final Map requestBody = new HashMap<>();
Optional.ofNullable(request.getSize()).ifPresent(parameter -> requestBody.put("size", parameter));
Optional.ofNullable(request.getFrom()).ifPresent(parameter -> requestBody.put("from", parameter));
if (!request.getSorts().isEmpty()) {
requestBody.put("sort", request.getSorts());
}
Optional.ofNullable(request.getQuery()).ifPresent(parameter -> requestBody.put("query", parameter));
Optional.ofNullable(parameters).ifPresent(p -> Arrays.stream(p).forEachOrdered(parameter -> requestBody.put(parameter.key(), parameter.value())));
return requestBody;
}
}
