[feature] Implement Iperf3 check #385

openwisp_monitoring/check/apps.py

@@ -32,3 +32,11 @@ def _connect_signals(self):
                 sender=load_model('config', 'Device'),
                 dispatch_uid='auto_config_check',
             )
+        if app_settings.AUTO_IPERF:
+            from .base.models import auto_iperf_check_receiver
+
+            post_save.connect(
+                auto_iperf_check_receiver,
+                sender=load_model('config', 'Device'),
+                dispatch_uid='auto_iperf_check',
+            )

openwisp_monitoring/check/base/models.py

@@ -9,7 +9,11 @@
 from jsonfield import JSONField
 
 from openwisp_monitoring.check import settings as app_settings
-from openwisp_monitoring.check.tasks import auto_create_config_check, auto_create_ping
+from openwisp_monitoring.check.tasks import (
+    auto_create_config_check,
+    auto_create_iperf_check,
+    auto_create_ping,
+)
 from openwisp_utils.base import TimeStampedEditableModel
 
 from ...utils import transaction_on_commit
@@ -116,3 +120,21 @@ def auto_config_check_receiver(sender, instance, created, **kwargs):
             object_id=str(instance.pk),
         )
     )
+
+
+def auto_iperf_check_receiver(sender, instance, created, **kwargs):
+    """
+    Implements OPENWISP_MONITORING_AUTO_IPERF
+    The creation step is executed in the background
+    """
+    # we need to skip this otherwise this task will be executed
+    # every time the configuration is requested via checksum
+    if not created:
+        return
+    transaction_on_commit(
+        lambda: auto_create_iperf_check.delay(
+            model=sender.__name__.lower(),
+            app_label=sender._meta.app_label,
+            object_id=str(instance.pk),
+        )
+    )

openwisp_monitoring/check/classes/__init__.py

@@ -1,2 +1,3 @@
 from .config_applied import ConfigApplied  # noqa
+from .iperf import Iperf  # noqa
 from .ping import Ping  # noqa