diff --git a/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java b/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java
index ab7acde..ffbaefa 100644
--- a/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java
+++ b/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java
@@ -218,8 +218,8 @@ public class GeneratorUtil {
      * @param secondaryPackageName 二级包名称（实体所在的包位置除去主包名后的位置）
      * @param type                 类型 po entity dto mapper service repository controller
      */
-    public static void generatorByTemplate(List<Column> list,String tableName, String packageName, String secondaryPackageName, String type) {
-        String templatePath = String.format("src/main/resources/template/%s.ftl",toLowerCase(type));
+    public static void generatorByTemplate(List<Column> list, String tableName, String packageName, String secondaryPackageName, String type) {
+        String templatePath = String.format("src/main/resources/template/%s.ftl", toLowerCase(type));
         String moduleName;
         //如果有前缀去除
         String className = tableName;
@@ -248,6 +248,7 @@ public class GeneratorUtil {
         }
         String dir = String.format("%s-%s/%s/%s/", CONFIG.getProjectName(), moduleName, package2Path(CONFIG.getPackageName()), package2Path(secondaryPackageName));
         String filePath = dir + className + SUFFIX;
+        File file = new File(filePath);
         try {
             // 文件夹不存在创建
             if (!new File(dir).exists()) {
@@ -255,7 +256,7 @@ public class GeneratorUtil {
                     // System.out.println("PO 文件夹创建成功");
                 }
                 //创建文件
-                if (!new File(filePath).exists()) {
+                if (!file.exists()) {
                     new File(filePath).createNewFile();
                     System.out.println("=== " + type + ":[" + className + " ] 创建成功 ===");
                 }
@@ -263,7 +264,12 @@ public class GeneratorUtil {
             }
             Configuration configuration = new Configuration(Configuration.VERSION_2_3_0);
             Template template = configuration.getTemplate(templatePath);
-
+            Map<String, Object> data = new HashMap<>();
+            data.put("package", packageName + '.' + secondaryPackageName);
+            data.put("table_name", tableName);
+            data.put("class_name", className);
+            data.put("columns", list);
+            template.process(data, new FileWriter(file));
         } catch (TemplateNotFoundException e) {
             e.printStackTrace();
         } catch (ParseException e) {
@@ -272,6 +278,8 @@ public class GeneratorUtil {
             e.printStackTrace();
         } catch (IOException e) {
             e.printStackTrace();
+        } catch (TemplateException e) {
+            e.printStackTrace();
         }
 
     }