diff --git a/modules/core/src/main/java/com/jeesite/modules/sys/db/InitCoreData.java b/modules/core/src/main/java/com/jeesite/modules/sys/db/InitCoreData.java index 586a0fd3..ed09db25 100644 --- a/modules/core/src/main/java/com/jeesite/modules/sys/db/InitCoreData.java +++ b/modules/core/src/main/java/com/jeesite/modules/sys/db/InitCoreData.java @@ -86,7 +86,7 @@ public class InitCoreData extends BaseInitDataTests { /** * 区域、行政区划表 */ - public void initArea() throws Exception{ + public void initArea(String... prefixes) throws Exception{ clearTable(Area.class); initExcelData(Area.class, new MethodCallback() { @Override @@ -95,7 +95,10 @@ public class InitCoreData extends BaseInitDataTests { if("save".equals(action)){ Area entity = (Area)params[1]; entity.setIsNewRecord(true); - areaService.save(entity); + if (prefixes == null || prefixes.length == 0 + || StringUtils.startsWithAny(entity.getAreaCode(), prefixes)){ + areaService.save(entity); + } return null; } return null; diff --git a/web/src/test/java/com/jeesite/test/InitCoreData.java b/web/src/test/java/com/jeesite/test/InitCoreData.java index de19706c..609c8ba0 100644 --- a/web/src/test/java/com/jeesite/test/InitCoreData.java +++ b/web/src/test/java/com/jeesite/test/InitCoreData.java @@ -24,7 +24,7 @@ public class InitCoreData extends com.jeesite.modules.sys.db.InitCoreData { public void initCoreData() throws Exception{ createTable(); initLog(); - initArea(); + initArea("3700","3701","3702"); initConfig(); initModule(); initDict();