feat: 支持oracle版本

Author: jaguarliu

database_schema/factory.py

@@ -2,7 +2,8 @@
 from database_schema.inspectors import (
     MySQLInspector,
     SQLServerInspector,
-    PostgreSQLInspector
+    PostgreSQLInspector,
+    OracleInspector
 )
 
 class InspectorFactory:
@@ -13,7 +14,8 @@ def create_inspector(db_type: str, **kwargs) -> object:
         mapping = {
             'mysql': MySQLInspector,
             'sqlserver': SQLServerInspector,
-            'postgresql': PostgreSQLInspector
+            'postgresql': PostgreSQLInspector,
+            'oracle': OracleInspector
         }
 
         if db_type not in mapping:

database_schema/inspectors/oracle.py

@@ -0,0 +1,59 @@
+from sqlalchemy.sql import text
+from .base import BaseInspector
+from sqlalchemy.engine import reflection
+from urllib.parse import quote_plus
+
+class OracleInspector(BaseInspector):
+    """Oracle元数据获取实现"""
+    
+    def __init__(self, host: str, port: int, database: str,
+                username: str, password: str, schema_name: str = None, **kwargs):
+        super().__init__(host, port, database, username, password, schema_name)
+        self.schema_name = username.upper()  # Oracle模式名通常与用户名一致[3,8](@ref)
+    
+    def build_conn_str(self, host: str, port: int, database: str,
+                      username: str, password: str) -> str:
+        # 使用cx_Oracle驱动，支持SID或Service Name[6,8](@ref)
+        return (
+            f"oracle+cx_oracle://{quote_plus(username)}:{quote_plus(password)}"
+            f"@{host}:{port}/?service_name={database}"
+        )
+    
+    def get_table_names(self, inspector: reflection.Inspector) -> list[str]:
+        return inspector.get_table_names(schema=self.schema_name)  # 需指定schema[3](@ref)
+    
+    def get_table_comment(self, inspector: reflection.Inspector,
+                         table_name: str) -> str:
+        # 查询ALL_TAB_COMMENTS视图[3,5](@ref)
+        with self.engine.connect() as conn:
+            sql = text("""
+                SELECT COMMENTS 
+                FROM ALL_TAB_COMMENTS 
+                WHERE OWNER = :owner 
+                    AND TABLE_NAME = :table_name
+            """)
+            return conn.execute(sql, {
+                'owner': self.schema_name,
+                'table_name': table_name
+            }).scalar() or ""
+    
+    def get_column_comment(self, inspector: reflection.Inspector,
+                          table_name: str, column_name: str) -> str:
+        # 查询ALL_COL_COMMENTS视图[3](@ref)
+        with self.engine.connect() as conn:
+            sql = text("""
+                SELECT COMMENTS 
+                FROM ALL_COL_COMMENTS 
+                WHERE OWNER = :owner 
+                    AND TABLE_NAME = :table_name 
+                    AND COLUMN_NAME = :column_name
+            """)
+            return conn.execute(sql, {
+                'owner': self.schema_name,
+                'table_name': table_name,
+                'column_name': column_name
+            }).scalar() or ""
+    
+    def normalize_type(self, raw_type: str) -> str:
+        # 标准化Oracle类型（如去除精度信息）[3,5](@ref)
+        return raw_type.split('(')[0].split('%')[0].upper()

prompt_templates/sql_generation/oracle_prompt.jinja

@@ -0,0 +1,49 @@
+{# Oracle专用模板 #}
+{% extends "base_prompt.jinja" %}
+
+{% block optimization_rules %}
+1. 索引优化策略：
+   - 优先创建复合索引覆盖WHERE+SELECT字段
+   - 对高频查询字段建立函数索引
+   - 大数据量表使用位图索引
+2. 查询优化规范：
+   - 使用绑定变量避免硬解析
+   - 分区表查询必须指定分区键
+   - 关联查询驱动表记录量需最小
+3. 执行计划控制：
+   - 强制走索引使用/*+ INDEX(table_name index_name) */
+   - 避免全表扫描的FILTER操作
+4. 特殊语法要求：
+   - 时间范围使用TO_DATE()显式转换
+   - 分页查询使用ROWNUM伪列
+{% endblock %}
+
+{% block validation_rules %}
+1. 统计信息验证：
+   - 自动检查LAST_ANALYZED时间，超过7天提示更新
+2. 执行计划验证：
+   - 通过DBMS_XPLAN检查ACCESS_PREDICATES
+   - 确保INDEX RANGE SCAN优于TABLE ACCESS FULL
+3. 性能基线检查：
+   - WHERE条件字段必须有统计直方图
+   - 多表关联必须存在连接条件索引
+{% endblock %}
+
+{% block example_section %}
+## 输出示例：
+SELECT /*+ INDEX(emp emp_dept_idx) */ 
+    e.employee_id AS "工号",
+    (e.salary * 1.1) AS "调整薪资",
+    TO_CHAR(e.hire_date, 'YYYY-MM-DD') AS "入职日期"
+FROM 
+    employees e
+INNER JOIN 
+    departments d ON e.department_id = d.department_id  
+WHERE 
+    e.job_id = 'IT_PROG'
+    AND d.location_id IN (1700, 1800)
+    AND e.hire_date BETWEEN TO_DATE('2020-01-01', 'YYYY-MM-DD') AND SYSDATE
+ORDER BY 
+    e.hire_date DESC
+FETCH FIRST 100 ROWS ONLY;
+{% endblock %}

tools/rookie_text2data.yaml

@@ -35,6 +35,9 @@ parameters:
       - label:
           en_US: SQL Server
         value: sqlserver
+      - label:
+          en_US: Oracle
+        value: oracle
   - name: limit
     type: number
     required: false
@@ -76,7 +79,7 @@ parameters:
   - name: host
     type: string
     required: true
-    form: form
+    form: llm
     label:
       en_US: Database ip/host
       zh_Hans: 数据库IP/域名
@@ -89,7 +92,7 @@ parameters:
   - name: port
     type: number
     required: true
-    form: form
+    form: llm
     min: 1
     max: 65535
     label:
@@ -104,7 +107,7 @@ parameters:
   - name: db_name
     type: string
     required: true
-    form: form
+    form: llm
     label:
       en_US: Database name
       zh_Hans: 数据库名称
@@ -143,7 +146,7 @@ parameters:
   - name: username
     type: string
     required: true
-    form: form
+    form: llm
     label:
       en_US: Username
       zh_Hans: 用户名
@@ -156,7 +159,7 @@ parameters:
   - name: password
     type: secret-input
     required: true
-    form: form
+    form: llm
     label:
       en_US: Password
       zh_Hans: 密码