代码优化提交
This commit is contained in:
+4
-4
@@ -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.scanner.service.config.ScanConfig;
|
||||
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 org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Service;
|
||||
@@ -28,7 +28,7 @@ public class MyBatisExportService {
|
||||
private FileOperationHelper fileOperationHelper;
|
||||
|
||||
@Autowired
|
||||
private OutputService outputService;
|
||||
private OutputHelper outputHelper;
|
||||
|
||||
private static final String[] SQL_TAG_TYPES = {"select", "insert", "update", "delete"};
|
||||
|
||||
@@ -98,8 +98,8 @@ public class MyBatisExportService {
|
||||
}
|
||||
}
|
||||
|
||||
String fileName = outputService.generateOutputFileName(fileNamePrefix, "csv");
|
||||
outputService.writeRecordsToCsv(allRecords, fileName, Arrays.asList(EXPORT_HEADERS));
|
||||
String fileName = outputHelper.generateOutputFileName(fileNamePrefix, "csv");
|
||||
outputHelper.writeRecordsToCsv(allRecords, fileName, Arrays.asList(EXPORT_HEADERS));
|
||||
|
||||
log.info("导出完成。已处理 {} 个项目,共 {} 条记录",
|
||||
processedCount.get(), allRecords.size());
|
||||
|
||||
@@ -3,7 +3,7 @@ package com.example.workutils.biz.scanner.service;
|
||||
import com.example.workutils.biz.scanner.service.config.ScanConfig;
|
||||
import com.example.workutils.system.config.WorkUtilsProperties;
|
||||
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 org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.stereotype.Service;
|
||||
@@ -26,7 +26,7 @@ public class ScannerService {
|
||||
@Autowired
|
||||
private FileOperationHelper fileOperationHelper;
|
||||
@Autowired
|
||||
private OutputService outputService;
|
||||
private OutputHelper outputHelper;
|
||||
@Autowired
|
||||
private WorkUtilsProperties properties;
|
||||
|
||||
@@ -186,7 +186,7 @@ public class ScannerService {
|
||||
}
|
||||
|
||||
fileOperationHelper.ensureDirectoryExists(properties.getTodayExportPath());
|
||||
outputService.writeRecordsToCsv(records, outputFileName);
|
||||
outputHelper.writeRecordsToCsv(records, outputFileName);
|
||||
log.info("Scan results exported to: {}", outputFileName);
|
||||
}
|
||||
|
||||
@@ -207,19 +207,19 @@ public class ScannerService {
|
||||
|
||||
public void scanKeywordsWithDefaultName(String pattern) throws Exception {
|
||||
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);
|
||||
}
|
||||
|
||||
public void scanRedisKeysWithDefaultName(String pattern) throws Exception {
|
||||
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);
|
||||
}
|
||||
|
||||
public void scanDubboInterfacesWithDefaultName(String pattern) throws Exception {
|
||||
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);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user