SIENTIAPDE-1110

Update Druid configuration in values.yaml to change port from 8081 to 8888. Refactor pydruid.py to use druid_engine for SQLAlchemy connections, enhancing clarity and consistency in data loading operations.
This commit is contained in:
vitor-aignosi
2025-07-02 10:35:53 -03:00
parent 6caec63977
commit 208491498f
2 changed files with 4 additions and 4 deletions

View File

@@ -19,7 +19,7 @@ class Druid(BaseActivity):
self.host = host self.host = host
self.port = port self.port = port
self.engine = create_engine( self.druid_engine = create_engine(
f'druid://{self.host}:{self.port}/druid/v2/sql/') f'druid://{self.host}:{self.port}/druid/v2/sql/')
logger.info( logger.info(
f"Druid client initialized with host: {self.host}, port: {self.port}") f"Druid client initialized with host: {self.host}, port: {self.port}")
@@ -51,10 +51,10 @@ class Druid(BaseActivity):
f"Loading data from Druid: {datasource} with query: {query}" f"Loading data from Druid: {datasource} with query: {query}"
) )
places = Table(datasource, MetaData(), autoload_with=self.engine) places = Table(datasource, MetaData(), autoload_with=self.druid_engine)
stmt = select(places).where(text(query)) stmt = select(places).where(text(query))
result = pd.read_sql(stmt, self.engine) result = pd.read_sql(stmt, self.druid_engine)
result["inserted_at"] = pd.to_datetime(result["__time"]).dt.strftime( result["inserted_at"] = pd.to_datetime(result["__time"]).dt.strftime(
"%Y-%m-%d %H:%M:%S.%f") "%Y-%m-%d %H:%M:%S.%f")

View File

@@ -185,7 +185,7 @@ env:
- name: DRUID_HOST - name: DRUID_HOST
value: "druid-router.druid.svc.cluster.local" value: "druid-router.druid.svc.cluster.local"
- name: DRUID_PORT - name: DRUID_PORT
value: "8081" value: "8888"
ssh: ssh:
enabled: true enabled: true