yiciheng/workutils
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

代码优化提交

Browse Source
This commit is contained in:
张家训
2026-06-09 14:39:28 +08:00
parent c5773063d7
commit d14c2fbebf
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/example/workutils/biz/mybatisexport/service/MyBatisExportService.java
+4 -4
View File
@@ -3,7 +3,7 @@ package com.example.workutils.biz.mybatisexport.service;
import com.example.workutils.biz.mybatisexport.service.model.MybatisTag; import com.example.workutils.biz.mybatisexport.service.model.MybatisTag;
import com.example.workutils.biz.scanner.service.config.ScanConfig; import com.example.workutils.biz.scanner.service.config.ScanConfig;
import com.example.workutils.system.support.FileOperationHelper; import com.example.workutils.system.support.FileOperationHelper;
import com.example.workutils.system.support.OutputService; import com.example.workutils.system.support.OutputHelper;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@@ -28,7 +28,7 @@ public class MyBatisExportService {
private FileOperationHelper fileOperationHelper; private FileOperationHelper fileOperationHelper;
@Autowired @Autowired
private OutputService outputService; private OutputHelper outputHelper;
private static final String[] SQL_TAG_TYPES = {"select", "insert", "update", "delete"}; private static final String[] SQL_TAG_TYPES = {"select", "insert", "update", "delete"};
@@ -98,8 +98,8 @@ public class MyBatisExportService {
} }
} }
String fileName = outputService.generateOutputFileName(fileNamePrefix, "csv"); String fileName = outputHelper.generateOutputFileName(fileNamePrefix, "csv");
outputService.writeRecordsToCsv(allRecords, fileName, Arrays.asList(EXPORT_HEADERS)); outputHelper.writeRecordsToCsv(allRecords, fileName, Arrays.asList(EXPORT_HEADERS));
log.info("导出完成。已处理 {} 个项目，共 {} 条记录", log.info("导出完成。已处理 {} 个项目，共 {} 条记录",
processedCount.get(), allRecords.size()); processedCount.get(), allRecords.size());
src/main/java/com/example/workutils/biz/scanner/service/ScannerService.java
+6 -6
View File
@@ -3,7 +3,7 @@ package com.example.workutils.biz.scanner.service;
import com.example.workutils.biz.scanner.service.config.ScanConfig; import com.example.workutils.biz.scanner.service.config.ScanConfig;
import com.example.workutils.system.config.WorkUtilsProperties; import com.example.workutils.system.config.WorkUtilsProperties;
import com.example.workutils.system.support.FileOperationHelper; import com.example.workutils.system.support.FileOperationHelper;
import com.example.workutils.system.support.OutputService; import com.example.workutils.system.support.OutputHelper;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
@@ -26,7 +26,7 @@ public class ScannerService {
@Autowired @Autowired
private FileOperationHelper fileOperationHelper; private FileOperationHelper fileOperationHelper;
@Autowired @Autowired
private OutputService outputService; private OutputHelper outputHelper;
@Autowired @Autowired
private WorkUtilsProperties properties; private WorkUtilsProperties properties;
@@ -186,7 +186,7 @@ public class ScannerService {
} }
fileOperationHelper.ensureDirectoryExists(properties.getTodayExportPath()); fileOperationHelper.ensureDirectoryExists(properties.getTodayExportPath());
outputService.writeRecordsToCsv(records, outputFileName); outputHelper.writeRecordsToCsv(records, outputFileName);
log.info("Scan results exported to: {}", outputFileName); log.info("Scan results exported to: {}", outputFileName);
} }
@@ -207,19 +207,19 @@ public class ScannerService {
public void scanKeywordsWithDefaultName(String pattern) throws Exception { public void scanKeywordsWithDefaultName(String pattern) throws Exception {
ScanResult result = scanAllProjects(ScanType.KEYWORD, pattern); ScanResult result = scanAllProjects(ScanType.KEYWORD, pattern);
String defaultFileName = outputService.generateOutputFileName("keyword_scan_result", "csv"); String defaultFileName = outputHelper.generateOutputFileName("keyword_scan_result", "csv");
exportScanResults(result, defaultFileName); exportScanResults(result, defaultFileName);
} }
public void scanRedisKeysWithDefaultName(String pattern) throws Exception { public void scanRedisKeysWithDefaultName(String pattern) throws Exception {
ScanResult result = scanAllProjects(ScanType.REDIS_KEY, pattern); ScanResult result = scanAllProjects(ScanType.REDIS_KEY, pattern);
String defaultFileName = outputService.generateOutputFileName("redis_key_scan_result", "csv"); String defaultFileName = outputHelper.generateOutputFileName("redis_key_scan_result", "csv");
exportScanResults(result, defaultFileName); exportScanResults(result, defaultFileName);
} }
public void scanDubboInterfacesWithDefaultName(String pattern) throws Exception { public void scanDubboInterfacesWithDefaultName(String pattern) throws Exception {
ScanResult result = scanAllProjects(ScanType.DUBBO, pattern); ScanResult result = scanAllProjects(ScanType.DUBBO, pattern);
String defaultFileName = outputService.generateOutputFileName("dubbo_scan_result", "csv"); String defaultFileName = outputHelper.generateOutputFileName("dubbo_scan_result", "csv");
exportScanResults(result, defaultFileName); exportScanResults(result, defaultFileName);
} }