提交工作更新

backend-java/src/main/java/com/nanri/aiimage/modules/permission/service/PermissionMenuSchemaInitializer.java

@@ -6,12 +6,19 @@ import lombok.extern.slf4j.Slf4j;
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.stereotype.Component;
 
+import java.util.List;
+
 @Component
 @RequiredArgsConstructor
 @Slf4j
 public class PermissionMenuSchemaInitializer {
 
     private final JdbcTemplate jdbcTemplate;
+    private static final List<DefaultAdminMenu> DEFAULT_ADMIN_MENUS = List.of(
+            new DefaultAdminMenu("数据去重总数据", "admin_dedupe_total_data", "dedupe-total-data"),
+            new DefaultAdminMenu("店铺管理", "admin_shop_manage", "shop-manage"),
+            new DefaultAdminMenu("跳过跟价ASIN", "admin_skip_price_asin", "skip-price-asin")
+    );
 
     @PostConstruct
     public void initialize() {
@@ -27,6 +34,7 @@ public class PermissionMenuSchemaInitializer {
         executeQuietly("ALTER TABLE columns ADD COLUMN menu_type VARCHAR(20) NOT NULL DEFAULT 'app' COMMENT '菜单类型: app/admin' AFTER column_key");
         executeQuietly("ALTER TABLE columns ADD COLUMN route_path VARCHAR(255) NOT NULL DEFAULT '' COMMENT '菜单路由或页面标识' AFTER menu_type");
         executeQuietly("UPDATE columns SET menu_type = 'app' WHERE menu_type IS NULL OR menu_type = ''");
+        ensureDefaultAdminMenus();
         executeQuietly("""
                 INSERT INTO columns (name, column_key, menu_type, route_path)
                 SELECT '店铺管理', 'admin_shop_manage', 'admin', 'shop-manage'
@@ -45,6 +53,18 @@ public class PermissionMenuSchemaInitializer {
                 """);
     }
 
+    private void ensureDefaultAdminMenus() {
+        for (DefaultAdminMenu menu : DEFAULT_ADMIN_MENUS) {
+            executeQuietly("""
+                    INSERT INTO columns (name, column_key, menu_type, route_path)
+                    SELECT '%s', '%s', 'admin', '%s'
+                    WHERE NOT EXISTS (
+                        SELECT 1 FROM columns WHERE column_key = '%s'
+                    )
+                    """.formatted(menu.name(), menu.columnKey(), menu.routePath(), menu.columnKey()));
+        }
+    }
+
     private void executeQuietly(String sql) {
         try {
             jdbcTemplate.execute(sql);
@@ -52,4 +72,7 @@ public class PermissionMenuSchemaInitializer {
             log.debug("[permission-menu] schema init skipped: {}", ex.getMessage());
         }
     }
+
+    private record DefaultAdminMenu(String name, String columnKey, String routePath) {
+    }
 }