feat：增加备份管理模块

flyfish-backup/pom.xml

@@ -0,0 +1,26 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <parent>
+        <artifactId>flyfish-framework</artifactId>
+        <groupId>com.flyfish.framework</groupId>
+        <version>0.0.1-SNAPSHOT</version>
+    </parent>
+    <modelVersion>4.0.0</modelVersion>
+
+    <artifactId>flyfish-backup</artifactId>
+
+    <properties>
+        <maven.compiler.source>8</maven.compiler.source>
+        <maven.compiler.target>8</maven.compiler.target>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>com.flyfish.framework</groupId>
+            <artifactId>flyfish-web</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+    </dependencies>
+</project>

flyfish-backup/src/main/java/com/flyfish/framework/backup/controller/BackupController.java

@@ -0,0 +1,36 @@
+package com.flyfish.framework.backup.controller;
+
+import com.flyfish.framework.backup.domain.Backup;
+import com.flyfish.framework.backup.scheduler.BackupScheduler;
+import com.flyfish.framework.bean.Result;
+import com.flyfish.framework.controller.reactive.ReactiveBaseController;
+import com.flyfish.framework.domain.base.NameLikeQo;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.annotation.Resource;
+
+/**
+ * 备份controller
+ *
+ * @author wangyu
+ */
+@RestController
+@RequestMapping("backups")
+public class BackupController extends ReactiveBaseController<Backup, NameLikeQo<Backup>> {
+
+    @Resource
+    private BackupScheduler backupScheduler;
+
+    /**
+     * 立即创建备份
+     *
+     * @return 结果
+     */
+    @PostMapping("task")
+    public Result<String> createTask() {
+        backupScheduler.backup();
+        return Result.accept("已创建备份，完成后会出现在列表中，请稍后刷新再试。");
+    }
+}

flyfish-backup/src/main/java/com/flyfish/framework/backup/domain/Backup.java

@@ -1,10 +1,11 @@
-package com.flyfish.framework.domain.po;
+package com.flyfish.framework.backup.domain;
 
 import com.flyfish.framework.domain.base.AuditDomain;
 import com.flyfish.framework.enums.NamedEnum;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.Setter;
+import org.springframework.data.mongodb.core.mapping.Document;
 
 /**
  * 系统备份
@@ -13,6 +14,7 @@ import lombok.Setter;
  */
 @Getter
 @Setter
