Apache/nifi
1
0
Fork 0
mirror of https://github.com/apache/nifi.git synced 2025-02-07 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

NIFI-9741: Make the close() method of WriteAvroResultWithExternalSchema idempotent

Browse Source 
Signed-off-by: Matthew Burgess <mattyb149@apache.org>

This closes #5818
This commit is contained in:
Mark Payne 2022-03-01 11:25:03 -05:00 committed by Matthew Burgess
parent d17fcf6427
commit a327ba478a
No known key found for this signature in database
GPG Key ID: 05D3DEB8126DAD24
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nifi-nar-bundles/nifi-standard-services/nifi-record-serialization-services-bundle/nifi-record-serialization-services/src/main/java/org/apache/nifi/avro/WriteAvroResultWithExternalSchema.java
View File

@ -43,6 +43,7 @@ public class WriteAvroResultWithExternalSchema extends AbstractRecordSetWriter {
private final OutputStream buffered;
private final DatumWriter<GenericRecord> datumWriter;
private final BlockingQueue<BinaryEncoder> recycleQueue;
private boolean closed = false;
public WriteAvroResultWithExternalSchema(final Schema avroSchema, final RecordSchema recordSchema, final SchemaAccessWriter schemaAccessWriter,
final OutputStream out, final BlockingQueue<BinaryEncoder> recycleQueue, final ComponentLog logger) {
@ -102,6 +103,11 @@ public class WriteAvroResultWithExternalSchema extends AbstractRecordSetWriter {
@Override
public void close() throws IOException {
if (closed) {
return;
}
closed = true;
if (encoder != null) {
flush();
recycleQueue.offer(encoder);