ÎļþÃû´Ó src/main/java/com/product/system/backup/service/SystemBackupService2.java ÐÞ¸Ä |
| | |
| | | import java.io.InputStream; |
| | | import java.io.InputStreamReader; |
| | | import java.nio.charset.StandardCharsets; |
| | | import java.nio.file.Files; |
| | | import java.nio.file.Path; |
| | | import java.nio.file.Paths; |
| | | import java.text.SimpleDateFormat; |
| | | import java.util.Date; |
| | | import java.util.List; |
| | | import java.util.Map; |
| | | import java.util.Properties; |
| | | import java.util.stream.Collectors; |
| | | import java.util.zip.ZipEntry; |
| | | import java.util.zip.ZipOutputStream; |
| | | |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.stereotype.Service; |
| | | |
| | | import com.alibaba.druid.util.StringUtils; |
| | |
| | | import com.jcraft.jsch.SftpException; |
| | | import com.jcraft.jsch.SftpProgressMonitor; |
| | | import com.product.core.config.Global; |
| | | import com.product.core.dao.BaseDao; |
| | | import com.product.core.entity.DataTableEntity; |
| | | import com.product.core.entity.FieldSetEntity; |
| | | import com.product.core.service.support.AbstractBaseService; |
| | | import com.product.system.backup.entity.BackupLogger; |
| | | |
| | | @Service("systemBackService") |
| | | public class SystemBackupService2 extends AbstractBaseService{ |
| | | |
| | | @Autowired |
| | | BaseDao baseDao; |
| | | /** |
| | | * ç³»ç»å¤ä»½ |
| | | * æ°æ®åºå¤ä»½ï¼æ¯å¤©æ§è¡ä»£ç å¤ä»½ |
| | | * éä»¶å¤ä»½ï¼æ¯å¤©æ§è¡ä»£ç å¤ä»½å½å¤©éä»¶ |
| | | * éçé®é¢ï¼é件被修æ¹ï¼ä¸è½éå®éå½å¤©è¢«ä¿®æ¹çéä»¶ |
| | | */ |
| | | @Service("systemBackService1") |
| | | public class SystemBackupServiceV1 { |
| | | |
| | | // å¤ä»½é
ç½® |
| | | Properties config; |
| | |
| | | // æ°åæ¶é´åæ°åæ¥æï¼ç¨åæä»¶å¤¹æè
æä»¶åï¼ |
| | | String NUMBER_TIME; //ä¾å¦ï¼20250428091001ï¼ |
| | | String NUMBER_DATE; //ä¾å¦ï¼20240428ï¼ |
| | | String STANDARD_START_TIME; // ä¾å¦ï¼2024-04-28 09:10:00ï¼ |
| | | String STANDARD_FINAL_TIME; // ä¾å¦ï¼2024-04-28 09:10:00ï¼ |
| | | |
| | | // æ°åæ¥ææ ¼å¼ |
| | | SimpleDateFormat numberTimeFormat = new SimpleDateFormat("yyyyMMddHHmm00"); |
| | | SimpleDateFormat numberTimeFormat = new SimpleDateFormat("yyyyMMddHHmmss"); |
| | | SimpleDateFormat numberDateFormat = new SimpleDateFormat("yyyyMMdd"); |
| | | SimpleDateFormat standardTimeFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:00"); |
| | | |
| | | // å¤ä»½é
ç½®æä»¶ |
| | | private static final String CONFIG_FILE_PATH = "systemBackup.properties"; |
| | | |
| | | /** |
| | | * ç³»ç»å¤ä»½å¤å® |
| | | * ç³»ç»å¤ä»½å
¥å£ |
| | | * è°ç¨å°ç¹ï¼å®æ¶ä»»å¡é
ç½®åè½(bean) |
| | | */ |
| | | public void systemBackupInit() { |
| | | // è·åç³»ç»æ¶é´ |
| | | Date finalTime = new Date(); |
| | | Date startTime = null; |
| | | // è·åç³»ç»å¤ä»½å¨æ(åé) |
| | | int backupCycle = Integer.valueOf(Global.getSystemConfig("SYSTEM_BACK_CYCLE", "20")); |
| | | // è·åç³»ç»æå¤§å¤ä»½æ¶é´ |
| | | StringBuilder sbSql = new StringBuilder(); |
| | | sbSql.append("SELECT TIMESTAMPDIFF(MINUTE, end_time, ?) AS diff_minutes,end_time \n"); |
| | | sbSql.append("FROM product_sys_backup_log \n"); |
| | | sbSql.append("WHERE backup_status = 1 \n"); |
| | | sbSql.append("ORDER BY end_time DESC \n"); |
| | | sbSql.append("LIMIT 1 \n"); |
| | | FieldSetEntity fseMaxLogTime = baseDao.getFieldSetEntityBySQL(sbSql.toString(), new Object[] {standardTimeFormat.format(finalTime)}, false); |
| | | if(fseMaxLogTime != null && !StringUtils.isEmpty(fseMaxLogTime.getString("diff_minutes"))) { |
| | | Integer diffMinutes = fseMaxLogTime.getInteger("diff_minutes"); |
| | | if (diffMinutes < backupCycle) { |
| | | // å°äºå¤ä»½å¨æï¼ä¸å¤ä»½ |
| | | return ; |
| | | } |
| | | startTime = fseMaxLogTime.getDate("end_time"); |
| | | } |
| | | try { |
| | | // æ§è¡å¤ä»½ |
| | | systemBackup(startTime, finalTime); |
| | | } catch (Exception e) { |
| | | e.printStackTrace(); |
| | | } |
| | | FieldSetEntity fseBackLog = new FieldSetEntity("product_sys_backup_log"); |
| | | fseBackLog.setValue("start_time", startTime); |
| | | fseBackLog.setValue("end_time", finalTime); |
| | | fseBackLog.setValue("backup_status", 1); |
| | | baseDao.saveFieldSetEntity(fseBackLog); |
| | | } |
| | | |
| | | /** |
| | | * ç³»ç»å¤ä»½å
¥å£ |
| | | * |
| | | */ |
| | | public void systemBackup(Date startTime, Date finalTime){ |
| | | |
| | | // åå§æ¥å¿æä»¶ |
| | | log = new BackupLogger(); |
| | | // 卿¥å¿ä¸è®°å½æä½ç³»ç»ä¿¡æ¯ï¼ä¾¿äºè°è¯ |
| | | log.writeInfo("ãç³»ç»å¤ä»½å
¥å£ãæä½ç³»ç»: " + System.getProperty("os.name"), BackupLogger.INFO_TYPE); |
| | | log.writeInfo("ãç³»ç»å¤ä»½å
¥å£ãæä»¶åé符: " + File.separator, BackupLogger.INFO_TYPE); |
| | | log.writeInfo("ãç³»ç»å¤ä»½å
¥å£ãç³»ç»å¤ä»½å¼å§..................", BackupLogger.INFO_TYPE); |
| | | NUMBER_TIME = numberTimeFormat.format(finalTime); |
| | | NUMBER_DATE = numberDateFormat.format(finalTime); |
| | | if(startTime == null) { |
| | | STANDARD_START_TIME = "2000-01-01 00:00:00"; |
| | | } else { |
| | | STANDARD_START_TIME = standardTimeFormat.format(startTime); |
| | | } |
| | | STANDARD_FINAL_TIME = standardTimeFormat.format(finalTime); |
| | | NUMBER_TIME = numberTimeFormat.format(new Date()); |
| | | NUMBER_DATE = numberDateFormat.format(new Date()); |
| | | log.writeInfo("ãç³»ç»å¤ä»½å
¥å£ãç³»ç»æ¥æï¼" + NUMBER_DATE, BackupLogger.INFO_TYPE); |
| | | log.writeInfo("ãç³»ç»å¤ä»½å
¥å£ãç³»ç»æ¶é´ï¼" + NUMBER_TIME, BackupLogger.INFO_TYPE); |
| | | log.writeInfo("ãç³»ç»æ¬åå
¥å£ãç³»ç»å¤ä»½å¼å§æ¶é´ï¼" + STANDARD_START_TIME, BackupLogger.INFO_TYPE); |
| | | log.writeInfo("ãç³»ç»å¤ä»½å
¥å£ãç³»ç»å¤ä»½æªæ¢æ¶é´ï¼" + STANDARD_FINAL_TIME, BackupLogger.INFO_TYPE); |
| | | // è¿å
¥å¤ä»½è¿ç¨ |
| | | backupProcess(); |
| | | log.closeLogger(); |
| | |
| | | if (!status) |
| | | return; |
| | | // ç¬¬åæ¥ï¼ä¸ä¼ å缩å¤ä»½æä»¶å°FTP |
| | | status = sftpTransferService(); |
| | | status = sftpTransferService(true); |
| | | // uploadBackupMachine2(); |
| | | if (!status) |
| | | return; |
| | | // ç¬¬äºæ¥ï¼æ¸
餿°æ® |
| | |
| | | // è·åç³»ç»æä»¶è·¯å¾ |
| | | String systemFileFolder = Global.getSystemConfig("local.dir", ""); |
| | | File newFile= new File(systemFileFolder); |
| | | // ç³»ç»éä»¶åæ¾ç®å½ |
| | | config.setProperty("DOCUMENT_ROOT", newFile.getAbsolutePath() + File.separator + "00000000-0000-0000-0000-000000000000"); |
| | | // ç³»ç»éä»¶åæ¾ç®å½(å½å¤©éä»¶) |
| | | config.setProperty("DOCUMENT_ROOT", newFile.getAbsolutePath() + File.separator + "00000000-0000-0000-0000-000000000000" + File.separator + NUMBER_DATE); |
| | | // æ°æ®åºå¤ä»½ç®å½ï¼æ°æ®åºå¤ä»½æ ¹ç®å½+æ¶é´æä»¶å+.sqlï¼ |
| | | config.setProperty("DATABASE_BACKUP", config.getProperty("DATABASE_ROOT") + File.separator + NUMBER_TIME + ".sql"); |
| | | // å¤ä»½ç®æ æä»¶ï¼ZIPå¤ä»½æ ¹ç®å½+æ¶é´æä»¶å+.zipï¼ |
| | |
| | | return true; |
| | | } |
| | | |
| | | private static void addFolderToZip(File folder, String parentPath, ZipOutputStream zos) throws IOException { |
| | | File[] files = folder.listFiles(); |
| | | if (files != null) { |
| | | for (File file : files) { |
| | | if (file.isDirectory()) { |
| | | // éå½å¤çåç®å½ï¼ä¿æè·¯å¾ç»æ |
| | | addFolderToZip(file, parentPath + file.getName() + "/", zos); |
| | | } else { |
| | | // æ·»å æä»¶å°ZIPï¼ä¿æè·¯å¾ç»æ |
| | | addFileToZip(file, parentPath + file.getName(), zos); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | |
| | | private static void addFileToZip(File file, String entryName, ZipOutputStream zos) throws IOException { |
| | | try (FileInputStream fis = new FileInputStream(file)) { |
| | | zos.putNextEntry(new ZipEntry(entryName)); |
| | |
| | | } |
| | | |
| | | /** |
| | | * å缩å¢ééä»¶ |
| | | * @param groupFolderFile å¢éæä»¶å¤¹ï¼æä»¶å |
| | | * @param baseDir éä»¶æ ¹ç®å½ |
| | | * @param outputZipPath å缩æä»¶ |
| | | * @throws IOException |
| | | * ç¬¬ä¸æ¥ï¼å缩å¤ä»½æä»¶ |
| | | */ |
| | | public static void compressExistingFiles(Map<Integer, List<String>> groupFolderFile, String baseDir, ZipOutputStream zos) throws IOException { |
| | | for (Map.Entry<Integer, List<String>> entry : groupFolderFile.entrySet()) { |
| | | int folderName = entry.getKey(); // æä»¶å¤¹åï¼å¦ 20250611ï¼ |
| | | List<String> files = entry.getValue(); // 该æä»¶å¤¹ä¸çæä»¶åå表 |
| | | Path folderPath = Paths.get(baseDir).resolve(String.valueOf(folderName)); |
| | | for (String fileName : files) { |
| | | Path filePath = folderPath.resolve(fileName); |
| | | // æ£æ¥æä»¶æ¯å¦åå¨ |
| | | if (Files.exists(filePath) && !Files.isDirectory(filePath)) { |
| | | // è®¡ç® ZIP å
çç¸å¯¹è·¯å¾ï¼å¦ "20250611/file1.txt"ï¼ |
| | | ZipEntry zipEntry = new ZipEntry( |
| | | String.valueOf(folderName) + "/" + fileName |
| | | ); |
| | | zos.putNextEntry(zipEntry); |
| | | |
| | | // åå
¥æä»¶å
å®¹å° ZIP |
| | | Files.copy(filePath, zos); |
| | | zos.closeEntry(); |
| | | } else { |
| | | System.err.println("æä»¶ä¸å卿䏿¯æä»¶: " + filePath); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | |
| | | public boolean zipDataBackup() { |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãå¼å§å缩å¤ä»½æä»¶..................", BackupLogger.INFO_TYPE); |
| | | String documentPath = config.getProperty("DOCUMENT_ROOT"); |
| | |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãéä»¶åå¨ç®å½è·¯å¾ï¼" + documentPath, BackupLogger.INFO_TYPE); |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãæ°æ®åºå¤ä»½å®æ´è·¯å¾ï¼" + databasePath, BackupLogger.INFO_TYPE); |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãå¤ä»½å缩æä»¶å®æ´è·¯å¾ï¼" + zipFilePath, BackupLogger.INFO_TYPE); |
| | | |
| | | try (FileOutputStream fos = new FileOutputStream(zipFilePath); |
| | | ZipOutputStream zos = new ZipOutputStream(fos)){ |
| | | StringBuilder sbSql = new StringBuilder(); |
| | | sbSql.append("SELECT SUBSTRING_INDEX(attachment_url, '/', -1) id,attachment_title AS uuid,attachment_url \n"); |
| | | sbSql.append("FROM product_sys_attachments \n"); |
| | | sbSql.append("WHERE (created_utc_datetime > ? AND created_utc_datetime <= ?) \n"); |
| | | sbSql.append("AND (updated_utc_datetime > ? AND updated_utc_datetime <= ?) \n"); |
| | | DataTableEntity dtTable = baseDao.listTable(sbSql.toString(), new Object[] {STANDARD_START_TIME, STANDARD_FINAL_TIME, STANDARD_START_TIME, STANDARD_FINAL_TIME}); |
| | | if(dtTable != null && dtTable.getRows() > 0) { |
| | | Map<Integer, List<String>> groupFolderFile = dtTable.getData().stream() |
| | | .collect(Collectors.groupingBy( |
| | | FieldSetEntity::getId, |
| | | Collectors.mapping( |
| | | FieldSetEntity::getUUID, |
| | | Collectors.toList() |
| | | ) |
| | | )); |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãå缩系ç»å¢ééä»¶", BackupLogger.INFO_TYPE); |
| | | compressExistingFiles(groupFolderFile, documentPath, zos); |
| | | } |
| | | |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãå缩系ç»éä»¶", BackupLogger.INFO_TYPE); |
| | | // å缩æä»¶å¤¹ï¼ä¿ç宿´è·¯å¾ç»æï¼ |
| | | // 注æï¼è¿éæä»¬ä¼ éäºæ ¹ç®å½åç§°ä½ä¸ºåå§parentPath |
| | | addFolderToZip(new File(documentPath), NUMBER_DATE + "/", zos); |
| | | |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãåç¼©æ°æ®åºå¤ä»½æä»¶", BackupLogger.INFO_TYPE); |
| | | // å缩SQLæä»¶ï¼æ¾å¨æå®è·¯å¾ä¸ï¼ |
| | | addFileToZip(new File(databasePath), "database/" + NUMBER_TIME + ".sql", zos); |
| | | |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãåç¼©å®æ", BackupLogger.INFO_TYPE); |
| | | } catch (IOException e) { |
| | | e.printStackTrace(); |
| | | } finally { |
| | | File file = new File(databasePath); |
| | | file.delete(); |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãæ¸
çæ°æ®åºå¤ä»½æä»¶å®æ", BackupLogger.INFO_TYPE); |
| | | } |
| | | log.writeInfo("ãå缩å¤ä»½æä»¶ãå缩失败ï¼" + e.getMessage(), BackupLogger.ERROR_TYPE); |
| | | return false; |
| | | } |
| | | return true; |
| | | } |
| | | |
| | | /** |
| | |
| | | * @param isUpload æ¯å¦ä¸ä¼ |
| | | * @return |
| | | */ |
| | | public boolean sftpTransferService() { |
| | | String sftpTitle = "ä¸ä¼ ç³»ç»å¤ä»½æä»¶";; |
| | | String localFilePath = config.getProperty("ZIPFILE_BACKUP"); // æ¬å°æä»¶è·¯å¾ |
| | | String sftpFilePath = config.getProperty("UPLOAD_BACKUP_DIR") + NUMBER_TIME + ".zip"; // SFTPæå¡æä»¶è·¯å¾; |
| | | |
| | | public boolean sftpTransferService(boolean isUpload) { |
| | | String sftpTitle = null; |
| | | String localFilePath = null; |
| | | String sftpFilePath = null; |
| | | if(isUpload) { |
| | | sftpTitle = "ä¸ä¼ ç³»ç»å¤ä»½æä»¶"; |
| | | localFilePath = config.getProperty("ZIPFILE_BACKUP"); // æ¬å°æä»¶è·¯å¾ |
| | | sftpFilePath = config.getProperty("UPLOAD_BACKUP_DIR") + NUMBER_TIME + ".zip"; // SFTPæå¡æä»¶è·¯å¾ |
| | | } else { |
| | | // sftpTitle = "è·åæ°æ®åºå¤ä»½æä»¶"; |
| | | // localFilePath = config.getProperty("DATABASE_BACKUP"); // æ¬å°æä»¶è·¯å¾ |
| | | // sftpFilePath = config.getProperty("DATABASE_BACKUP"); // SFTPæå¡æä»¶è·¯å¾ |
| | | } |
| | | String host = config.getProperty("UPLOAD_SFTP_HOST"); // SFTPæå¡IPå°å |
| | | String port = config.getProperty("UPLOAD_SFTP_PORT"); // SSHç«¯å£ |
| | | String user = config.getProperty("UPLOAD_SFTP_USER"); // SFTPæå¡çç¨æ·å |
| | |
| | | if (!remoteDir.startsWith("/")) { |
| | | remoteDir = "/" + remoteDir; |
| | | } |
| | | if(isUpload) { |
| | | // å建对åºç®å½æä»¶å¤¹ |
| | | createRemoteDirectory(sftpChannel, remoteDir); |
| | | sftpChannel.put(localFilePath, sftpFilePath, new SftpProgressMonitor() { |
| | |
| | | public boolean count(long count) { return true; } |
| | | public void end() { log.writeInfo("ãä¸ä¼ ç³»ç»å¤ä»½æä»¶ãä¼ è¾å®æ", BackupLogger.INFO_TYPE); } |
| | | }); |
| | | } else { |
| | | // å建对åºç®å½æä»¶å¤¹ |
| | | File fileDirectory = new File(remoteDir); |
| | | if (!fileDirectory.exists()) { |
| | | fileDirectory.mkdirs(); |
| | | } |
| | | sftpChannel.get(localFilePath, sftpFilePath, new SftpProgressMonitor() { |
| | | public void init(int op, String src, String dest, long max) { |
| | | log.writeInfo("ãè·åæ°æ®åºå¤ä»½æä»¶ãå¼å§ä¼ è¾: " + src + " -> " + dest, BackupLogger.INFO_TYPE); |
| | | } |
| | | public boolean count(long count) { return true; } |
| | | public void end() { log.writeInfo("ãè·åæ°æ®åºå¤ä»½æä»¶ãä¼ è¾å®æ", BackupLogger.INFO_TYPE); } |
| | | }); |
| | | } |
| | | sftpChannel.exit(); |
| | | session.disconnect(); |
| | | log.writeInfo(String.format("ã%sã%sæå", sftpTitle, sftpTitle), BackupLogger.INFO_TYPE); |
| | |
| | | } |
| | | |
| | | /** |
| | | * SFTPéå½å建è¿ç¨ç®å½ |
| | | * éå½å建è¿ç¨ç®å½ |
| | | * @param sftpChannel |
| | | * @param remoteDir |
| | | * @throws SftpException |