Skip to content

Commit f15a954

Browse files
add data platform to main api entry point (#139)
* add data platform to main api entry point * fix
1 parent ee55bea commit f15a954

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

src/quartz_api/cmd/main.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,15 +34,17 @@
3434
from typing import Any
3535

3636
import uvicorn
37+
from dp_sdk.ocf import dp
3738
from fastapi import FastAPI, status
3839
from fastapi.middleware.cors import CORSMiddleware
3940
from fastapi.openapi.utils import get_openapi
41+
from grpclib.client import Channel
4042
from pydantic import BaseModel
4143
from pyhocon import ConfigFactory
4244
from starlette.responses import FileResponse
4345
from starlette.staticfiles import StaticFiles
4446

45-
from quartz_api.internal.backends import DummyClient, QuartzClient
47+
from quartz_api.internal.backends import DataPlatformClient, DummyClient, QuartzClient
4648
from quartz_api.internal.middleware import audit, auth
4749
from quartz_api.internal.models import DatabaseInterface, get_db_client
4850
from quartz_api.internal.service import regions, sites
@@ -156,6 +158,14 @@ def redoc_html() -> FileResponse:
156158
case "dummydb":
157159
db_instance = DummyClient()
158160
log.warning("disabled backend. NOT recommended for production")
161+
case "dataplatform":
162+
163+
channel = Channel(
164+
host=conf.get_string("backend.dataplatform.host"),
165+
port=conf.get_int("backend.dataplatform.port"),
166+
)
167+
client = dp.DataPlatformDataServiceStub(channel=channel)
168+
db_instance = DataPlatformClient.from_dp(dp_client=client)
159169
case _:
160170
raise ValueError(
161171
"Unknown backend. "

src/quartz_api/cmd/server.conf

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,10 @@ backend {
2222
database_url = ${?DB_URL}
2323
}
2424
dummydb {}
25+
dataplatform {
26+
host = ${?DATA_PLATFORM_HOST}
27+
port = ${?DATA_PLATFORM_PORT}
28+
}
2529
}
2630

2731
// Auth0 configuration

0 commit comments

Comments
 (0)