Skip to content

Commit 3e2629b

Browse files
committed
refactor processor run status handling and update error state return values
1 parent 6e15b54 commit 3e2629b

File tree

3 files changed

+10
-27
lines changed

3 files changed

+10
-27
lines changed

minifi/minifi-nar-bundles/minifi-framework-bundle/minifi-framework/minifi-framework-core/src/main/java/org/apache/nifi/minifi/c2/C2NifiClientService.java

Lines changed: 5 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -435,29 +435,14 @@ private ProcessorStatus convertProcessorStatus(org.apache.nifi.controller.status
435435
result.setProcessingNanos(processorStatus.getProcessingNanos());
436436
result.setActiveThreadCount(processorStatus.getActiveThreadCount());
437437
result.setTerminatedThreadCount(processorStatus.getTerminatedThreadCount());
438-
RunStatus mappedRunStatus = mapProcessorRunStatus(processorStatus.getRunStatus());
439-
if (mappedRunStatus != null) {
440-
result.setRunStatus(mappedRunStatus);
441-
}
438+
result.setRunStatus(
439+
processorStatus.getRunStatus() == org.apache.nifi.controller.status.RunStatus.Running
440+
? RUNNING
441+
: STOPPED
442+
);
442443
return result;
443444
}
444445

445-
private RunStatus mapProcessorRunStatus(org.apache.nifi.controller.status.RunStatus controllerRunStatus) {
446-
if (controllerRunStatus == null) {
447-
return null;
448-
}
449-
switch (controllerRunStatus) {
450-
case Running:
451-
return RUNNING;
452-
case Stopped:
453-
case Disabled:
454-
case Invalid:
455-
case Validating:
456-
return STOPPED;
457-
}
458-
return null;
459-
}
460-
461446
private RunStatus getRunStatus() {
462447
ProcessGroup processGroup = flowController.getFlowManager().getRootGroup();
463448
return isProcessGroupRunning(processGroup)

minifi/minifi-nar-bundles/minifi-framework-bundle/minifi-framework/minifi-framework-core/src/main/java/org/apache/nifi/minifi/c2/command/DefaultProcessorStateStrategy.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@
2727

2828
import static org.apache.nifi.c2.protocol.api.C2OperationState.OperationState.FULLY_APPLIED;
2929
import static org.apache.nifi.c2.protocol.api.C2OperationState.OperationState.NOT_APPLIED;
30-
import static org.apache.nifi.c2.protocol.api.C2OperationState.OperationState.PARTIALLY_APPLIED;
3130

3231
public class DefaultProcessorStateStrategy implements ProcessorStateStrategy {
3332

@@ -67,7 +66,7 @@ private OperationState changeState(String processorId, boolean start) {
6766
return FULLY_APPLIED;
6867
} catch (Exception e) {
6968
LOGGER.error("Failed to change state for processor {}", processorId, e);
70-
return PARTIALLY_APPLIED;
69+
return NOT_APPLIED;
7170
}
7271
}
7372
}

minifi/minifi-nar-bundles/minifi-framework-bundle/minifi-framework/minifi-framework-core/src/test/java/org/apache/nifi/minifi/c2/command/DefaultProcessorStateStrategyTest.java

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@
3030

3131
import static org.apache.nifi.c2.protocol.api.C2OperationState.OperationState.FULLY_APPLIED;
3232
import static org.apache.nifi.c2.protocol.api.C2OperationState.OperationState.NOT_APPLIED;
33-
import static org.apache.nifi.c2.protocol.api.C2OperationState.OperationState.PARTIALLY_APPLIED;
3433
import static org.junit.jupiter.api.Assertions.assertEquals;
3534
import static org.mockito.Mockito.doThrow;
3635
import static org.mockito.Mockito.mock;
@@ -79,15 +78,15 @@ public void testStartProcessorNotAppliedWhenMissing() {
7978
}
8079

8180
@Test
82-
public void testStartProcessorPartiallyAppliedOnError() {
81+
public void testStartProcessorNotAppliedOnError() {
8382
ProcessorNode node = mock(ProcessorNode.class);
8483
when(flowManager.getProcessorNode(PROCESSOR_ID)).thenReturn(node);
8584
when(node.getProcessGroupIdentifier()).thenReturn(GROUP_ID);
8685
doThrow(new RuntimeException("boom")).when(flowController).startProcessor(GROUP_ID, PROCESSOR_ID, true);
8786

8887
OperationState result = victim.startProcessor(PROCESSOR_ID);
8988

90-
assertEquals(PARTIALLY_APPLIED, result);
89+
assertEquals(NOT_APPLIED, result);
9190
}
9291

9392
@Test
@@ -103,15 +102,15 @@ public void testStopProcessorFullyApplied() {
103102
}
104103

105104
@Test
106-
public void testStopProcessorPartiallyAppliedOnError() {
105+
public void testStopProcessorNotAppliedOnError() {
107106
ProcessorNode node = mock(ProcessorNode.class);
108107
when(flowManager.getProcessorNode(PROCESSOR_ID)).thenReturn(node);
109108
when(node.getProcessGroupIdentifier()).thenReturn(GROUP_ID);
110109
doThrow(new RuntimeException("boom")).when(flowController).stopProcessor(GROUP_ID, PROCESSOR_ID);
111110

112111
OperationState result = victim.stopProcessor(PROCESSOR_ID);
113112

114-
assertEquals(PARTIALLY_APPLIED, result);
113+
assertEquals(NOT_APPLIED, result);
115114
}
116115

117116
@Test

0 commit comments

Comments
 (0)