+@Document(collection = "backups")
 public class Backup extends AuditDomain {
 
     // 文件路径

flyfish-backup/src/main/java/com/flyfish/framework/backup/repository/BackupRepository.java

@@ -0,0 +1,12 @@
+package com.flyfish.framework.backup.repository;
+
+import com.flyfish.framework.backup.domain.Backup;
+import com.flyfish.framework.repository.DefaultReactiveRepository;
+
+/**
+ * 备份仓库
+ *
+ * @author wangyu
+ */
+public interface BackupRepository extends DefaultReactiveRepository<Backup> {
+}

flyfish-backup/src/main/java/com/flyfish/framework/backup/scheduler/BackupIndex.java

@@ -1,9 +1,8 @@
-package com.flyfish.framework.scheduler;
+package com.flyfish.framework.backup.scheduler;
 
 import lombok.AllArgsConstructor;
 import lombok.Data;
 
-import java.nio.ByteBuffer;
 import java.util.List;
 
 /**

flyfish-backup/src/main/java/com/flyfish/framework/backup/scheduler/BackupScheduler.java

@@ -1,8 +1,8 @@
-package com.flyfish.framework.scheduler;
+package com.flyfish.framework.backup.scheduler;
 
 import com.alibaba.fastjson.JSON;
+import com.flyfish.framework.backup.domain.Backup;
 import com.flyfish.framework.domain.base.DomainService;
-import com.flyfish.framework.domain.po.Backup;
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
@@ -20,6 +20,7 @@ import org.springframework.util.unit.DataSize;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
 
+import java.io.IOException;
 import java.nio.channels.AsynchronousFileChannel;
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
@@ -29,6 +30,7 @@ import java.nio.file.StandardOpenOption;
 import java.util.Date;
 import java.util.List;
 import java.util.UUID;
+import java.util.concurrent.atomic.AtomicLong;
 import java.util.stream.Collectors;
 
 /**
@@ -75,9 +77,8 @@ public class BackupScheduler {
     public void backup() {
         // 本次备份任务代号
         String code = UUID.randomUUID().toString();
-        // 本次备份父目录
+        // 本次备份父目录，出错时替换目录，保证备份成功
-        String parent = backupPath + "/" + code;
+        String parent = createIfNotExists(backupPath + "/" + code);
-        createIfNotExists(parent);
         // 开始备份，先构造一个指示器
         Backup backup = new Backup();
         backup.setCreateTime(new Date());
@@ -85,6 +86,8 @@ public class BackupScheduler {
         backup.setCode(code);
         backup.setStatus(Backup.Status.RUNNING);
         backup.setFilepath(parent + "/back.zip");
+        // 数据大小计数器
+        AtomicLong sizeCounter = new AtomicLong(0);
         // 开始备份
         operations.save(backup)
                 .thenMany(Flux.fromIterable(this.collections))
@@ -98,7 +101,9 @@ public class BackupScheduler {
                     BackupIndex index = new BackupIndex();
                     index.setId(code);
                     index.setItems(list.stream().map(content -> {
-                        DataSize size = DataSize.parse(new String(content.getContent(), StandardCharsets.UTF_8));
+                        long currentSize = content.getContent().length;
+                        sizeCounter.addAndGet(currentSize);
+                        DataSize size = DataSize.ofBytes(currentSize);
                         BackupIndex.BackupItem item = new BackupIndex.BackupItem();
                         item.setCollection(content.getCollection());
                         item.setPath(parent + "/" + content.getCollection() + ".json");
@@ -112,6 +117,7 @@ public class BackupScheduler {
                     backup.setLog("成功备份");
                     backup.setPeriod(new Date().getTime() - backup.getCreateTime().getTime());
                     backup.setStatus(Backup.Status.SUCCESS);
+                    backup.setSize(DataSize.ofBytes(sizeCounter.get()).toKilobytes() + "KB");
                     return operations.save(backup);
                 }))
                 .subscribe(v -> {
@@ -131,11 +137,17 @@ public class BackupScheduler {
      * @param filepath 路径
      */
     @SneakyThrows
-    private void createIfNotExists(String filepath) {
+    private String createIfNotExists(String filepath) {
         Path path = Paths.get(filepath);
         if (!Files.exists(path)) {
-            Files.createDirectories(path);
+            try {
+                Files.createDirectories(path);
+            } catch (IOException e) {
+                log.error(e.getMessage());
+                return createIfNotExists(System.getProperty("user.home") + filepath);
+            }
         }
+        return filepath;
     }
 
     /**
@@ -157,7 +169,7 @@ public class BackupScheduler {
      * @return 结果
      */
     private Mono<Void> write(byte[] content, String path) {
-        return Mono.fromCallable(() -> AsynchronousFileChannel.open(Paths.get(path), StandardOpenOption.WRITE))
+        return Mono.fromCallable(() -> AsynchronousFileChannel.open(Files.createFile(Paths.get(path)), StandardOpenOption.WRITE))
                 .flatMapMany(channel -> DataBufferUtils.write(Mono.just(factory.wrap(content)), channel, 0))
                 .then();
     }

flyfish-backup/src/main/java/com/flyfish/framework/backup/service/BackupService.java

@@ -0,0 +1,24 @@
+package com.flyfish.framework.backup.service;
+
+import com.flyfish.framework.backup.domain.Backup;
+import com.flyfish.framework.service.impl.BaseReactiveServiceImpl;
+import org.springframework.stereotype.Service;
+import reactor.core.publisher.Mono;
+
+/**
+ * 备份服务
+ *
+ * @author wangyu
+ */
+@Service
+public class BackupService extends BaseReactiveServiceImpl<Backup> {
+
+    /**
+     * 恢复指定备份
+     *
+     * @return 结果
+     */
+    public Mono<Void> restore(String id) {
+        return Mono.empty();
+    }
+}

pom.xml

@@ -46,6 +46,7 @@
         <module>flyfish-dict</module>
         <module>flyfish-form</module>
         <module>flyfish-approval</module>
+        <module>flyfish-backup</module>
     </modules>
 
     <repositories>