Spider/spider-flow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修复输出 csv 文件被锁定的问题

Browse Source
This commit is contained in:
nekolr 2020-04-12 16:51:29 +08:00
parent 7b42f4c0a5
commit 0c0cdca38c
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spider-flow-core/src/main/java/org/spiderflow/core/executor/shape/OutputExecutor.java
View File

@ -170,6 +170,9 @@ public class OutputExecutor implements ShapeExecutor, SpiderListener {
List<String> records = new ArrayList<>(data.size());
String[] headers = data.keySet().toArray(new String[data.size()]);
try {
if (printer == null) {
synchronized (cachePrinter) {
printer = cachePrinter.get(key);
if (printer == null) {
CSVFormat format = CSVFormat.DEFAULT.withHeader(headers);
FileOutputStream os = new FileOutputStream(csvName);
@ -184,10 +187,12 @@ public class OutputExecutor implements ShapeExecutor, SpiderListener {
printer = new CSVPrinter(osw, format);
cachePrinter.put(key, printer);
}
}
}
for (int i = 0; i < headers.length; i++) {
records.add(data.get(headers[i]).toString());
}
synchronized (printer) {
synchronized (cachePrinter) {
printer.printRecord(records);
}
} catch (IOException e) {