拓元日志

sys-api-service/src/main/java/com/xy/service/SysDictServiceImpl.java

@@ -1,5 +1,6 @@
 package com.xy.service;
 
+import cn.hutool.core.collection.CollUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.StringUtils;
@@ -24,6 +25,7 @@ import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestHeader;
 
+import java.util.ArrayList;
 import java.util.List;
 
 import static com.xy.utils.Beans.copy;
@@ -56,6 +58,26 @@ public class SysDictServiceImpl extends ServiceImpl<SysDictMapper, SysDict> impl
         SysDicts.forEach(sysDict -> refuRedis(sysDict));
     }
 
+    @ApiOperation("同步到生产环境")
+    @PostMapping("syncProd")
+    public R syncProd(@RequestBody @Validated SysDictDto.SyncProd syncProd) {
+        String paterCode = syncProd.getPaterCode();
+        LambdaQueryWrapper<SysDict> lambdaQueryWrapper = new LambdaQueryWrapper<SysDict>().eq(SysDict::getCode, paterCode);
+        SysDict sysDict = getOne(lambdaQueryWrapper);
+        if (sysDict == null) {
+            return R.ok();
+        }
+        List<SysDict> sysDictsAll = new ArrayList<>();
+        sysDictsAll.add(sysDict);
+        LambdaQueryWrapper<SysDict> lambdaQueryWrapperSub = new LambdaQueryWrapper<SysDict>().eq(SysDict::getCode, paterCode);
+        List<SysDict> sysDicts = list(lambdaQueryWrapperSub);
+        if (CollUtil.isNotEmpty(sysDicts)) {
+            sysDictsAll.addAll(sysDicts);
+        }
+        //TODO 同步到生产环境
+        return R.ok();
+    }
+
     @ApiOperation("添加")
     @PostMapping("save")
     public R save(@RequestBody @Validated SysDictDto.Save save) {

sys-api/src/main/java/com/xy/dto/SysDictDto.java

@@ -21,6 +21,14 @@ import java.util.List;
 @Accessors(chain = true)
 public class SysDictDto {
 
+    @Data
+    @Accessors(chain = true)
+    public static class SyncProd {
+        @NotBlank(message = "父编码不能为空")
+        @ApiModelProperty("父编码")
+        private String paterCode;
+    }
+
     @Data
     @Accessors(chain = true)
     public static class Save {

sys-start/src/main/java/com/xy/GenCode.java

@@ -5,6 +5,8 @@ import com.xy.utils.MybatisGenerator2;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Component;
 
+import java.io.File;
+
 /**
  * 自动生成代码
  */
@@ -12,32 +14,59 @@ import org.springframework.stereotype.Component;
 @Component
 public class GenCode {
 
-    public static void main(String[] args) {
+    /**
+     * 服务名，不含xy-
+     */
+    public static final String SERVICE_NAME = "sys";
+    /**
+     * 作者
+     */
+    public static final String AUTHOR = "谭斌";
 
-//        MybatisGenerator.Generator.builder().build().create();
-        gen2();
 
+    public static void main(String[] args) {
+        gen2();
     }
 
     public static void gen2() {
         MybatisGenerator2.Generator.builder()
-                .author("谭斌")
-                .commPath("D:\\xy_workspace\\xy-new\\xy-sys")
-                .dtoModule("sys-api")
+                .author(AUTHOR)
+                .commPath(getCommPath(SERVICE_NAME))
+                .dtoModule(SERVICE_NAME + "-api")
                 .dtoPackage("com.xy")
-                .serviceModule("sys-api")
+                .serviceModule(SERVICE_NAME + "-api")
                 .servicePackage("com.xy")
-                .implModule("sys-api-service")
+                .implModule(SERVICE_NAME + "-api-service")
                 .implPackage("com.xy")
-                .mapperModule("sys-api-service")
+                .mapperModule(SERVICE_NAME + "-api-service")
                 .mapperPackage("com.xy")
-                .feignModule("sys-api-cloud")
+                .feignModule(SERVICE_NAME + "-api-cloud")
                 .feignPackage("com.xy")
-                .entityModule("sys-api-service")
+                .entityModule(SERVICE_NAME + "-api-service")
                 .entityPackage("com.xy")
                 .idType(IdType.ASSIGN_ID)
                 .autoFill(true)
                 .deleteFieldName("deleted")
                 .build().create();
     }
-}
+
+    /**
+     * 获取项目的根路径
+     *
+     * @return
+     */
+    private static String getCommPath(String serviceName) {
+        // "xy-merc\merc-start\target\classes"
+        String strReplace = String.format("xy-%s\\%s-start\\target\\classes", serviceName, serviceName);
+        // 获取当前类的路径
+        String className = GenCode.class.getName().replace('.', '/') + ".class";
+        String classPath = GenCode.class.getClassLoader().getResource(className).getPath();
+        // 获取当前类的目录
+        File classFile = new File(classPath);
+        String classDir = classFile.getParentFile().getParentFile().getAbsolutePath();
+        // 构建 commPath
+        String commPath = classDir.substring(0, classDir.lastIndexOf(File.separator));
+        commPath = commPath.replace(strReplace, "") + "xy-" + serviceName;
+        return commPath;
+    }
+}