com.azure.cosmos.models.IndexingPolicy Maven / Gradle / Ivy
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.cosmos.models;
import com.azure.cosmos.implementation.Constants;
import com.azure.cosmos.implementation.Index;
import com.azure.cosmos.implementation.JsonSerializable;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.azure.cosmos.implementation.apachecommons.lang.StringUtils;
import com.fasterxml.jackson.databind.node.ObjectNode;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/**
* Represents the indexing policy configuration for a container in the Azure Cosmos DB database service.
*/
public final class IndexingPolicy {
private static final String DEFAULT_PATH = "/*";
private List includedPaths;
private List excludedPaths;
private List> compositeIndexes;
private List spatialIndexes;
private JsonSerializable jsonSerializable;
/**
* Constructor.
*/
public IndexingPolicy() {
this.jsonSerializable = new JsonSerializable();
this.setAutomatic(true);
this.setIndexingMode(IndexingMode.CONSISTENT);
}
/**
* Initializes a new instance of the IndexingPolicy class with the specified set of indexes as
* default index specifications for the root path.
*
* The following example shows how to override the default indexingPolicy for root path:
*
* {@code
* HashIndex hashIndexOverride = Index.HASH(DataType.STRING, 5);
* RangeIndex rangeIndexOverride = Index.RANGE(DataType.NUMBER, 2);
* SpatialIndex spatialIndexOverride = Index.SPATIAL(DataType.POINT);
*
* IndexingPolicy indexingPolicy = new IndexingPolicy(hashIndexOverride, rangeIndexOverride, spatialIndexOverride);
* }
*
*
* @param defaultIndexOverrides comma separated set of indexes that serve as default index specifications for the
* root path.
* @throws IllegalArgumentException throws when defaultIndexOverrides is null
*/
IndexingPolicy(Index[] defaultIndexOverrides) {
this();
if (defaultIndexOverrides == null) {
throw new IllegalArgumentException("defaultIndexOverrides is null.");
}
IncludedPath includedPath = new IncludedPath(IndexingPolicy.DEFAULT_PATH);
includedPath.setIndexes(new ArrayList(Arrays.asList(defaultIndexOverrides)));
this.getIncludedPaths().add(includedPath);
}
/**
* Constructor.
*
* @param jsonString the json string that represents the indexing policy.
*/
IndexingPolicy(String jsonString) {
this.jsonSerializable = new JsonSerializable(jsonString);
}
/**
* Constructor.
*
* @param objectNode the object node that represents the indexing policy.
*/
IndexingPolicy(ObjectNode objectNode) {
this.jsonSerializable = new JsonSerializable(objectNode);
}
/**
* Gets whether automatic indexing is enabled for a container.
*
* In automatic indexing, items can be explicitly excluded from indexing using RequestOptions. In manual
* indexing, items can be explicitly included.
*
* @return the automatic
*/
public Boolean isAutomatic() {
return this.jsonSerializable.getBoolean(Constants.Properties.AUTOMATIC);
}
/**
* Sets whether automatic indexing is enabled for a container.
*
* In automatic indexing, items can be explicitly excluded from indexing using RequestOptions. In manual
* indexing, items can be explicitly included.
*
* @param automatic the automatic
* @return the Indexing Policy.
*/
public IndexingPolicy setAutomatic(boolean automatic) {
this.jsonSerializable.set(Constants.Properties.AUTOMATIC, automatic);
return this;
}
/**
* Gets the indexing mode (consistent or lazy).
*
* @return the indexing mode.
*/
public IndexingMode getIndexingMode() {
IndexingMode result = IndexingMode.LAZY;
try {
result = IndexingMode.valueOf(StringUtils.upperCase(this.jsonSerializable.getString(Constants.Properties.INDEXING_MODE)));
} catch (IllegalArgumentException e) {
this.jsonSerializable.getLogger().warn("INVALID indexingMode value {}.",
this.jsonSerializable.getString(Constants.Properties.INDEXING_MODE));
}
return result;
}
/**
* Sets the indexing mode (consistent or lazy).
*
* @param indexingMode the indexing mode.
* @return the Indexing Policy.
*/
public IndexingPolicy setIndexingMode(IndexingMode indexingMode) {
this.jsonSerializable.set(Constants.Properties.INDEXING_MODE, indexingMode.toString());
return this;
}
/**
* Gets the paths that are chosen to be indexed by the user.
*
* @return the included paths.
*/
public List getIncludedPaths() {
if (this.includedPaths == null) {
this.includedPaths = this.jsonSerializable.getList(Constants.Properties.INCLUDED_PATHS, IncludedPath.class);
if (this.includedPaths == null) {
this.includedPaths = new ArrayList();
}
}
return this.includedPaths;
}
/**
* Sets included paths.
*
* @param includedPaths the included paths
* @return the included paths
*/
public IndexingPolicy setIncludedPaths(List includedPaths) {
this.includedPaths = includedPaths;
return this;
}
/**
* Gets the paths that are not indexed.
*
* @return the excluded paths.
*/
public List getExcludedPaths() {
if (this.excludedPaths == null) {
this.excludedPaths = this.jsonSerializable.getList(Constants.Properties.EXCLUDED_PATHS, ExcludedPath.class);
if (this.excludedPaths == null) {
this.excludedPaths = new ArrayList();
}
}
return this.excludedPaths;
}
/**
* Sets excluded paths.
*
* @param excludedPaths the excluded paths
* @return the excluded paths
*/
public IndexingPolicy setExcludedPaths(List excludedPaths) {
this.excludedPaths = excludedPaths;
return this;
}
/**
* Gets the composite indexes for additional indexes.
*
* @return the composite indexes.
*/
public List> getCompositeIndexes() {
if (this.compositeIndexes == null) {
this.compositeIndexes = new ArrayList<>();
ArrayNode compositeIndexes = (ArrayNode) this.jsonSerializable.get(Constants.Properties.COMPOSITE_INDEXES);
if (compositeIndexes == null) {
return this.compositeIndexes;
}
for (int i = 0; i < compositeIndexes.size(); i++) {
ArrayNode compositeIndex = (ArrayNode) compositeIndexes.get(i);
ArrayList compositePaths = new ArrayList();
for (int j = 0; j < compositeIndex.size(); j++) {
CompositePath candidateCompositePath = new CompositePath(compositeIndex.get(j).toString());
compositePaths.add(candidateCompositePath);
}
this.compositeIndexes.add(compositePaths);
}
}
return this.compositeIndexes;
}
/**
* Sets the composite indexes for additional indexes.
*
* @param compositeIndexes the composite indexes.
* @return the Indexing Policy.
*/
public IndexingPolicy setCompositeIndexes(List> compositeIndexes) {
this.compositeIndexes = compositeIndexes;
this.jsonSerializable.set(Constants.Properties.COMPOSITE_INDEXES, this.compositeIndexes);
return this;
}
/**
* Sets the spatial indexes for additional indexes.
*
* @return the spatial indexes.
*/
public List getSpatialIndexes() {
if (this.spatialIndexes == null) {
this.spatialIndexes = this.jsonSerializable.getList(Constants.Properties.SPATIAL_INDEXES, SpatialSpec.class);
if (this.spatialIndexes == null) {
this.spatialIndexes = new ArrayList();
}
}
return this.spatialIndexes;
}
/**
* Sets the spatial indexes for additional indexes.
*
* @param spatialIndexes the spatial indexes.
* @return the Indexing Policy.
*/
public IndexingPolicy setSpatialIndexes(List spatialIndexes) {
this.spatialIndexes = spatialIndexes;
this.jsonSerializable.set(Constants.Properties.SPATIAL_INDEXES, this.spatialIndexes);
return this;
}
void populatePropertyBag() {
this.jsonSerializable.populatePropertyBag();
// If indexing mode is not 'none' and not paths are set, set them to the defaults
if (this.getIndexingMode() != IndexingMode.NONE && this.getIncludedPaths().size() == 0
&& this.getExcludedPaths().size() == 0) {
IncludedPath includedPath = new IncludedPath(IndexingPolicy.DEFAULT_PATH);
this.getIncludedPaths().add(includedPath);
}
if (this.includedPaths != null) {
for (IncludedPath includedPath : this.includedPaths) {
includedPath.populatePropertyBag();
}
this.jsonSerializable.set(Constants.Properties.INCLUDED_PATHS, this.includedPaths);
}
if (this.excludedPaths != null) {
for (ExcludedPath excludedPath : this.excludedPaths) {
excludedPath.populatePropertyBag();
}
this.jsonSerializable.set(Constants.Properties.EXCLUDED_PATHS, this.excludedPaths);
}
}
JsonSerializable getJsonSerializable() { return this.jsonSerializable; }
}