diff --git a/src/views/tool/gen/index.vue b/src/views/tool/gen/index.vue
index ff10f0759edb3736986b3afc9d65a637b163ad06..863d52e1d89de222e426f7b8e4572e90430e9496 100644
--- a/src/views/tool/gen/index.vue
+++ b/src/views/tool/gen/index.vue
@@ -285,7 +285,12 @@ function handleDelete(row) {
 }
 /** 生成代码操作 */
 function handleGenTable(row) {
-    proxy.$download.zip("/tool/gen/genCode/" + row.tableId, "baize.zip");
+  const tableId = row.tableId || ids.value[0];
+  if (!tableId) {
+    proxy.$modal.msgError("请选择要生成的数据，且一次只能选择一条");
+    return;
+  }
+  proxy.$download.zip("/tool/gen/genCode/" + tableId, "baize.zip");
 }
 getList();
 </script>