Merge pull request #1144 from moonstream-to/fix-engine

Update engine.
moonstream-to · Nov 27, 2024 · 83850d7 · 83850d7
2 parents 79b464d + b12fb0f
commit 83850d7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/crawlers/mooncrawl/mooncrawl/db.py b/crawlers/mooncrawl/mooncrawl/db.py
@@ -7,7 +7,7 @@
     MOONSTREAM_POOL_SIZE,
     create_moonstream_engine,
 )
-from moonstreamdbv3.db import MoonstreamCustomDBEngine
+from moonstreamdbv3.db import DBEngine
 from sqlalchemy.orm import Session, sessionmaker
 
 from .settings import (

diff --git a/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py b/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py
@@ -15,7 +15,7 @@
 from ..db import (
     RO_pre_ping_query_engine,
     MOONSTREAM_DB_URI_READ_ONLY,
-    MoonstreamCustomDBEngine,
+    DBEngine,
 )
 from ..reporter import reporter
 from ..settings import (
@@ -85,9 +85,7 @@ def data_generate(
         db_uri = get_customer_db_uri(customer_id, instance_id, "customer")
         label = SEER_CRAWLER_LABEL
 
-        engine = MoonstreamCustomDBEngine(
-            url=db_uri, schema=MOONSTREAM_DB_V3_SCHEMA_NAME
-        )
+        engine = DBEngine(url=db_uri, schema=MOONSTREAM_DB_V3_SCHEMA_NAME).engine
     else:
         engine = RO_pre_ping_query_engine