sync

backend/src/backend/primary/routers/surface/router.py

@@ -1,6 +1,6 @@
 import logging
 from typing import List, Union, Optional
-
+import asyncio
 from fastapi import APIRouter, Depends, HTTPException, Query, Response
 
 from src.services.sumo_access.surface_access import SurfaceAccess
@@ -184,6 +184,23 @@ async def get_realization_surface_data_as_png(
     return surf_data_response
 
 
+class AsyncRange:
+    def __init__(self, start, end):
+        self.current = start
+        self.end = end
+
+    def __aiter__(self):
+        return self
+
+    async def __anext__(self):
+        if self.current < self.end:
+            number = self.current
+            self.current += 1
+            return number
+        else:
+            raise StopAsyncIteration
+
+
 @router.get("/statistical_surface_data_as_png/")
 async def get_statistical_surface_data_as_png(
     response: Response,
@@ -210,13 +227,21 @@ async def get_statistical_surface_data_as_png(
     )
     if service_stat_func_to_compute is None:
         raise HTTPException(status_code=404, detail="Invalid statistic requested")
-
-    xtgeo_surf = await access.get_statistical_surface_data_async(
-        statistic_function=service_stat_func_to_compute,
-        name=name,
-        attribute=attribute,
-        time_or_interval_str=time_or_interval,
-    )
+    xtg_arr = []
+    print("before loop", flush=True)
+    for i in range(0, 100):
+        print(i, flush=True)
+        xtg_arr.append(
+            access.get_statistical_surface_data_async(
+                statistic_function=service_stat_func_to_compute,
+                name=name,
+                attribute=attribute,
+                time_or_interval_str=time_or_interval,
+            )
+        )
+    print("before await", flush=True)
+    await asyncio.gather(*xtg_arr)
+    print("after await", flush=True)
     perf_metrics.record_lap("sumo-calc")
 
     if not xtgeo_surf: