♻️ refactor(model): ??????????

Closes #97

Author: ZhaoXingPeng

src/dbjavagenix/core/models.py

@@ -68,6 +68,7 @@ class ColumnInfo:
     max_length: Optional[int] = None
     precision: Optional[int] = None
     scale: Optional[int] = None
+    auto_increment: bool = False
 
 
 @dataclass

src/dbjavagenix/database/codegen_tools.py

@@ -160,12 +160,11 @@ def _build_table_info(
                 primary_key=col["primary_key"],
                 default_value=col["default_value"],
                 comment=col["comment"],
+                auto_increment=col["auto_increment"],
+                max_length=col["max_length"],
+                precision=col.get("precision"),
+                scale=col.get("scale"),
             )
-            # 添加额外属性
-            column_obj.auto_increment = col["auto_increment"]
-            column_obj.max_length = col["max_length"]
-            column_obj.precision = col.get("precision")
-            column_obj.scale = col.get("scale")
             column_objects.append(column_obj)
 
         table_obj = TableInfo(
@@ -193,10 +192,10 @@ def _column_to_dict(self, column: ColumnInfo) -> Dict[str, Any]:
             "primary_key": column.primary_key,
             "default_value": column.default_value,
             "comment": column.comment,
-            "auto_increment": getattr(column, "auto_increment", False),
-            "max_length": getattr(column, "max_length", None),
-            "precision": getattr(column, "precision", None),
-            "scale": getattr(column, "scale", None),
+            "auto_increment": column.auto_increment,
+            "max_length": column.max_length,
+            "precision": column.precision,
+            "scale": column.scale,
             "java_type": column.java_type,
         }
 

tests/unit/test_codegen_analyzer.py

@@ -1,5 +1,6 @@
 """Unit tests for the public code-generation analysis workflow."""
 
+import asyncio
 from types import SimpleNamespace
 
 import pytest
@@ -164,6 +165,39 @@ async def test_table_analysis_forwards_schema_to_all_metadata_queries():
     assert result["relationships"]["primary_keys"] == ["id"]
 
 
+def test_table_analysis_keeps_column_metadata_typed():
+    analyzer = CodegenAnalyzer(object())
+    introspector = _SchemaAwareIntrospector()
+
+    def get_columns(_connection_id, _table_name, _schema=None):
+        return [
+            {
+                "name": "id",
+                "type": "BIGINT",
+                "nullable": False,
+                "primary_key": True,
+                "default_value": None,
+                "comment": "Primary key",
+                "auto_increment": True,
+                "max_length": None,
+                "precision": None,
+                "scale": None,
+            }
+        ]
+
+    introspector.get_columns = get_columns
+    analyzer.introspector = introspector
+
+    result = asyncio.run(
+        analyzer.analyze_table_for_codegen("mysql-1", "users", template_category="Default")
+    )
+
+    column = result["table_info"]["columns"][0]
+    assert column["auto_increment"] is True
+    assert column["precision"] is None
+    assert column["scale"] is None
+
+
 def test_codegen_entrypoints_expose_optional_schema():
     atomic = next(
         tool for tool in get_atomic_codegen_tools() if tool.name == "codegen_build_context"

tests/unit/test_models.py

@@ -21,6 +21,19 @@ def test_column_info_creation():
     assert column.name == "user_id"
     assert column.java_type == "Long"
     assert column.primary_key is True
+    assert column.auto_increment is False
+
+
+def test_column_info_persists_auto_increment_metadata():
+    column = ColumnInfo(
+        name="id",
+        data_type="BIGINT",
+        java_type="Long",
+        primary_key=True,
+        auto_increment=True,
+    )
+
+    assert column.auto_increment is True
 
 
 def test_table_info_entity_name():