increase timeout. use threading

Author: HansKallekleiv

backend.Dockerfile

@@ -23,4 +23,4 @@ RUN pip install poetry \
     && pip install -r requirements.txt
 
 
-CMD exec uvicorn --proxy-headers  --host=0.0.0.0 $UVICORN_ENTRYPOINT 
+CMD exec uvicorn --proxy-headers --timeout-keep-alive 300 --host=0.0.0.0 $UVICORN_ENTRYPOINT 

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

@@ -3,7 +3,9 @@
 
 from typing import List, Tuple
 import logging
+from concurrent.futures import ThreadPoolExecutor
 from aiocache import Cache
+
 import numpy as np
 from fastapi.responses import ORJSONResponse
 
@@ -14,7 +16,7 @@
 from src.backend.auth.auth_helper import AuthenticatedUser, AuthHelper
 
 from src.services.sumo_access.surface_access import SurfaceAccess
-
+import asyncio
 from src.backend.primary.routers.surface import schemas
 from .test_async import async_get_cached_surf
 from .test_go import go_get_surface_blobs
@@ -97,11 +99,6 @@ async def well_intersection_reals_from_user_session(
         # await cache.set(f"{authenticated_user._user_id}-{uuid}", surface)
         surfaces.append(surface)
 
-    # print(surfaces)
-    # return []
-
-    intersections = []
-
     fence_arr = np.array(
         [
             polyline.x_points,
@@ -110,13 +107,21 @@ async def well_intersection_reals_from_user_session(
             polyline.cum_length,
         ]
     ).T
-    for surf in surfaces:
+
+    def worker(surf):
         line = surf.get_randomline(fence_arr)
         intersection = schemas.SurfaceIntersectionPoints(
             name=f"{surf.name}",
             cum_length=line[:, 0].tolist(),
             z_array=line[:, 1].tolist(),
         )
-        intersections.append(intersection)
+        return intersection
+
+    loop = asyncio.get_running_loop()
+
+    with ThreadPoolExecutor() as executor:
+        tasks = [loop.run_in_executor(executor, worker, surf) for surf in surfaces]
+        intersections = await asyncio.gather(*tasks)
 
-    return ORJSONResponse([section.dict() for section in intersections])
+    result = [intersection.dict() for intersection in intersections]
+    return ORJSONResponse(result)

backend/src/backend/user_session/routers/surface/test_go.py

@@ -30,11 +30,11 @@ def go_get_surface_blobs(sumo_token: str, case_uuid: str, object_ids: list[str])
     res = GetZippedBlobs(json.dumps(new_request).encode("utf-8"))
     res_string = ctypes.string_at(res).decode("ascii")
     data_map_b64 = json.loads(res_string)
-
+    print("Done downloading surfaces", flush=True)
     surfaces = {}
     for object_id, b64_blob in data_map_b64.items():
         bytestr = base64.b64decode(b64_blob)
         xtgeo_surface = xtgeo.surface_from_file(BytesIO(bytestr), fformat="irap_binary")
         surfaces[object_id] = xtgeo_surface
-
+    print("Done decoding surfaces", flush=True)
     return surfaces