Skip to content

Commit

Permalink
resolve code review
Browse files Browse the repository at this point in the history
Signed-off-by: yuye-aws <yuyezhu@amazon.com>
  • Loading branch information
yuye-aws committed Apr 28, 2024
1 parent 51adb22 commit d155a1b
Showing 1 changed file with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
import java.util.List;
import java.util.Objects;

import org.apache.commons.lang3.math.NumberUtils;
import org.opensearch.cluster.metadata.IndexMetadata;
import org.opensearch.env.Environment;
import org.opensearch.cluster.service.ClusterService;
Expand Down Expand Up @@ -144,18 +145,19 @@ private boolean isListOfString(final Object value) {
}

private int getMaxTokenCount(final Map<String, Object> sourceAndMetadataMap) {
int defaultMaxTokenCount = IndexSettings.MAX_TOKEN_COUNT_SETTING.get(environment.settings());
String indexName = sourceAndMetadataMap.get(IndexFieldMapper.NAME).toString();
IndexMetadata indexMetadata = clusterService.state().metadata().index(indexName);
if (Objects.isNull(indexMetadata)) {
return IndexSettings.MAX_TOKEN_COUNT_SETTING.get(environment.settings());
return defaultMaxTokenCount;
}
// if the index is specified in the metadata, read maxTokenCount from the index setting
String maxTokenCountString = indexMetadata.getSettings().get("index.analyze.max_token_count");
// if maxTokenCount is not specified in the index, return the default setting
if (Objects.isNull(maxTokenCountString)) {
return IndexSettings.MAX_TOKEN_COUNT_SETTING.get(environment.settings());
return defaultMaxTokenCount;
}
return Integer.parseInt(maxTokenCountString);
return NumberUtils.toInt(maxTokenCountString, defaultMaxTokenCount);
}

/**
Expand Down

0 comments on commit d155a1b

Please sign in to comment.