38 changed files with 609 additions and 61 deletions
@ -0,0 +1,98 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.queue; |
||||
|
|
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.params.ParameterizedTest; |
||||
|
import org.junit.jupiter.params.provider.Arguments; |
||||
|
import org.junit.jupiter.params.provider.MethodSource; |
||||
|
import org.thingsboard.server.common.data.EntityType; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.rule.RuleNode; |
||||
|
import org.thingsboard.server.common.msg.queue.RuleEngineException; |
||||
|
import org.thingsboard.server.common.msg.queue.RuleNodeException; |
||||
|
import org.thingsboard.server.common.msg.tools.TbRateLimitsException; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
import java.util.stream.Stream; |
||||
|
|
||||
|
import static org.mockito.ArgumentMatchers.any; |
||||
|
import static org.mockito.Mockito.mock; |
||||
|
import static org.mockito.Mockito.never; |
||||
|
import static org.mockito.Mockito.spy; |
||||
|
import static org.mockito.Mockito.verify; |
||||
|
|
||||
|
class TbMsgPackCallbackTest { |
||||
|
|
||||
|
TenantId tenantId; |
||||
|
UUID msgId; |
||||
|
TbMsgPackProcessingContext ctx; |
||||
|
TbMsgPackCallback callback; |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
tenantId = TenantId.fromUUID(UUID.randomUUID()); |
||||
|
msgId = UUID.randomUUID(); |
||||
|
ctx = mock(TbMsgPackProcessingContext.class); |
||||
|
callback = spy(new TbMsgPackCallback(msgId, tenantId, ctx)); |
||||
|
} |
||||
|
|
||||
|
private static Stream<Arguments> testOnFailure_NotRateLimitException() { |
||||
|
return Stream.of( |
||||
|
Arguments.of(new RuleEngineException("rule engine no cause")), |
||||
|
Arguments.of(new RuleEngineException("rule engine caused 1 lvl", new RuntimeException())), |
||||
|
Arguments.of(new RuleEngineException("rule engine caused 2 lvl", new RuntimeException(new Exception()))), |
||||
|
Arguments.of(new RuleEngineException("rule engine caused 2 lvl Throwable", new RuntimeException(new Throwable()))), |
||||
|
Arguments.of(new RuleNodeException("rule node no cause", "RuleChain", new RuleNode())) |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
@ParameterizedTest |
||||
|
@MethodSource |
||||
|
void testOnFailure_NotRateLimitException(RuleEngineException ree) { |
||||
|
callback.onFailure(ree); |
||||
|
|
||||
|
verify(callback, never()).onRateLimit(any()); |
||||
|
verify(callback, never()).onSuccess(); |
||||
|
verify(ctx, never()).onSuccess(any()); |
||||
|
} |
||||
|
|
||||
|
private static Stream<Arguments> testOnFailure_RateLimitException() { |
||||
|
return Stream.of( |
||||
|
Arguments.of(new RuleEngineException("caused lvl 1", new TbRateLimitsException(EntityType.ASSET))), |
||||
|
Arguments.of(new RuleEngineException("caused lvl 2", new RuntimeException(new TbRateLimitsException(EntityType.ASSET)))), |
||||
|
Arguments.of( |
||||
|
new RuleEngineException("caused lvl 3", |
||||
|
new RuntimeException( |
||||
|
new Exception( |
||||
|
new TbRateLimitsException(EntityType.ASSET))))) |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
@ParameterizedTest |
||||
|
@MethodSource |
||||
|
void testOnFailure_RateLimitException(RuleEngineException ree) { |
||||
|
callback.onFailure(ree); |
||||
|
|
||||
|
verify(callback).onRateLimit(any()); |
||||
|
verify(callback).onFailure(any()); |
||||
|
verify(callback, never()).onSuccess(); |
||||
|
verify(ctx).onSuccess(msgId); |
||||
|
verify(ctx).onSuccess(any()); |
||||
|
verify(ctx, never()).onFailure(any(), any(), any()); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,41 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.exception; |
||||
|
|
||||
|
public abstract class AbstractRateLimitException extends RuntimeException { |
||||
|
|
||||
|
public AbstractRateLimitException() { |
||||
|
super(); |
||||
|
} |
||||
|
|
||||
|
public AbstractRateLimitException(String message) { |
||||
|
super(message); |
||||
|
} |
||||
|
|
||||
|
public AbstractRateLimitException(String message, Throwable cause) { |
||||
|
super(message, cause); |
||||
|
} |
||||
|
|
||||
|
public AbstractRateLimitException(Throwable cause) { |
||||
|
super(cause); |
||||
|
} |
||||
|
|
||||
|
protected AbstractRateLimitException(String message, Throwable cause, |
||||
|
boolean enableSuppression, |
||||
|
boolean writableStackTrace) { |
||||
|
super(message, cause, enableSuppression, writableStackTrace); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,54 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.queue.kafka; |
||||
|
|
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.kafka.common.header.Header; |
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.thingsboard.server.queue.TbQueueMsg; |
||||
|
|
||||
|
import java.nio.charset.StandardCharsets; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.mockito.ArgumentMatchers.any; |
||||
|
import static org.mockito.BDDMockito.willCallRealMethod; |
||||
|
import static org.mockito.BDDMockito.willReturn; |
||||
|
import static org.mockito.Mockito.mock; |
||||
|
|
||||
|
@Slf4j |
||||
|
class TbKafkaProducerTemplateTest { |
||||
|
|
||||
|
TbKafkaProducerTemplate<TbQueueMsg> producerTemplate; |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
producerTemplate = mock(TbKafkaProducerTemplate.class); |
||||
|
willCallRealMethod().given(producerTemplate).addAnalyticHeaders(any()); |
||||
|
willReturn("tb-core-to-core-notifications-tb-core-3").given(producerTemplate).getClientId(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testAddAnalyticHeaders() { |
||||
|
List<Header> headers = new ArrayList<>(); |
||||
|
producerTemplate.addAnalyticHeaders(headers); |
||||
|
assertThat(headers).isNotEmpty(); |
||||
|
headers.forEach(r -> log.info("RecordHeader key [{}] value [{}]", r.key(), new String(r.value(), StandardCharsets.UTF_8))); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,20 @@ |
|||||
|
<?xml version="1.0" encoding="UTF-8" ?> |
||||
|
|
||||
|
<configuration> |
||||
|
<appender name="console" class="ch.qos.logback.core.ConsoleAppender"> |
||||
|
<encoder> |
||||
|
<pattern>%d{ISO8601} [%thread] %-5level %logger{36} - %msg%n</pattern> |
||||
|
</encoder> |
||||
|
</appender> |
||||
|
|
||||
|
<!-- TbKafkaProducerTemplate will add headers for each message when log level: |
||||
|
- DEBUG - producerId and thread name |
||||
|
- TRACE - will add stacktrace. |
||||
|
Kafka compression is highly recommended --> |
||||
|
<logger name="org.thingsboard.server.queue.kafka.TbKafkaProducerTemplate" level="TRACE"/> |
||||
|
|
||||
|
<root level="INFO"> |
||||
|
<appender-ref ref="console"/> |
||||
|
</root> |
||||
|
|
||||
|
</configuration> |
||||
@ -0,0 +1,67 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.common.util; |
||||
|
|
||||
|
import com.google.gson.JsonParseException; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.thingsboard.server.common.data.StringUtils; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
|
||||
|
import javax.script.ScriptException; |
||||
|
import java.io.PrintWriter; |
||||
|
import java.io.StringWriter; |
||||
|
|
||||
|
@Slf4j |
||||
|
public class ExceptionUtil { |
||||
|
|
||||
|
@SuppressWarnings("unchecked") |
||||
|
public static <T extends Exception> T lookupException(Throwable source, Class<T> clazz) { |
||||
|
Exception e = lookupExceptionInCause(source, clazz); |
||||
|
if (e != null) { |
||||
|
return (T) e; |
||||
|
} else { |
||||
|
return null; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public static Exception lookupExceptionInCause(Throwable source, Class<? extends Exception>... clazzes) { |
||||
|
while (source != null) { |
||||
|
for (Class<? extends Exception> clazz : clazzes) { |
||||
|
if (clazz.isAssignableFrom(source.getClass())) { |
||||
|
return (Exception) source; |
||||
|
} |
||||
|
} |
||||
|
source = source.getCause(); |
||||
|
} |
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
public static String toString(Exception e, EntityId componentId, boolean stackTraceEnabled) { |
||||
|
Exception exception = lookupExceptionInCause(e, ScriptException.class, JsonParseException.class); |
||||
|
if (exception != null && StringUtils.isNotEmpty(exception.getMessage())) { |
||||
|
return exception.getMessage(); |
||||
|
} else { |
||||
|
if (stackTraceEnabled) { |
||||
|
StringWriter sw = new StringWriter(); |
||||
|
e.printStackTrace(new PrintWriter(sw)); |
||||
|
return sw.toString(); |
||||
|
} else { |
||||
|
log.debug("[{}] Unknown error during message processing", componentId, e); |
||||
|
return "Please contact system administrator"; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,74 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.common.util; |
||||
|
|
||||
|
import org.junit.jupiter.api.Test; |
||||
|
|
||||
|
import java.io.IOException; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
|
||||
|
class ExceptionUtilTest { |
||||
|
|
||||
|
final Exception cause = new RuntimeException(); |
||||
|
|
||||
|
@Test |
||||
|
void givenRootCause_whenLookupExceptionInCause_thenReturnRootCauseAndNoStackOverflow() { |
||||
|
Exception e = cause; |
||||
|
for (int i = 0; i <= 16384; i++) { |
||||
|
e = new Exception(e); |
||||
|
} |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(e, RuntimeException.class)).isSameAs(cause); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void givenCause_whenLookupExceptionInCause_thenReturnCause() { |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(cause), RuntimeException.class)).isSameAs(cause); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void givenNoCauseAndExceptionIsWantedCauseClass_whenLookupExceptionInCause_thenReturnSelf() { |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(cause, RuntimeException.class)).isSameAs(cause); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void givenNoCause_whenLookupExceptionInCause_thenReturnNull() { |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(), RuntimeException.class)).isNull(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void givenNotWantedCause_whenLookupExceptionInCause_thenReturnNull() { |
||||
|
final Exception cause = new IOException(); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(cause), RuntimeException.class)).isNull(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void givenCause_whenLookupExceptionInCauseByMany_thenReturnFirstCause() { |
||||
|
final Exception causeIAE = new IllegalAccessException(); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIAE))).isNull(); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIAE), IOException.class, NoSuchFieldException.class)).isNull(); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIAE), IllegalAccessException.class, IOException.class, NoSuchFieldException.class)).isSameAs(causeIAE); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIAE), IOException.class, NoSuchFieldException.class, IllegalAccessException.class)).isSameAs(causeIAE); |
||||
|
|
||||
|
final Exception causeIOE = new IOException(causeIAE); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIOE))).isNull(); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIAE), ClassNotFoundException.class, NoSuchFieldException.class)).isNull(); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIOE), IOException.class, NoSuchFieldException.class)).isSameAs(causeIOE); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIOE), IllegalAccessException.class, IOException.class, NoSuchFieldException.class)).isSameAs(causeIOE); |
||||
|
assertThat(ExceptionUtil.lookupExceptionInCause(new Exception(causeIOE), IOException.class, NoSuchFieldException.class, IllegalAccessException.class)).isSameAs(causeIOE); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue