优化代码生成器

generator/src/main/java/io/geekidea/springbootplus/generator/SpringBootPlusGenerator.java

@@ -25,6 +25,8 @@ import io.geekidea.springbootplus.generator.constant.GeneratorConstant;
 import io.geekidea.springbootplus.generator.properties.GeneratorProperties;
 import io.geekidea.springbootplus.generator.properties.GeneratorProperties;
 import org.springframework.stereotype.Component;
 import org.springframework.stereotype.Component;
 
 
+import java.util.Scanner;
+
 /**
 /**
  * spring-boot-plus代码生成器入口类
  * spring-boot-plus代码生成器入口类
  *
  *
@@ -40,7 +42,12 @@ public class SpringBootPlusGenerator {
      * @param args
      * @param args
      */
      */
     public static void main(String[] args) {
     public static void main(String[] args) {
-        getCode("123");
+
+        Scanner sc = new Scanner(System.in);
+        System.out.println("请输入表名称：");
+        String name = sc.nextLine();
+
+        getCode(name);
     }
     }
 
 
     private static void getCode(String tableName) {
     private static void getCode(String tableName) {
@@ -51,7 +58,7 @@ public class SpringBootPlusGenerator {
                 .setMavenModuleName("common")
                 .setMavenModuleName("common")
                 .setParentPackage("com.jumeirah.common")
                 .setParentPackage("com.jumeirah.common")
 //                .setModuleName("api-app")
 //                .setModuleName("api-app")
-                .setAuthor("wei")
+                .setAuthor("xxx") // 设置作者名称
                 .setFileOverride(true);
                 .setFileOverride(true);
 
 
         // 设置表信息
         // 设置表信息