Expose onboarding metrics on worker level

Author: mzbroch

netbox_onboarding/metrics.py

@@ -0,0 +1,18 @@
+"""Plugin additions to the NetBox navigation menu.
+
+(c) 2020 Network To Code
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+  http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+from prometheus_client import Counter
+
+onboardingtask_results_counter = Counter(
+    name="onboardingtask_results_total", documentation="Count of results for Onboarding Task", labelnames=("status",)
+)

netbox_onboarding/worker.py

@@ -15,19 +15,25 @@
 
 from django.core.exceptions import ValidationError
 from django_rq import job
+from prometheus_client import Summary
 
 from dcim.models import Device
 
 from .choices import OnboardingFailChoices
 from .choices import OnboardingStatusChoices
 from .exceptions import OnboardException
+from .metrics import onboardingtask_results_counter
 from .models import OnboardingDevice
 from .models import OnboardingTask
 from .onboard import OnboardingManager
 
 logger = logging.getLogger("rq.worker")
 
 
+REQUEST_TIME = Summary("onboardingtask_processing_seconds", "Time spent processing onboarding request")
+
+
+@REQUEST_TIME.time()
 @job("default")
 def onboard_device(task_id, credentials):  # pylint: disable=too-many-statements
     """Process a single OnboardingTask instance."""
@@ -98,4 +104,6 @@ def onboard_device(task_id, credentials):  # pylint: disable=too-many-statements
         ot.save()
         onboarding_status = False
 
+    onboardingtask_results_counter.labels(status=ot.status).inc()
+
     return dict(ok=onboarding_status)