Apache
/
nifi
mirror of https://github.com/apache/nifi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

NIFI-7286 ListenTCPRecord cleanup changed from @OnStopped to @OnUnscheduled

This commit is contained in:
Tamas Palfy 2020-03-26 15:28:03 +01:00 committed by Bryan Bende
parent 483f23a8aa
commit c79473baf5
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/main/java/org/apache/nifi/processors/standard/ListenTCPRecord.java
View File

@ -25,7 +25,7 @@ import org.apache.nifi.annotation.behavior.WritesAttributes;
import org.apache.nifi.annotation.documentation.CapabilityDescription;
import org.apache.nifi.annotation.documentation.Tags;
import org.apache.nifi.annotation.lifecycle.OnScheduled;
import org.apache.nifi.annotation.lifecycle.OnStopped;
import org.apache.nifi.annotation.lifecycle.OnUnscheduled;
import org.apache.nifi.components.AllowableValue;
import org.apache.nifi.components.PropertyDescriptor;
import org.apache.nifi.components.ValidationContext;
@ -300,8 +300,8 @@ public class ListenTCPRecord extends AbstractProcessor {
readerThread.start();
}
@OnStopped
public void onStopped() {
@OnUnscheduled
public void onUnscheduled() {
if (dispatcher != null) {
dispatcher.close();
dispatcher = null;
@ -309,7 +309,11 @@ public class ListenTCPRecord extends AbstractProcessor {
SocketChannelRecordReader socketRecordReader;
while ((socketRecordReader = socketReaders.poll()) != null) {
IOUtils.closeQuietly(socketRecordReader.getRecordReader());
try {
socketRecordReader.close();
} catch (Exception e) {
getLogger().error("Couldn't close " + socketRecordReader, e);
}
}
}

2
nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestListenTCPRecord.java
View File

@ -258,7 +258,7 @@ public class TestListenTCPRecord {
runner.assertTransferCount(ListenTCPRecord.REL_SUCCESS, expectedTransferred);
} finally {
// unschedule to close connections
proc.onStopped();
proc.onUnscheduled();
IOUtils.closeQuietly(sender);
}
}