metron git commit: Merge branch 'master' into feature/METRON-1416-upgrade-solr
Repository: metron Updated Branches: refs/heads/feature/METRON-1416-upgrade-solr 476856192 -> 4c4452e59 Merge branch 'master' into feature/METRON-1416-upgrade-solr Project: http://git-wip-us.apache.org/repos/asf/metron/repo Commit: http://git-wip-us.apache.org/repos/asf/metron/commit/4c4452e5 Tree: http://git-wip-us.apache.org/repos/asf/metron/tree/4c4452e5 Diff: http://git-wip-us.apache.org/repos/asf/metron/diff/4c4452e5 Branch: refs/heads/feature/METRON-1416-upgrade-solr Commit: 4c4452e59c93306a049e6879f7b6fed2ee86c74e Parents: 4768561 Author: cstella Authored: Tue Jun 12 21:40:01 2018 -0400 Committer: cstella Committed: Tue Jun 12 21:40:01 2018 -0400 -- .../dao/ElasticsearchSearchDao.java | 30 +-- .../ElasticsearchMetaAlertIntegrationTest.java | 223 ++- 2 files changed, 134 insertions(+), 119 deletions(-) -- http://git-wip-us.apache.org/repos/asf/metron/blob/4c4452e5/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchSearchDao.java -- diff --git a/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchSearchDao.java b/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchSearchDao.java index 5725534..5cd0a4d 100644 --- a/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchSearchDao.java +++ b/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchSearchDao.java @@ -443,22 +443,24 @@ public class ElasticsearchSearchDao implements SearchDao { private List getGroupResults(GroupRequest groupRequest, int index, Aggregations aggregations, Map commonColumnMetadata) { List groups = groupRequest.getGroups(); String field = groups.get(index).getField(); -Terms terms = aggregations.get(getGroupByAggregationName(field)); List searchResultGroups = new ArrayList<>(); -for(Bucket bucket: terms.getBuckets()) { - GroupResult groupResult = new GroupResult(); - groupResult.setKey(formatKey(bucket.getKey(), commonColumnMetadata.get(field))); - groupResult.setTotal(bucket.getDocCount()); - Optional scoreField = groupRequest.getScoreField(); - if (scoreField.isPresent()) { -Sum score = bucket.getAggregations().get(getSumAggregationName(scoreField.get())); -groupResult.setScore(score.getValue()); - } - if (index < groups.size() - 1) { -groupResult.setGroupedBy(groups.get(index + 1).getField()); -groupResult.setGroupResults(getGroupResults(groupRequest, index + 1, bucket.getAggregations(), commonColumnMetadata)); +if(aggregations != null) { + Terms terms = aggregations.get(getGroupByAggregationName(field)); + for (Bucket bucket : terms.getBuckets()) { +GroupResult groupResult = new GroupResult(); +groupResult.setKey(formatKey(bucket.getKey(), commonColumnMetadata.get(field))); +groupResult.setTotal(bucket.getDocCount()); +Optional scoreField = groupRequest.getScoreField(); +if (scoreField.isPresent()) { + Sum score = bucket.getAggregations().get(getSumAggregationName(scoreField.get())); + groupResult.setScore(score.getValue()); +} +if (index < groups.size() - 1) { + groupResult.setGroupedBy(groups.get(index + 1).getField()); + groupResult.setGroupResults(getGroupResults(groupRequest, index + 1, bucket.getAggregations(), commonColumnMetadata)); +} +searchResultGroups.add(groupResult); } - searchResultGroups.add(groupResult); } return searchResultGroups; } http://git-wip-us.apache.org/repos/asf/metron/blob/4c4452e5/metron-platform/metron-elasticsearch/src/test/java/org/apache/metron/elasticsearch/integration/ElasticsearchMetaAlertIntegrationTest.java -- diff --git a/metron-platform/metron-elasticsearch/src/test/java/org/apache/metron/elasticsearch/integration/ElasticsearchMetaAlertIntegrationTest.java b/metron-platform/metron-elasticsearch/src/test/java/org/apache/metron/elasticsearch/integration/ElasticsearchMetaAlertIntegrationTest.java index 99f0490..224ec56 100644 --- a/metron-platform/metron-elasticsearch/src/test/java/org/apache/metron/elasticsearch/integration/ElasticsearchMetaAlertIntegrationTest.java +++ b/metron-platform/metron-elasticsearch/src/test/java/org/apache/metron/elasticsearch/integration/ElasticsearchMetaAlertIntegrationTest.java @@ -1,20 +1,21 @@ + /* - * Licensed to the Apache Software Foundation (ASF) under one - * or more contributor license agreements. See the NOTICE file - * distributed with this work for
[1/2] metron git commit: Merge branch 'master' into feature/METRON-1416-upgrade-solr
Repository: metron Updated Branches: refs/heads/feature/METRON-1416-upgrade-solr 0717cfc25 -> 476856192 Merge branch 'master' into feature/METRON-1416-upgrade-solr Project: http://git-wip-us.apache.org/repos/asf/metron/repo Commit: http://git-wip-us.apache.org/repos/asf/metron/commit/e7233b0f Tree: http://git-wip-us.apache.org/repos/asf/metron/tree/e7233b0f Diff: http://git-wip-us.apache.org/repos/asf/metron/diff/e7233b0f Branch: refs/heads/feature/METRON-1416-upgrade-solr Commit: e7233b0f82f372e6b1339de56a115fa7e4581f07 Parents: 4170887 Author: cstella Authored: Mon Jun 11 21:41:06 2018 -0400 Committer: cstella Committed: Tue Jun 12 09:46:46 2018 -0400 -- .../org/apache/metron/common/Constants.java | 1 + .../elasticsearch/dao/ElasticsearchDao.java | 4 ++ .../dao/ElasticsearchMetaAlertDao.java | 30 +- .../dao/ElasticsearchMetaAlertUpdateDao.java| 4 +- .../indexing/dao/metaalert/MetaAlertConfig.java | 43 .../AbstractLuceneMetaAlertUpdateDao.java | 3 ++ .../dao/metaalert/MetaAlertIntegrationTest.java | 2 +- .../AbstractLuceneMetaAlertUpdateDaoTest.java | 28 + .../metron/solr/dao/SolrMetaAlertDao.java | 29 + .../metron/solr/dao/SolrMetaAlertSearchDao.java | 15 --- .../metron/solr/dao/SolrMetaAlertUpdateDao.java | 2 +- .../SolrMetaAlertIntegrationTest.java | 28 + 12 files changed, 129 insertions(+), 60 deletions(-) -- http://git-wip-us.apache.org/repos/asf/metron/blob/e7233b0f/metron-platform/metron-common/src/main/java/org/apache/metron/common/Constants.java -- diff --git a/metron-platform/metron-common/src/main/java/org/apache/metron/common/Constants.java b/metron-platform/metron-common/src/main/java/org/apache/metron/common/Constants.java index f74660c..4a8bea2 100644 --- a/metron-platform/metron-common/src/main/java/org/apache/metron/common/Constants.java +++ b/metron-platform/metron-common/src/main/java/org/apache/metron/common/Constants.java @@ -28,6 +28,7 @@ public class Constants { public static final long DEFAULT_CONFIGURED_BOLT_TIMEOUT = 5000; public static final String SENSOR_TYPE = "source.type"; public static final String SENSOR_TYPE_FIELD_PROPERTY = "source.type.field"; + public static final String THREAT_SCORE_FIELD_PROPERTY = "threat.triage.score.field"; public static final String ENRICHMENT_TOPIC = "enrichments"; public static final String INDEXING_TOPIC = "indexing"; public static final String ERROR_STREAM = "error"; http://git-wip-us.apache.org/repos/asf/metron/blob/e7233b0f/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchDao.java -- diff --git a/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchDao.java b/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchDao.java index eae0a39..3eb86ce 100644 --- a/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchDao.java +++ b/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchDao.java @@ -85,6 +85,10 @@ public class ElasticsearchDao implements IndexDao { //uninitialized. } + public AccessConfig getAccessConfig() { +return accessConfig; + } + @Override public synchronized void init(AccessConfig config) { if (this.client == null) { http://git-wip-us.apache.org/repos/asf/metron/blob/e7233b0f/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchMetaAlertDao.java -- diff --git a/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchMetaAlertDao.java b/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchMetaAlertDao.java index f73a640..649077e 100644 --- a/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchMetaAlertDao.java +++ b/metron-platform/metron-elasticsearch/src/main/java/org/apache/metron/elasticsearch/dao/ElasticsearchMetaAlertDao.java @@ -68,6 +68,7 @@ import java.util.Map.Entry; import java.util.Optional; import java.util.Set; import java.util.UUID; +import java.util.function.Supplier; import java.util.stream.Collectors; import static org.apache.metron.common.Constants.GUID; @@ -83,10 +84,8 @@ public class ElasticsearchMetaAlertDao implements MetaAlertDao { public static final String THREAT_TRIAGE_FIELD = MetaAlertConstants.THREAT_FIELD_DEFAULT