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

Merge pull request #3 from nekolr/dev

Browse Source 
修复输出 csv 文件被锁定的问题
This commit is contained in:
javamxd 2020-04-12 03:55:26 -05:00 committed by GitHub
commit 1404f03cc8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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) {