diff options
| author | yinjiangyi <[email protected]> | 2021-09-01 19:16:28 +0800 |
|---|---|---|
| committer | yinjiangyi <[email protected]> | 2021-09-01 19:16:28 +0800 |
| commit | 314cdbed28dff240669d4ac48e800b76d7d1386e (patch) | |
| tree | 99019c0e14f8691a14b0e199985cf3d0d9e00933 /src/main/java | |
| parent | a124ac05ea9b50648b7a5e7c7c3cc869f7149f07 (diff) | |
增加参数配置baseline保存level
Diffstat (limited to 'src/main/java')
| -rw-r--r-- | src/main/java/cn/mesalab/config/ApplicationConfig.java | 1 | ||||
| -rw-r--r-- | src/main/java/cn/mesalab/service/BaselineSingleThread.java | 4 |
2 files changed, 4 insertions, 1 deletions
diff --git a/src/main/java/cn/mesalab/config/ApplicationConfig.java b/src/main/java/cn/mesalab/config/ApplicationConfig.java index ffbfafb..6c28e90 100644 --- a/src/main/java/cn/mesalab/config/ApplicationConfig.java +++ b/src/main/java/cn/mesalab/config/ApplicationConfig.java @@ -22,6 +22,7 @@ public class ApplicationConfig { public static final Integer HISTORICAL_GRAD = ConfigUtils.getIntProperty("historical.grad"); public static final String TIME_FORMAT = ConfigUtils.getStringProperty("time.format"); public static final String BASELINE_METRIC_TYPE = ConfigUtils.getStringProperty("baseline.metric.type"); + public static final Integer BASELINE_SAVE_LEVEL = ConfigUtils.getIntProperty("baseline.save.level"); public static final String HBASE_BASELINE_GENERATION_TYPE_SUFFIX = ConfigUtils.getStringProperty("hbase.baseline.generation.type.suffix"); public static final String HBASE_BASELINE_ZERO_REPLACE_VALUE_SUFFIX = ConfigUtils.getStringProperty("hbase.baseline.zero.replace.value.suffix"); diff --git a/src/main/java/cn/mesalab/service/BaselineSingleThread.java b/src/main/java/cn/mesalab/service/BaselineSingleThread.java index 2d8098d..e48b3a3 100644 --- a/src/main/java/cn/mesalab/service/BaselineSingleThread.java +++ b/src/main/java/cn/mesalab/service/BaselineSingleThread.java @@ -29,6 +29,8 @@ import java.util.stream.Collectors; public class BaselineSingleThread extends Thread { private static final Logger LOG = LoggerFactory.getLogger(BaselineSingleThread.class); + private final int BASELINE_SAVE_LEVEL = ApplicationConfig.BASELINE_SAVE_LEVEL; + private final HbaseUtils hbaseUtils; private final Table hbaseTable; private final List<String> attackTypeList; @@ -100,7 +102,7 @@ public class BaselineSingleThread extends Thread { int[] ipBaseline = tuple._1; int generateType = tuple._2; int zeroReplaceValue = tuple._3; - if ((ipBaseline!= null ) && (ip.length()>0)){ + if ((BASELINE_SAVE_LEVEL >= generateType) && (ipBaseline!= null ) && (ip.length()>0)){ hbaseUtils.cachedInPut(putList, ip, ipBaseline, attackType, ApplicationConfig.BASELINE_METRIC_TYPE); hbaseUtils.cachedInPut(putList, ip, generateType, attackType, ApplicationConfig.BASELINE_METRIC_TYPE + "_" + ApplicationConfig.HBASE_BASELINE_GENERATION_TYPE_SUFFIX); |