feat(executors): add redis instrumentation

Author: AndreyMarkinPPC

libs/executors/garf/executors/entrypoints/server.py

@@ -35,6 +35,7 @@
 from opentelemetry.instrumentation.celery import CeleryInstrumentor
 from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
 from opentelemetry.instrumentation.logging import LoggingInstrumentor
+from opentelemetry.instrumentation.redis import RedisInstrumentor
 from typing_extensions import Annotated
 
 OTEL_SERVICE_NAME = 'garf'
@@ -49,6 +50,7 @@
 logger.addHandler(initialize_logger())
 
 CeleryInstrumentor().instrument()
+RedisInstrumentor().instrument()
 app = fastapi.FastAPI(
   title='Garf API',
   version=garf.executors.__version__,
@@ -131,7 +133,9 @@ def execute_batch(
   return ApiExecutorResponse(results=results)
 
 
-@app.post('/api/execute:batch_task')
+@app.post(
+  '/api/execute:batch_task', status_code=fastapi.status.HTTP_202_ACCEPTED
+)
 async def execute_batch_task(
   request: tasks.ApiExecutorBatchRequest,
 ) -> dict[str, str]:
@@ -163,7 +167,9 @@ def execute_workflow(
   )
 
 
-@app.post('/api/execute:workflow_task')
+@app.post(
+  '/api/execute:workflow_task', status_code=fastapi.status.HTTP_202_ACCEPTED
+)
 async def execute_workflow_task(
   workflow_file: Optional[fastapi.UploadFile] = fastapi.File(None),
   enable_cache: bool = False,

libs/executors/garf/executors/entrypoints/tasks.py

@@ -28,6 +28,7 @@
 from garf.executors.workflows import workflow, workflow_runner
 from garf.io import reader
 from opentelemetry.instrumentation.celery import CeleryInstrumentor
+from opentelemetry.instrumentation.redis import RedisInstrumentor
 
 redis_url = os.getenv('CELERY_BROKER_URL', 'redis://localhost:6379/0')
 
@@ -88,6 +89,7 @@ def init_celery_telemetry(*args, **kwargs):
     loglevel='INFO', logger_type='local', name=otel_service_name
   )
   logger.addHandler(initialize_logger(otel_service_name))
+  RedisInstrumentor().instrument()
   CeleryInstrumentor().instrument()
 
 

libs/executors/pyproject.toml

@@ -64,6 +64,7 @@ server=[
   "grpcio-reflection",
   "celery[redis]",
   "opentelemetry-instrumentation-celery",
+  "opentelemetry-instrumentation-redis",
 
 ]
 tests = [