diff --git a/src/views/tool/gen/index.vue b/src/views/tool/gen/index.vue
index 8c52ddc..8785ea2 100644
--- a/src/views/tool/gen/index.vue
+++ b/src/views/tool/gen/index.vue
@@ -289,21 +289,10 @@ function handleQuery() {
 }
 /** 生成代码操作 */
 function handleGenTable(row) {
-  const tbNames = row.tableName || tableNames.value;
-  if (tbNames == "") {
-    proxy.$modal.msgError("请选择要生成的数据");
-    return;
-  }
-  if (row.genType === "1") {
-    genCode(row.tableName).then((response) => {
-      proxy.$modal.msgSuccess("成功生成到自定义路径：" + row.genPath);
-    });
-  } else {
-    proxy.$download.zip(
-      "/tool/gen/batchGenCode?tables=" + tbNames,
-      "ruoyi.zip"
-    );
-  }
+  proxy.$download.zip(
+      "/tool/gen/download/" + row.tableId,
+      row.tableName + ".zip"
+  );
 }
 /** 同步数据库操作 */
 function handleSynchDb(row) {