temp

Author: Yuri Zmytrakov

stac_fastapi/core/stac_fastapi/core/core.py

@@ -329,20 +329,6 @@ async def all_collections(
             if parsed_sort:
                 sort = parsed_sort
 
-        current_url = str(request.url)
-        redis_enable = get_bool_env("REDIS_ENABLE", default=False)
-
-        redis = None
-        if redis_enable:
-            try:
-                redis = await connect_redis()
-                logger.info("Redis connection established successfully")
-            except Exception as e:
-                redis = None
-                logger.warning(
-                    f"Redis connection failed, continuing without Redis: {e}"
-                )
-
         # Convert q to a list if it's a string
         q_list = None
         if q is not None:
@@ -441,21 +427,71 @@ async def all_collections(
             },
         ]
 
-        if redis_enable and redis:
-            if next_token:
-                await save_self_link(redis, next_token, current_url)
+        async def handle_pagination_links(
+                current_url: str,
+                token: str,
+                next_token: str, 
+                links: list
+            ) -> None:
 
-            prev_link = await get_prev_link(redis, token)
-            if prev_link:
-                links.insert(
-                    0,
-                    {
-                        "rel": "prev",
-                        "type": "application/json",
-                        "method": "GET",
-                        "href": prev_link,
-                    },
-                )
+            redis_enable = get_bool_env("REDIS_ENABLE", default=False)
+
+            redis = None
+            if redis_enable:
+                try:
+                    redis = await connect_redis()
+                    logger.info("Redis connection established successfully")
+                except Exception as e:
+                    redis = None
+                    logger.warning(f"Redis connection failed, continuing without Redis: {e}")
+
+            if redis_enable and redis:
+                if next_token:
+                    await save_self_link(redis, next_token, current_url)
+                
+                prev_link = await get_prev_link(redis, token)
+                if prev_link:
+                    links.insert(
+                        0,
+                        {
+                            "rel": "prev",
+                            "type": "application/json",
+                            "method": "GET", 
+                            "href": prev_link,
+                        },
+                    )
+
+        current_url = str(request.url)
+
+        handle_pagination_links(current_url, token, next_token, links)
+
+        # redis_enable = get_bool_env("REDIS_ENABLE", default=False)
+
+        # redis = None
+        # if redis_enable:
+        #     try:
+        #         redis = await connect_redis()
+        #         logger.info("Redis connection established successfully")
+        #     except Exception as e:
+        #         redis = None
+        #         logger.warning(
+        #             f"Redis connection failed, continuing without Redis: {e}"
+        #         )
+        # if redis_enable and redis:
+        #     if next_token:
+        #         await save_self_link(redis, next_token, current_url)
+
+        #     prev_link = await get_prev_link(redis, token)
+        #     if prev_link:
+        #         links.insert(
+        #             0,
+        #             {
+        #                 "rel": "prev",
+        #                 "type": "application/json",
+        #                 "method": "GET",
+        #                 "href": prev_link,
+        #             },
+        #         )
 
         if next_token:
             next_link = PagingLinks(next=next_token, request=request).link_next()