honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[ML] Minor refactoring to improve code readability 

Original commit: elastic/x-pack-elasticsearch@ef2d5fcab6
This commit is contained in:
Dimitrios Athanasiou 2017-03-24 10:51:25 +00:00
parent c85498462c
commit 28d6b505ed
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
plugin/src/main/java/org/elasticsearch/xpack/ml/MachineLearningFeatureSet.java
View File

@ -36,11 +36,15 @@ import java.io.IOException;
import java.util.Collections; import java.util.Collections;
import java.util.HashMap; import java.util.HashMap;
import java.util.LinkedHashMap; import java.util.LinkedHashMap;
import java.util.List;
import java.util.Locale; import java.util.Locale;
import java.util.Map; import java.util.Map;
import java.util.Objects; import java.util.Objects;
import java.util.concurrent.TimeoutException; import java.util.concurrent.TimeoutException;
import static org.elasticsearch.xpack.ml.action.GetDatafeedsStatsAction.Response.DatafeedStats;
import static org.elasticsearch.xpack.ml.action.GetJobsStatsAction.Response.JobStats;
public class MachineLearningFeatureSet implements XPackFeatureSet { public class MachineLearningFeatureSet implements XPackFeatureSet {
private final boolean enabled; private final boolean enabled;
@ -214,8 +218,8 @@ public class MachineLearningFeatureSet implements XPackFeatureSet {
Map<JobState, StatsAccumulator> modelSizeStatsByState = new HashMap<>(); Map<JobState, StatsAccumulator> modelSizeStatsByState = new HashMap<>();
Map<String, Job> jobs = mlMetadata.getJobs(); Map<String, Job> jobs = mlMetadata.getJobs();
for (GetJobsStatsAction.Response.JobStats jobStats List<JobStats> jobsStats = response.getResponse().results();
: response.getResponse().results()) { for (JobStats jobStats : jobsStats) {
ModelSizeStats modelSizeStats = jobStats.getModelSizeStats(); ModelSizeStats modelSizeStats = jobStats.getModelSizeStats();
int detectorsCount = jobs.get(jobStats.getJobId()).getAnalysisConfig() int detectorsCount = jobs.get(jobStats.getJobId()).getAnalysisConfig()
.getDetectors().size(); .getDetectors().size();
@ -255,8 +259,8 @@ public class MachineLearningFeatureSet implements XPackFeatureSet {
private void addDatafeedsUsage(GetDatafeedsStatsAction.Response response) { private void addDatafeedsUsage(GetDatafeedsStatsAction.Response response) {
Map<DatafeedState, Counter> datafeedCountByState = new HashMap<>(); Map<DatafeedState, Counter> datafeedCountByState = new HashMap<>();
for (GetDatafeedsStatsAction.Response.DatafeedStats datafeedStats List<DatafeedStats> datafeedsStats = response.getResponse().results();
: response.getResponse().results()) { for (DatafeedStats datafeedStats : datafeedsStats) {
datafeedCountByState.computeIfAbsent(datafeedStats.getDatafeedState(), datafeedCountByState.computeIfAbsent(datafeedStats.getDatafeedState(),
ds -> Counter.newCounter()).addAndGet(1); ds -> Counter.newCounter()).addAndGet(1);
} }