add xarray dependency in titiler.application (developmentseed#1321)

* add xarray dependency in titiler.application

* use local titiler-xarray in dockerfile

* make sure to update titile-xarray version in application pyproject.toml

* update docs

* remove useless install

Author: vincentsarago

CHANGES.md

@@ -89,7 +89,7 @@ def __init__(
         permissions = permissions or []
         environment = environment or {}
 
-        # COG / STAC / MosaicJSON
+        # COG / STAC / MosaicJSON / Zarr endpoints
         lambda_function = aws_lambda.Function(
             self,
             f"{id}-lambda",
@@ -122,39 +122,6 @@ def __init__(
         )
         CfnOutput(self, "Endpoint", value=api.url)
 
-        # Xarray
-        xarray_lambda_function = aws_lambda.Function(
-            self,
-            f"{id}-xarray-lambda",
-            runtime=runtime,
-            code=aws_lambda.Code.from_docker_build(
-                path=os.path.abspath(code_dir),
-                file="lambda/Dockerfile.xarray",
-                platform="linux/amd64",
-                build_args={
-                    "PYTHON_VERSION": "3.14",
-                },
-            ),
-            handler="handler.handler",
-            memory_size=memory,
-            reserved_concurrent_executions=concurrent,
-            timeout=Duration.seconds(timeout),
-            environment=environment,
-            log_retention=logs.RetentionDays.ONE_WEEK,
-        )
-
-        for perm in permissions:
-            xarray_lambda_function.add_to_role_policy(perm)
-
-        xarray_api = apigw.HttpApi(
-            self,
-            f"{id}-xarray-endpoint",
-            default_integration=HttpLambdaIntegration(
-                f"{id}-xarray-integration", handler=xarray_lambda_function
-            ),
-        )
-        CfnOutput(self, "Xarray-Endpoint", value=xarray_api.url)
-
 
 app = App()
 settings = StackSettings()

README.md

@@ -8,7 +8,7 @@ WORKDIR /tmp
 RUN dnf install -y gcc-c++ && dnf clean all
 
 RUN python -m pip install pip -U
-RUN python -m pip install "titiler-application==2.0.0dev0" "mangum==0.19.0" -t /asset --no-binary pydantic
+RUN python -m pip install "titiler-application==2.0.0dev0" "mangum==0.19.0" "cftime" -t /asset --no-binary pydantic,xarray,numpy,pandas
 
 # Reduce package size and remove useless files
 RUN cd /asset && find . -type f -name '*.pyc' | while read f; do n=$(echo $f | sed 's/__pycache__\///' | sed 's/.cpython-[0-9]*//'); cp $f $n; done;

deployment/aws/app.py

@@ -28,6 +28,7 @@ services:
       # - TITILER_API_DISABLE_STAC=TRUE/FALSE
       # - TITILER_API_DISABLE_MOSAIC=TRUE/FALSE
       # - TITILER_API_DISABLE_COG=TRUE/FALSE
+      # - TITILER_API_DISABLE_ZARR=TRUE/FALSE
       # - TITILER_API_CORS_ORIGIN=url.io,url.xyz
       # - TITILER_API_CACHECONTROL=public, max-age=3600
       # - TITILER_API_DEBUG=TRUE/FALSE
@@ -82,16 +83,6 @@ services:
       - nginx-titiler
     command: ["nginx-debug", "-g", "daemon off;"]
 
-  titiler-xarray:
-    extends:
-      service: titiler
-    build:
-      context: .
-      dockerfile: dockerfiles/Dockerfile.xarray
-    ports:
-      - "8082:8082"
-    command: ["uvicorn", "titiler.xarray.main:app", "--host", "0.0.0.0", "--port", "8082", "--workers", "1"]
-
   otel-collector:
     image: otel/opentelemetry-collector-contrib:latest
     command: ["--config=/etc/otel-collector-config.yaml"]

deployment/aws/lambda/Dockerfile

@@ -17,7 +17,7 @@ RUN python -m pip install -U pip
 RUN python -m pip install uvicorn uvicorn-worker gunicorn
 
 COPY src/titiler/ /tmp/titiler/
-RUN python -m pip install /tmp/titiler/core /tmp/titiler/extensions["cogeo,stac"] /tmp/titiler/mosaic /tmp/titiler/application --no-cache-dir --upgrade
+RUN python -m pip install /tmp/titiler/core /tmp/titiler/extensions["cogeo,stac"] /tmp/titiler/mosaic["mosaicjson"] /tmp/titiler/xarray /tmp/titiler/application --no-cache-dir --upgrade
 RUN rm -rf /tmp/titiler
 
 ###################################################

deployment/aws/lambda/Dockerfile.xarray

@@ -249,6 +249,11 @@ filename = "src/titiler/application/pyproject.toml"
 search = 'titiler-core[telemetry]=={current_version}'
 replace = 'titiler-core[telemetry]=={new_version}'
 
+[[tool.bumpversion.files]]
+filename = "src/titiler/application/pyproject.toml"
+search = 'titiler-xarray=={current_version}'
+replace = 'titiler-xarray=={new_version}'
+
 [[tool.bumpversion.files]]
 filename = "src/titiler/application/pyproject.toml"
 search = 'titiler-extensions[cogeo,stac]=={current_version}'