dendi-wang/thingsboard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clear interruption status after exiting consumer while loop

Browse Source
This commit is contained in:
ViacheslavKlimov 2023-10-19 14:00:04 +03:00
parent 5108e98f38
commit 049f40a62e
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
application/src/main/java/org/thingsboard/server/service/queue/ruleengine/TbQueueConsumerTask.java
View File

@ -41,8 +41,8 @@ public class TbQueueConsumerTask {
private CountDownLatch completionLatch;
public void setTask(Future<?> task) {
this.task = task;
this.completionLatch = new CountDownLatch(1);
this.task = task;
}
public void subscribe(Set<TopicPartitionInfo> partitions) {
@ -63,14 +63,14 @@ public class TbQueueConsumerTask {
if (isRunning()) {
try {
if (!completionLatch.await(30, TimeUnit.SECONDS)) {
task = null;
throw new IllegalStateException("timeout of 30 seconds expired");
}
log.trace("[{}] Awaited finish", key);
} catch (Exception e) {
log.warn("[{}] Failed to await for consumer to stop", key, e);
}
task = null;
}
log.trace("[{}] Awaited finish", key);
}
public boolean isRunning() {
@ -79,6 +79,7 @@ public class TbQueueConsumerTask {
public void finished() {
completionLatch.countDown();
task = null;
}
}

8
application/src/main/java/org/thingsboard/server/service/queue/ruleengine/TbRuleEngineQueueConsumerManager.java
View File

@ -237,7 +237,11 @@ public class TbRuleEngineQueueConsumerManager {
log.info("[{}] Launching consumer", consumerTask.getKey());
Future<?> consumerLoop = ctx.getConsumersExecutor().submit(() -> {
ThingsBoardThreadFactory.updateCurrentThreadName(consumerTask.getKey().toString());
consumerLoop(consumerTask.getConsumer());
try {
consumerLoop(consumerTask.getConsumer());
} catch (Throwable e) {
log.error("Failure in consumer loop", e);
}
consumerTask.finished();
});
consumerTask.setTask(consumerLoop);
@ -262,7 +266,7 @@ public class TbRuleEngineQueueConsumerManager {
}
}
}
if (consumer.isStopped()) {
if (Thread.interrupted() || consumer.isStopped()) {
consumer.unsubscribe();
}
log.info("Rule Engine consumer stopped");