diff --git a/src/main/java/com/simon/common/code/CodeGenerator.java b/src/main/java/com/simon/common/code/CodeGenerator.java
index 55b9bd6e69d413b2cec91153885a7a4dd1eafc6b..f5efe489a36247444198a20005e072b8e993cb9f 100644
--- a/src/main/java/com/simon/common/code/CodeGenerator.java
+++ b/src/main/java/com/simon/common/code/CodeGenerator.java
@@ -51,11 +51,14 @@ public class CodeGenerator {
     private static String PACKAGE_PATH_SERVICE_IMPL;//生成的Service实现存放路径
     private static String PACKAGE_PATH_CONTROLLER;//生成的Controller存放路径
 
+    private static String GEN_MODULES;//要生成的模块
+
     protected static String AUTHOR;//@author
     protected static final String CREATE = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date());//@date
 
+    private static Properties prop;
     static {
-        Properties prop = new Properties();
+        prop = new Properties();
         try {
             prop.load(CodeGenerator.class.getResourceAsStream("/code-gen.properties"));
             JDBC_URL = prop.getProperty("jdbc_url");
@@ -79,6 +82,8 @@ public class CodeGenerator {
             PACKAGE_PATH_SERVICE = packageConvertPath(SERVICE_PACKAGE);
             PACKAGE_PATH_SERVICE_IMPL = packageConvertPath(SERVICE_IMPL_PACKAGE);
             PACKAGE_PATH_CONTROLLER = packageConvertPath(CONTROLLER_PACKAGE);
+
+            GEN_MODULES = prop.getProperty("gen_modules");
         } catch (IOException e) {
             e.printStackTrace();
         }
@@ -121,11 +126,25 @@ public class CodeGenerator {
      * @param modelName 自定义的 Model 名称
      */
     private static void genCodeByCustomModelName(String tableName, String modelName) {
-        genModelAndMapper(tableName, modelName);
-        //genRepository(tableName, modelName);
-		//genService(tableName, modelName);
-		//genController(tableName, modelName);
-
+        if(StringUtils.isEmpty(GEN_MODULES)){
+            genModelAndMapper(tableName, modelName);
+            genRepository(tableName, modelName);
+            genService(tableName, modelName);
+            //genController(tableName, modelName);
+        }else{
+            if(GEN_MODULES.toLowerCase().contains("modelandmapper")){
+                genModelAndMapper(tableName, modelName);
+            }
+            if(GEN_MODULES.toLowerCase().contains("repository")){
+                genRepository(tableName, modelName);
+            }
+            if(GEN_MODULES.toLowerCase().contains("service")){
+                genService(tableName, modelName);
+            }
+            if(GEN_MODULES.toLowerCase().contains("controller")){
+                genController(tableName, modelName);
+            }
+        }
     }
 
 
diff --git a/src/main/resources/code-gen.properties b/src/main/resources/code-gen.properties
index 6f17f0c54335fc9ce62cd55a01e3ec912a5b0c4e..dc6edc5bd0dbf86efa3f706731d8e7f4a85c6f37 100644
--- a/src/main/resources/code-gen.properties
+++ b/src/main/resources/code-gen.properties
@@ -12,4 +12,6 @@ resources_path=/src/test/resources
 # ��Ŀ���������ƣ������Լ�����Ŀ�޸�
 base_package=com.simon
 # Mapper��������ӿڵ���ȫ�޶���(�ڶ����ᵽ�ĺ��ļ̳нӿ�Mapper)
-mapper_interface_reference=com.simon.common.mapper.MyMapper
\ No newline at end of file
+mapper_interface_reference=com.simon.common.mapper.MyMapper
+# Ҫ���ɵ�ģ�飬��ѡֵ��modelAndMapper,repository,service,controller������ָ��������modelAndMapper,repository,service
+gen_modules=modelAndMapper,repository,service
\ No newline at end of file