Andrii Shvaika
3 years ago
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