Log events for function calls

Note that I would've preferred the function ID to go into the `obj_id`
field, but function IDs are strings, and the field is numeric.

changelog.d/20240903_155336_roman_function_events.md

@@ -0,0 +1,4 @@
+### Added
+
+- Added analytics events for function calls
+  (<https://github.com/cvat-ai/cvat/pull/8395>)

cvat/apps/events/event.py

@@ -27,6 +27,7 @@ class EventScopes:
         "annotations": ["create", "update", "delete"],
         "label": ["create", "update", "delete"],
         "dataset": ["export", "import"],
+        "function": ["call"],
     }
 
     @classmethod

cvat/apps/events/handlers.py

@@ -510,6 +510,26 @@ def handle_dataset_import(
 ) -> None:
     handle_dataset_io(instance, "import", format_name=format_name, cloud_storage_id=cloud_storage_id)
 
+def handle_function_call(
+    function_id: str,
+    target: Union[Task, Job],
+    **payload_fields,
+) -> None:
+    record_server_event(
+        scope=event_scope("call", "function"),
+        request_id=request_id(),
+        project_id=project_id(target),
+        task_id=task_id(target),
+        job_id=job_id(target),
+        user_id=user_id(),
+        user_name=user_name(),
+        user_email=user_email(),
+        payload={
+            "function": {"id": function_id},
+            **payload_fields,
+        },
+    )
+
 def handle_rq_exception(rq_job, exc_type, exc_value, tb):
     oid = rq_job.meta.get(RQJobMetaField.ORG_ID, None)
     oslug = rq_job.meta.get(RQJobMetaField.ORG_SLUG, None)

cvat/apps/lambda_manager/views.py

@@ -20,6 +20,7 @@
 import numpy as np
 import requests
 import rq
+from cvat.apps.events.handlers import handle_function_call
 from cvat.apps.lambda_manager.signals import interactive_function_call_signal
 from django.conf import settings
 from django.core.exceptions import ObjectDoesNotExist, ValidationError
@@ -1083,14 +1084,21 @@ def call(self, request, func_id):
         gateway = LambdaGateway()
         lambda_func = gateway.get(func_id)
 
-        return lambda_func.invoke(
+        response = lambda_func.invoke(
             db_task,
             request.data, # TODO: better to add validation via serializer for these data
             db_job=job,
             is_interactive=True,
             request=request
         )
 
+        handle_function_call(func_id, db_task,
+            category="interactive",
+            parameters={"frame": request.data["frame"]},
+        )
+
+        return response
+
 @extend_schema(tags=['lambda'])
 @extend_schema_view(
     retrieve=extend_schema(
@@ -1182,6 +1190,8 @@ def create(self, request):
         rq_job = queue.enqueue(lambda_func, threshold, task, quality,
             mapping, cleanup, conv_mask_to_poly, max_distance, request, job=job)
 
+        handle_function_call(function, job or task, category="batch")
+
         response_serializer = FunctionCallSerializer(rq_job.to_dict())
         return response_serializer.data
 

site/content/en/docs/administration/advanced/analytics.md

@@ -129,6 +129,8 @@ Server events:
 
 - `export:dataset`, `import:dataset`
 
+- `call:function`
+
 Client events:
 
 - `load:cvat`