diff --git a/xiaoqu-agent-service/src/main/java/me/iiv/agent/entity/m0/DistrictDistrictStrategy.java b/xiaoqu-agent-service/src/main/java/me/iiv/agent/entity/m0/DistrictDistrictStrategy.java index 1b009c5..02f92fd 100644 --- a/xiaoqu-agent-service/src/main/java/me/iiv/agent/entity/m0/DistrictDistrictStrategy.java +++ b/xiaoqu-agent-service/src/main/java/me/iiv/agent/entity/m0/DistrictDistrictStrategy.java @@ -1,6 +1,7 @@ package me.iiv.agent.entity.m0; import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableField; import com.baomidou.mybatisplus.annotation.TableId; import com.baomidou.mybatisplus.annotation.TableName; import lombok.Getter; @@ -35,6 +36,7 @@ public class DistrictDistrictStrategy implements Serializable { private String memo; private String contact; private String mail; + @TableField("buildCount") private Integer buildCount; private String districtNumber; private String phone; diff --git a/xiaoqu-agent-service/src/test/java/me/iiv/agent/mcp/tool/DistrictToolsTest.java b/xiaoqu-agent-service/src/test/java/me/iiv/agent/mcp/tool/DistrictToolsTest.java index 8de1d85..8a457dd 100644 --- a/xiaoqu-agent-service/src/test/java/me/iiv/agent/mcp/tool/DistrictToolsTest.java +++ b/xiaoqu-agent-service/src/test/java/me/iiv/agent/mcp/tool/DistrictToolsTest.java @@ -176,6 +176,13 @@ class DistrictToolsTest { .path("required").toString().contains("\"id\"")); } + @Test + void districtEntityKeepsLegacyBuildCountColumnName() { + assertEquals("buildCount", TableInfoHelper.getTableInfo(DistrictDistrictStrategy.class).getFieldList().stream() + .filter(field -> "buildCount".equals(field.getProperty())) + .findFirst().orElseThrow().getColumn()); + } + @Test void listRejectsUnknownDistrictStateBeforeQuerying() { assertThrows(IllegalArgumentException.class, () -> tools.districtListQuery(