feat: DBMS -> DB -> SCHEMA -> TABLE -> COLUMN 계층 조회 구현

Author: nahyeongjin1

app/api/user_db_api.py

@@ -7,7 +7,7 @@
 from app.core.response import ResponseMessage
 from app.core.status import CommonCode
 from app.schemas.user_db.db_profile_model import DBProfileInfo, UpdateOrCreateDBProfile
-from app.schemas.user_db.result_model import ColumnInfo, DBProfile, SchemaDetail, TableInfo
+from app.schemas.user_db.result_model import ColumnInfo, DBDetail, DBProfile, TableInfo
 from app.services.user_db_service import UserDbService, user_db_service
 
 user_db_service_dependency = Depends(lambda: user_db_service)
@@ -160,13 +160,13 @@ def find_all_schema_info(
 
 @router.get(
     "/find/hierarchical-schema/{profile_id}",
-    response_model=ResponseMessage[list[SchemaDetail]],
+    response_model=ResponseMessage[DBDetail],
     summary="특정 DB의 전체 스키마의 계층적 상세 정보 조회",
     description="스키마, 테이블, 컬럼, 제약조건, 인덱스를 포함한 모든 스키마 정보를 계층 구조로 반환합니다.",
 )
 def find_hierarchical_schema_info(
     profile_id: str, service: UserDbService = user_db_service_dependency
-) -> ResponseMessage[list[SchemaDetail]]:
+) -> ResponseMessage[DBDetail]:
     db_info = service.find_profile(profile_id)
     hierarchical_schema_info = service.get_hierarchical_schema_info(db_info)
 

app/schemas/user_db/result_model.py

@@ -118,3 +118,11 @@ class SchemaDetail(BaseModel):
 
     schema_name: str = Field(..., description="스키마 이름")
     tables: list[TableInfo] = Field([], description="테이블 목록")
+
+
+class DBDetail(BaseModel):
+    """계층적 조회에서 DB 정보를 담는 모델 (스키마 포함)"""
+
+    db_name: str | None = Field(None, description="데이터베이스 이름")
+    db_type: str = Field(..., description="데이터베이스 종류")
+    schemas: list[SchemaDetail] = Field([], description="스키마 목록")

app/services/user_db_service.py

@@ -19,6 +19,7 @@
     BasicResult,
     ChangeProfileResult,
     ColumnListResult,
+    DBDetail,
     SchemaDetail,
     SchemaInfoResult,
     TableInfo,
@@ -220,7 +221,7 @@ def get_full_schema_info(
 
     def get_hierarchical_schema_info(
         self, db_info: AllDBProfileInfo, repository: UserDbRepository = user_db_repository
-    ) -> list[SchemaDetail]:
+    ) -> DBDetail:
         """
         DB 프로필 정보를 받아 해당 데이터베이스의 전체 스키마 정보를
         계층적인 구조 (스키마 -> 테이블 -> 컬럼 등)로 조회하여 반환합니다.
@@ -243,7 +244,7 @@ def get_hierarchical_schema_info(
             if db_info.type.lower() == "sqlite" and not schemas_to_scan:
                 schemas_to_scan = ["main"]
 
-            hierarchical_schema_info = []
+            schema_details = []
             for schema_name in sorted(schemas_to_scan):
                 # For Oracle, schema names are uppercase.
                 effective_schema_name = schema_name
@@ -269,10 +270,10 @@ def get_hierarchical_schema_info(
                     table_details.append(table_info)
 
                 if table_details:
-                    hierarchical_schema_info.append(SchemaDetail(schema_name=schema_name, tables=table_details))
+                    schema_details.append(SchemaDetail(schema_name=schema_name, tables=table_details))
 
-            logging.info(f"Finished hierarchical schema scan. Total schemas found: {len(hierarchical_schema_info)}.")
-            return hierarchical_schema_info
+            logging.info(f"Finished hierarchical schema scan. Total schemas found: {len(schema_details)}.")
+            return DBDetail(db_name=db_info.name, db_type=db_info.type, schemas=schema_details)
         except APIException:
             raise
         except Exception as e: