diff --git a/src/api/generator.ts b/src/api/generator.ts
index 4686ad22d59635dbb72ca25cb6b74301a39ecc23..dc10cdae3afed0c3f6f2922231dbef6e3468689c 100644
--- a/src/api/generator.ts
+++ b/src/api/generator.ts
@@ -38,6 +38,26 @@ class GeneratorAPI {
       method: "get",
     });
   }
+  /**
+   * 下载 ZIP 文件
+   * @param url
+   * @param fileName
+   */
+  static downloadZip(tableName: string, fileName?: string) {
+    return request({
+      url: `${GENERATOR_BASE_URL}/${tableName}/downloadZip`,
+      method: "get",
+      responseType: "blob",
+    }).then((res) => {
+      const blob = new Blob([res.data], { type: "application/zip" });
+      const a = document.createElement("a");
+      const url = window.URL.createObjectURL(blob);
+      a.href = url;
+      a.download = fileName || "下载文件.zip";
+      a.click();
+      window.URL.revokeObjectURL(url);
+    });
+  }
 }
 
 export default GeneratorAPI;
diff --git a/src/views/generator/index.vue b/src/views/generator/index.vue
index fc4ae20c1bad75249664038c3270e1e3bf7361e7..a970af114e2ee84c0934e8ef1fca4c19a3e62807 100644
--- a/src/views/generator/index.vue
+++ b/src/views/generator/index.vue
@@ -1,12 +1,12 @@
 <template>
   <div class="app-container">
     <div class="search-container">
-      <el-form ref="queryFormRef" :model="queryParams" :inline="true">
-        <el-form-item prop="keywords" label="关键字">
+      <el-form ref="queryFormRef" :inline="true" :model="queryParams">
+        <el-form-item label="关键字" prop="keywords">
           <el-input
             v-model="queryParams.keywords"
-            placeholder="表名"
             clearable
+            placeholder="表名"
             @keyup.enter="handleQuery"
           />
         </el-form-item>
@@ -24,33 +24,33 @@
       </el-form>
     </div>
 
-    <el-card shadow="never" class="table-container">
+    <el-card class="table-container" shadow="never">
       <el-table
         ref="dataTableRef"
         v-loading="loading"
         :data="pageData"
-        highlight-current-row
         border
+        highlight-current-row
       >
-        <el-table-column type="selection" width="55" align="center" />
-        <el-table-column label="表名" prop="tableName" min-width="100" />
+        <el-table-column align="center" type="selection" width="55" />
+        <el-table-column label="表名" min-width="100" prop="tableName" />
         <el-table-column label="描述" prop="tableComment" width="150" />
 
-        <el-table-column label="存储引擎" align="center" prop="engine" />
+        <el-table-column align="center" label="存储引擎" prop="engine" />
 
         <el-table-column
-          label="排序规则"
           align="center"
+          label="排序规则"
           prop="tableCollation"
         />
-        <el-table-column label="创建时间" align="center" prop="createTime" />
+        <el-table-column align="center" label="创建时间" prop="createTime" />
 
         <el-table-column fixed="right" label="操作" width="150">
           <template #default="scope">
             <el-button
-              type="primary"
-              size="small"
               link
+              size="small"
+              type="primary"
               @click="handleOpenDialog(scope.row.tableName)"
             >
               <i-ep-MagicStick />
@@ -62,9 +62,9 @@
 
       <pagination
         v-if="total > 0"
-        v-model:total="total"
-        v-model:page="queryParams.pageNum"
         v-model:limit="queryParams.pageSize"
+        v-model:page="queryParams.pageNum"
+        v-model:total="total"
         @pagination="handleQuery"
       />
     </el-card>
@@ -72,8 +72,8 @@
     <el-drawer
       v-model="dialog.visible"
       :title="dialog.title"
-      @close="handleCloseDialog"
       size="80%"
+      @close="handleCloseDialog"
     >
       <el-steps :active="active" align-center finish-status="success" simple>
         <el-step title="基础配置" />
@@ -82,7 +82,7 @@
       </el-steps>
 
       <div class="mt-5">
-        <el-form v-show="active == 0" :model="formData" :label-width="100">
+        <el-form v-show="active == 0" :label-width="100" :model="formData">
           <el-row>
             <el-col :span="12">
               <el-form-item label="表名">
@@ -126,9 +126,9 @@
         <el-table
           v-show="active == 1"
           v-loading="loading"
+          :data="formData.fieldConfigs"
           :element-loading-text="loadingText"
           highlight--currentrow
-          :data="formData.fieldConfigs"
         >
           <el-table-column label="字段列名" width="100">
             <template #default="scope">
@@ -187,8 +187,8 @@
               <el-form-item>
                 <el-checkbox
                   v-model="scope.row.isShowInQuery"
-                  :true-value="1"
                   :false-value="0"
+                  :true-value="1"
                 />
               </el-form-item>
             </template>
@@ -199,8 +199,8 @@
               <el-form-item>
                 <el-checkbox
                   v-model="scope.row.isShowInList"
-                  :true-value="1"
                   :false-value="0"
+                  :true-value="1"
                 />
               </el-form-item>
             </template>
@@ -211,8 +211,8 @@
               <el-form-item>
                 <el-checkbox
                   v-model="scope.row.isShowInForm"
-                  :true-value="1"
                   :false-value="0"
+                  :true-value="1"
                 />
               </el-form-item>
             </template>
@@ -223,9 +223,9 @@
               <el-form-item>
                 <el-checkbox
                   v-model="scope.row.isRequired"
-                  :true-value="1"
-                  :false-value="0"
                   :disabled="scope.row.isShowInForm !== 1"
+                  :false-value="0"
+                  :true-value="1"
                 />
               </el-form-item>
             </template>
@@ -296,18 +296,20 @@
           <el-col :span="18">
             <el-scrollbar max-height="72vh">
               <div class="absolute-rt z-36 right-5 top-2">
-                <el-link @click="handleCopyCode" type="primary">
-                  <el-icon><CopyDocument /></el-icon>
+                <el-link type="primary" @click="handleCopyCode">
+                  <el-icon>
+                    <CopyDocument />
+                  </el-icon>
                   一键复制
                 </el-link>
               </div>
 
               <Codemirror
+                ref="cmRef"
                 v-model:value="code"
                 :options="cmOptions"
-                border
-                ref="cmRef"
                 :readonly="true"
+                border
                 height="100%"
                 width="100%"
               />
@@ -317,21 +319,27 @@
       </div>
 
       <template #footer>
-        <el-button type="success" @click="handlePrevClick" v-if="active !== 0">
-          <el-icon><Back /></el-icon>
+        <el-button v-if="active !== 0" type="success" @click="handlePrevClick">
+          <el-icon>
+            <Back />
+          </el-icon>
           {{ prevBtnText }}
         </el-button>
         <el-button type="primary" @click="handleNextClick">
           {{ nextBtnText }}
-          <el-icon v-if="active !== 2"><Right /></el-icon>
-          <el-icon v-else><Download /></el-icon>
+          <el-icon v-if="active !== 2">
+            <Right />
+          </el-icon>
+          <el-icon v-else>
+            <Download />
+          </el-icon>
         </el-button>
       </template>
     </el-drawer>
   </div>
 </template>
 
-<script setup lang="ts">
+<script lang="ts" setup>
 defineOptions({
   name: "Generator",
 });
@@ -396,13 +404,13 @@ function handlePrevClick() {
 }
 
 function handleNextClick() {
+  const tableName = formData.value.tableName;
+  if (!tableName) {
+    ElMessage.error("表名不能为空");
+    return;
+  }
   if (active.value === 1) {
     // 保存生成配置
-    const tableName = formData.value.tableName;
-    if (!tableName) {
-      ElMessage.error("表名不能为空");
-      return;
-    }
     loading.value = true;
     loadingText.value = "代码生成中，请稍后...";
     GeneratorAPI.saveGenConfig(tableName, formData.value)
@@ -418,6 +426,10 @@ function handleNextClick() {
       });
   } else {
     if (active.value++ >= 2) active.value = 2;
+    if (active.value === 2) {
+      GeneratorAPI.downloadZip(tableName, "youlai-code.zip");
+      // console.log("生成的代码", code.value);
+    }
   }
 }
 
@@ -445,6 +457,7 @@ function handleQuery() {
       loading.value = false;
     });
 }
+
 /** 重置查询 */
 function handleResetQuery() {
   queryFormRef.value.resetFields();