feat: add error threshold as config (#48)

Co-authored-by: AlucardleVash <[email protected]>

Author: AlucardleVash

README.md

@@ -118,7 +118,7 @@ Let’s explain the plugin fields:
 * `influxDBToken` - the influxdb bucket token, the default value should be updated, copy it from InfluxDB site.
 * `influxDBFlushInterval` - its interval to send data to InfluxDB, the default value is 4000 (4 seconds).
 * `influxDBMaxBatchSize` - the max size of the batch with metrics, the default 2000 (2000 items of JMeter results).
-
+* `influxDBThresholdError` - the error threshold before stopping the import, the default value is 5. (see [Important notes](#important-notes) for more detail.)
 
  ![](img/influx3.png)
 

build.gradle

@@ -120,6 +120,11 @@ publishing {
                         name = 'Mike Derevianko'
                         email = '[email protected]'
                     }
+                    developer {
+                        id = 'AlucardleVash'
+                        name = 'Pierre Brun'
+                        email = '[email protected]'
+                    }
                 }
                 scm {
                     connection = 'scm:git:git:github.com/mderevyankoaqa/jmeter-influxdb2-listener-plugin.git'

gradlew

@@ -157,6 +157,7 @@ public Arguments getDefaultParameters() {
         arguments.addArgument(InfluxDBConfig.KEY_INFLUX_DB_BUCKET, InfluxDBConfig.DEFAULT_BUCKET);
         arguments.addArgument(InfluxDBConfig.KEY_INFLUX_DB_FLUSH_INTERVAL, String.valueOf(InfluxDBConfig.DEFAULT_INFLUX_DB_FLUSH_INTERVAL));
         arguments.addArgument(InfluxDBConfig.KEY_INFLUX_DB_MAX_BATCH_SIZE, String.valueOf(InfluxDBConfig.DEFAULT_INFLUX_DB_MAX_BATCH_SIZE));
+        arguments.addArgument(InfluxDBConfig.KEY_INFLUX_DB_THRESHOLD_ERROR, Integer.toString(InfluxDBConfig.DEFAULT_THRESHOLD_ERROR));
         arguments.addArgument(KEY_SAMPLERS_LIST, ".*");
         arguments.addArgument(KEY_USE_REGEX_FOR_SAMPLER_LIST, "true");
         arguments.addArgument(KEY_RECORD_SUB_SAMPLES, "true");

src/main/java/io/github/mderevyankoaqa/influxdb2/visualizer/InfluxDatabaseBackendListenerClient.java

@@ -27,6 +27,11 @@ public class InfluxDBConfig {
      */
     public static final int DEFAULT_PORT = 8086;
 
+    /**
+     * Default threshold error.
+     */
+    public static final int DEFAULT_THRESHOLD_ERROR = 5;
+
     /**
      * Default token.
      */
@@ -88,6 +93,11 @@ public class InfluxDBConfig {
      */
     public static final String KEY_INFLUX_DB_FLUSH_INTERVAL = "influxDBFlushInterval";
 
+    /**
+     * Config key error threshold.
+     */
+    public static final String KEY_INFLUX_DB_THRESHOLD_ERROR = "influxDBThresholdError";
+
     /**
      * InfluxDB Host.
      */
@@ -128,6 +138,11 @@ public class InfluxDBConfig {
      */
     private int influxdbFlushInterval;
 
+    /**
+     * Default protection to avoid OOM error.
+     */
+    private int influxdbThresholdError;
+
     /**
      * Creates the new instance of {@link InfluxDBConfig}
      *
@@ -164,6 +179,9 @@ public InfluxDBConfig(BackendListenerContext context) {
         Arguments.checkNotNegativeNumber(influxdbFlushInterval, KEY_INFLUX_DB_FLUSH_INTERVAL);
         this.setInfluxdbFlushInterval(influxdbFlushInterval);
 
+        int influxdbThresholdError = context.getIntParameter(KEY_INFLUX_DB_THRESHOLD_ERROR, InfluxDBConfig.DEFAULT_THRESHOLD_ERROR);
+        Arguments.checkNotNegativeNumber(influxdbThresholdError, KEY_INFLUX_DB_THRESHOLD_ERROR);
+        this.setInfluxdbThresholdError(influxdbThresholdError);
     }
 
     /**
@@ -309,4 +327,22 @@ public int getInfluxdbFlushInterval() {
     public void setInfluxdbFlushInterval(int influxdbFlushInterval) {
         this.influxdbFlushInterval = influxdbFlushInterval;
     }
+
+    /**
+     * Sets error threshold.
+     *
+     * @param influxdbThresholdError is the threshold of error to stop sending data to influx db to avoid OOM error.
+     */
+    public void setInfluxdbThresholdError(int influxdbThresholdError) {
+        this.influxdbThresholdError = influxdbThresholdError;
+    }
+
+    /**
+     * Gets threshold error to stop InfluxDB import.
+     *
+     * @return Threshold of error represented in the int.
+     */
+    public int getInfluxdbThresholdError() {
+        return influxdbThresholdError;
+    }
 }

src/main/java/io/github/mderevyankoaqa/influxdb2/visualizer/config/InfluxDBConfig.java

@@ -100,13 +100,13 @@ public synchronized void close() {
      */
     public synchronized void importData() {
 
-        if (this.errorsAmount.size() >= 5)
+        if (this.errorsAmount.size() >= this.influxDBConfig.getInfluxdbThresholdError())
         {
             this.points.clear();
             this.LOGGER.warn("Importing of the results to Influx DB is skipping since 5 errors, has occurred!");
         }
 
-        if (this.points.size() != 0 && this.errorsAmount.size() <= 5) {
+        if (this.points.size() != 0 && this.errorsAmount.size() <= this.influxDBConfig.getInfluxdbThresholdError()) {
             try {
 
                 long start = System.currentTimeMillis();