fix: minor changes and fixes in the api on names

voltrondata · Oct 16, 2023 · 32ee764 · 32ee764
1 parent 110a327
commit 32ee764
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/api/app.py b/api/app.py
@@ -19,7 +19,6 @@
 app = FastAPI()
 
 allowed_urls = [
-    "substrait-fiddle.com",
     "substrait-fiddle.com",
     "*.substrait-fiddle.com",
     "127.0.0.1",
@@ -31,7 +30,7 @@
 
 
 async def get_duck_conn():
-    conn = app.state.duck_pool.initialize()
+    conn = app.state.duck_pool.get_connection()
     try:
         yield conn
     finally:
@@ -69,7 +68,7 @@ async def initialize():
     )
 
 
-@app.get("/health/duckcb/")
+@app.get("/health/duckdb/")
 def check_backend_conn(conn):
     check_duckdb_connection(conn)
 

diff --git a/api/backend/duckdb.py b/api/backend/duckdb.py
@@ -41,7 +41,7 @@ def check_pool(self):
                 conn.load_extension("substrait")
                 self.conn_pool.append(conn)
 
-    def initialize(self):
+    def get_connection(self):
         self.check_pool()
         con = self.conn_pool.pop(0)
         return con