Igor Kulikov
3 years ago
110 changed files with 2030 additions and 314 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,161 @@ |
|||||
|
/** |
||||
|
* 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.transport.coap; |
||||
|
|
||||
|
import org.eclipse.californium.core.coap.CoAP; |
||||
|
import org.eclipse.californium.core.coap.OptionSet; |
||||
|
import org.eclipse.californium.core.coap.Request; |
||||
|
import org.junit.jupiter.api.BeforeAll; |
||||
|
import org.junit.jupiter.params.ParameterizedTest; |
||||
|
import org.junit.jupiter.params.provider.Arguments; |
||||
|
import org.junit.jupiter.params.provider.MethodSource; |
||||
|
import org.thingsboard.server.coapserver.CoapServerService; |
||||
|
import org.thingsboard.server.common.data.StringUtils; |
||||
|
import org.thingsboard.server.common.msg.session.FeatureType; |
||||
|
import org.thingsboard.server.common.transport.TransportService; |
||||
|
import org.thingsboard.server.queue.scheduler.SchedulerComponent; |
||||
|
import org.thingsboard.server.transport.coap.client.CoapClientContext; |
||||
|
|
||||
|
import java.util.Random; |
||||
|
import java.util.stream.Stream; |
||||
|
|
||||
|
import static org.junit.jupiter.api.Assertions.assertEquals; |
||||
|
import static org.junit.jupiter.api.Assertions.assertTrue; |
||||
|
import static org.mockito.Mockito.mock; |
||||
|
import static org.mockito.Mockito.when; |
||||
|
|
||||
|
class CoapTransportResourceTest { |
||||
|
|
||||
|
private static final String V1 = "v1"; |
||||
|
private static final String API = "api"; |
||||
|
private static final String TELEMETRY = "telemetry"; |
||||
|
private static final String ATTRIBUTES = "attributes"; |
||||
|
private static final String RPC = "rpc"; |
||||
|
private static final String CLAIM = "claim"; |
||||
|
private static final String PROVISION = "provision"; |
||||
|
private static final String GET_ATTRIBUTES_URI_QUERY = "clientKeys=attribute1,attribute2&sharedKeys=shared1,shared2"; |
||||
|
|
||||
|
private static final Random RANDOM = new Random(); |
||||
|
|
||||
|
private static CoapTransportResource coapTransportResource; |
||||
|
|
||||
|
@BeforeAll |
||||
|
static void setUp() { |
||||
|
|
||||
|
var ctxMock = mock(CoapTransportContext.class); |
||||
|
var coapServerServiceMock = mock(CoapServerService.class); |
||||
|
var transportServiceMock = mock(TransportService.class); |
||||
|
var clientContextMock = mock(CoapClientContext.class); |
||||
|
var schedulerComponentMock = mock(SchedulerComponent.class); |
||||
|
|
||||
|
when(ctxMock.getTransportService()).thenReturn(transportServiceMock); |
||||
|
when(ctxMock.getClientContext()).thenReturn(clientContextMock); |
||||
|
when(ctxMock.getSessionReportTimeout()).thenReturn(1L); |
||||
|
when(ctxMock.getScheduler()).thenReturn(schedulerComponentMock); |
||||
|
|
||||
|
coapTransportResource = new CoapTransportResource(ctxMock, coapServerServiceMock, V1); |
||||
|
} |
||||
|
|
||||
|
@ParameterizedTest |
||||
|
@MethodSource("provideRequestAndFeatureType") |
||||
|
void givenRequest_whenGetFeatureType_thenReturnedExpectedFeatureType(Request request, FeatureType expectedFeatureType) { |
||||
|
var featureTypeOptional = coapTransportResource.getFeatureType(request); |
||||
|
|
||||
|
assertTrue(featureTypeOptional.isPresent(), "Optional<FeatureType> is empty"); |
||||
|
assertEquals(expectedFeatureType, featureTypeOptional.get(), "Feature type is invalid"); |
||||
|
} |
||||
|
|
||||
|
static Stream<Arguments> provideRequestAndFeatureType() { |
||||
|
return Stream.of( |
||||
|
// accessToken based tests
|
||||
|
Arguments.of(toAccessTokenRequest(CoAP.Code.POST, TELEMETRY), FeatureType.TELEMETRY), |
||||
|
Arguments.of(toAccessTokenRequest(CoAP.Code.POST, ATTRIBUTES), FeatureType.ATTRIBUTES), |
||||
|
Arguments.of(toGetAttributesAccessTokenRequest(), FeatureType.ATTRIBUTES), |
||||
|
Arguments.of(toAccessTokenRequest(CoAP.Code.GET, ATTRIBUTES), FeatureType.ATTRIBUTES), |
||||
|
Arguments.of(toAccessTokenRequest(CoAP.Code.GET, RPC), FeatureType.RPC), |
||||
|
Arguments.of(toRpcResponseAccessTokenRequest(), FeatureType.RPC), |
||||
|
Arguments.of(toAccessTokenRequest(CoAP.Code.POST, RPC), FeatureType.RPC), |
||||
|
Arguments.of(toAccessTokenRequest(CoAP.Code.POST, CLAIM), FeatureType.CLAIM), |
||||
|
// certificate based tests
|
||||
|
Arguments.of(toCertificateRequest(CoAP.Code.POST, TELEMETRY), FeatureType.TELEMETRY), |
||||
|
Arguments.of(toCertificateRequest(CoAP.Code.POST, ATTRIBUTES), FeatureType.ATTRIBUTES), |
||||
|
Arguments.of(toGetAttributesCertificateRequest(), FeatureType.ATTRIBUTES), |
||||
|
Arguments.of(toCertificateRequest(CoAP.Code.GET, ATTRIBUTES), FeatureType.ATTRIBUTES), |
||||
|
Arguments.of(toCertificateRequest(CoAP.Code.GET, RPC), FeatureType.RPC), |
||||
|
Arguments.of(toRpcResponseCertificateRequest(), FeatureType.RPC), |
||||
|
Arguments.of(toCertificateRequest(CoAP.Code.POST, RPC), FeatureType.RPC), |
||||
|
Arguments.of(toCertificateRequest(CoAP.Code.POST, CLAIM), FeatureType.CLAIM), |
||||
|
// provision request
|
||||
|
Arguments.of(toProvisionRequest(), FeatureType.PROVISION) |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
private static Request toAccessTokenRequest(CoAP.Code method, String featureType) { |
||||
|
return getAccessTokenRequest(method, featureType, null, null); |
||||
|
} |
||||
|
|
||||
|
private static Request toGetAttributesAccessTokenRequest() { |
||||
|
return getAccessTokenRequest(CoAP.Code.GET, CoapTransportResourceTest.ATTRIBUTES, null, CoapTransportResourceTest.GET_ATTRIBUTES_URI_QUERY); |
||||
|
} |
||||
|
|
||||
|
private static Request toRpcResponseAccessTokenRequest() { |
||||
|
return getAccessTokenRequest(CoAP.Code.POST, CoapTransportResourceTest.RPC, RANDOM.nextInt(100), null); |
||||
|
} |
||||
|
|
||||
|
private static Request toCertificateRequest(CoAP.Code method, String featureType) { |
||||
|
return getCertificateRequest(method, featureType, null, null); |
||||
|
} |
||||
|
|
||||
|
private static Request toGetAttributesCertificateRequest() { |
||||
|
return getCertificateRequest(CoAP.Code.GET, CoapTransportResourceTest.ATTRIBUTES, null, CoapTransportResourceTest.GET_ATTRIBUTES_URI_QUERY); |
||||
|
} |
||||
|
|
||||
|
private static Request toRpcResponseCertificateRequest() { |
||||
|
return getCertificateRequest(CoAP.Code.POST, CoapTransportResourceTest.RPC, RANDOM.nextInt(100), null); |
||||
|
} |
||||
|
|
||||
|
private static Request getAccessTokenRequest(CoAP.Code method, String featureType, Integer requestId, String uriQuery) { |
||||
|
return getRequest(method, featureType, false, requestId, uriQuery); |
||||
|
} |
||||
|
|
||||
|
private static Request getCertificateRequest(CoAP.Code method, String featureType, Integer requestId, String uriQuery) { |
||||
|
return getRequest(method, featureType, true, requestId, uriQuery); |
||||
|
} |
||||
|
|
||||
|
private static Request toProvisionRequest() { |
||||
|
return getRequest(CoAP.Code.POST, PROVISION, true, null, null); |
||||
|
} |
||||
|
|
||||
|
private static Request getRequest(CoAP.Code method, String featureType, boolean dtls, Integer requestId, String uriQuery) { |
||||
|
var request = new Request(method); |
||||
|
var options = new OptionSet(); |
||||
|
options.addUriPath(API); |
||||
|
options.addUriPath(V1); |
||||
|
if (!dtls) { |
||||
|
options.addUriPath(StringUtils.randomAlphanumeric(20)); |
||||
|
} |
||||
|
options.addUriPath(featureType); |
||||
|
if (requestId != null) { |
||||
|
options.addUriPath(String.valueOf(requestId)); |
||||
|
} |
||||
|
if (uriQuery != null) { |
||||
|
options.setUriQuery(uriQuery); |
||||
|
} |
||||
|
request.setOptions(options); |
||||
|
return request; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -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); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,68 @@ |
|||||
|
<!-- |
||||
|
|
||||
|
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. |
||||
|
|
||||
|
--> |
||||
|
|
||||
|
<mat-toolbar> |
||||
|
<h2>{{ "attribute.delete-timeseries.delete-strategy" | translate }}</h2> |
||||
|
<span fxFlex></span> |
||||
|
<button mat-icon-button |
||||
|
type="button" |
||||
|
(click)="cancel()"> |
||||
|
<mat-icon class="material-icons">close</mat-icon> |
||||
|
</button> |
||||
|
</mat-toolbar> |
||||
|
<form [formGroup]="deleteTimeseriesFormGroup" class="mat-content mat-padding tb-form-settings"> |
||||
|
<mat-form-field class="mat-block" subscriptSizing="dynamic"> |
||||
|
<mat-label translate>attribute.delete-timeseries.strategy</mat-label> |
||||
|
<mat-select formControlName="strategy"> |
||||
|
<mat-option *ngFor="let strategy of strategiesTranslationsMap.keys()" [value]="strategy"> |
||||
|
{{ strategiesTranslationsMap.get(strategy) | translate }} |
||||
|
</mat-option> |
||||
|
</mat-select> |
||||
|
</mat-form-field> |
||||
|
<div *ngIf="isPeriodStrategy()" class="tb-select-interval"> |
||||
|
<mat-form-field class="mat-block" subscriptSizing="dynamic"> |
||||
|
<mat-label translate>attribute.delete-timeseries.start-time</mat-label> |
||||
|
<mat-datetimepicker-toggle [for]="startDateTimePicker" matPrefix></mat-datetimepicker-toggle> |
||||
|
<mat-datetimepicker #startDateTimePicker type="datetime" openOnFocus="true"></mat-datetimepicker> |
||||
|
<input required matInput formControlName="startDateTime" [matDatetimepicker]="startDateTimePicker"> |
||||
|
</mat-form-field> |
||||
|
<mat-form-field class="mat-block" subscriptSizing="dynamic"> |
||||
|
<mat-label translate>attribute.delete-timeseries.ends-on</mat-label> |
||||
|
<mat-datetimepicker-toggle [for]="endDatePicker" matPrefix></mat-datetimepicker-toggle> |
||||
|
<mat-datetimepicker #endDatePicker type="datetime" openOnFocus="true"></mat-datetimepicker> |
||||
|
<input required matInput formControlName="endDateTime" [matDatetimepicker]="endDatePicker"> |
||||
|
</mat-form-field> |
||||
|
</div> |
||||
|
<mat-slide-toggle class="tb-slide-toggle" formControlName="rewriteLatest" *ngIf="isPeriodStrategy() || isDeleteLatestStrategy()"> |
||||
|
{{ "attribute.delete-timeseries.rewrite-latest-value" | translate }} |
||||
|
</mat-slide-toggle> |
||||
|
</form> |
||||
|
<div fxLayout="row" class="tb-panel-actions"> |
||||
|
<span fxFlex></span> |
||||
|
<button mat-button color="primary" |
||||
|
type="button" |
||||
|
(click)="cancel()" cdkFocusInitial> |
||||
|
{{ 'action.cancel' | translate }} |
||||
|
</button> |
||||
|
<button mat-button mat-raised-button color="primary" |
||||
|
type="button" |
||||
|
(click)="delete()" |
||||
|
[disabled]="deleteTimeseriesFormGroup.invalid"> |
||||
|
{{ 'action.apply' | translate }} |
||||
|
</button> |
||||
|
</div> |
||||
@ -0,0 +1,44 @@ |
|||||
|
/** |
||||
|
* 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. |
||||
|
*/ |
||||
|
@import '../../../../../scss/constants'; |
||||
|
|
||||
|
:host { |
||||
|
width: 100%; |
||||
|
|
||||
|
.mat-toolbar { |
||||
|
background: none; |
||||
|
} |
||||
|
|
||||
|
.tb-form-settings { |
||||
|
flex-direction: column; |
||||
|
gap: 16px; |
||||
|
padding-top: 0; |
||||
|
} |
||||
|
|
||||
|
.tb-select-interval { |
||||
|
display: flex; |
||||
|
flex-direction: row; |
||||
|
gap: 16px; |
||||
|
@media #{$mat-xs} { |
||||
|
flex-direction: column; |
||||
|
gap: 0; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
.tb-slide-toggle { |
||||
|
margin-bottom: 8px; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,151 @@ |
|||||
|
///
|
||||
|
/// 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.
|
||||
|
///
|
||||
|
|
||||
|
import { Component, Inject, InjectionToken, OnDestroy, OnInit } from '@angular/core'; |
||||
|
import { OverlayRef } from '@angular/cdk/overlay'; |
||||
|
import { |
||||
|
TimeseriesDeleteStrategy, |
||||
|
timeseriesDeleteStrategyTranslations |
||||
|
} from '@shared/models/telemetry/telemetry.models'; |
||||
|
import { MINUTE } from '@shared/models/time/time.models'; |
||||
|
import { FormBuilder, FormGroup, Validators } from '@angular/forms'; |
||||
|
import { Subject } from 'rxjs'; |
||||
|
import { takeUntil } from 'rxjs/operators'; |
||||
|
|
||||
|
export const DELETE_TIMESERIES_PANEL_DATA = new InjectionToken<any>('DeleteTimeseriesPanelData'); |
||||
|
|
||||
|
export interface DeleteTimeseriesPanelData { |
||||
|
isMultipleDeletion: boolean; |
||||
|
} |
||||
|
|
||||
|
export interface DeleteTimeseriesPanelResult { |
||||
|
strategy: TimeseriesDeleteStrategy; |
||||
|
startDateTime: Date; |
||||
|
endDateTime: Date; |
||||
|
rewriteLatest: boolean; |
||||
|
} |
||||
|
|
||||
|
@Component({ |
||||
|
selector: 'tb-delete-timeseries-panel', |
||||
|
templateUrl: './delete-timeseries-panel.component.html', |
||||
|
styleUrls: ['./delete-timeseries-panel.component.scss'] |
||||
|
}) |
||||
|
export class DeleteTimeseriesPanelComponent implements OnInit, OnDestroy { |
||||
|
|
||||
|
deleteTimeseriesFormGroup: FormGroup; |
||||
|
|
||||
|
result: DeleteTimeseriesPanelResult = null; |
||||
|
|
||||
|
strategiesTranslationsMap = timeseriesDeleteStrategyTranslations; |
||||
|
|
||||
|
private multipleDeletionStrategies = new Set<TimeseriesDeleteStrategy>([ |
||||
|
TimeseriesDeleteStrategy.DELETE_ALL_DATA, |
||||
|
TimeseriesDeleteStrategy.DELETE_ALL_DATA_EXCEPT_LATEST_VALUE |
||||
|
]); |
||||
|
|
||||
|
private destroy$ = new Subject<void>(); |
||||
|
|
||||
|
constructor(@Inject(DELETE_TIMESERIES_PANEL_DATA) private data: DeleteTimeseriesPanelData, |
||||
|
private overlayRef: OverlayRef, |
||||
|
private fb: FormBuilder) { } |
||||
|
|
||||
|
ngOnInit(): void { |
||||
|
const today = new Date(); |
||||
|
if (this.data.isMultipleDeletion) { |
||||
|
this.strategiesTranslationsMap = new Map(Array.from(this.strategiesTranslationsMap) |
||||
|
.filter(([strategy]) => this.multipleDeletionStrategies.has(strategy))) |
||||
|
} |
||||
|
this.deleteTimeseriesFormGroup = this.fb.group({ |
||||
|
strategy: [TimeseriesDeleteStrategy.DELETE_ALL_DATA], |
||||
|
startDateTime: [ |
||||
|
{ value: new Date(today.getFullYear(), today.getMonth() - 1, today.getDate()), disabled: true }, |
||||
|
Validators.required |
||||
|
], |
||||
|
endDateTime: [{ value: today, disabled: true }, Validators.required], |
||||
|
rewriteLatest: [true] |
||||
|
}) |
||||
|
|
||||
|
this.deleteTimeseriesFormGroup.get('strategy').valueChanges.pipe( |
||||
|
takeUntil(this.destroy$) |
||||
|
).subscribe(value => { |
||||
|
if (value === TimeseriesDeleteStrategy.DELETE_ALL_DATA_FOR_TIME_PERIOD) { |
||||
|
this.deleteTimeseriesFormGroup.get('startDateTime').enable({onlySelf: true, emitEvent: false}); |
||||
|
this.deleteTimeseriesFormGroup.get('endDateTime').enable({onlySelf: true, emitEvent: false}); |
||||
|
} else { |
||||
|
this.deleteTimeseriesFormGroup.get('startDateTime').disable({onlySelf: true, emitEvent: false}); |
||||
|
this.deleteTimeseriesFormGroup.get('endDateTime').disable({onlySelf: true, emitEvent: false}); |
||||
|
} |
||||
|
}) |
||||
|
this.deleteTimeseriesFormGroup.get('startDateTime').valueChanges.pipe( |
||||
|
takeUntil(this.destroy$) |
||||
|
).subscribe(value => this.onStartDateTimeChange(value)); |
||||
|
this.deleteTimeseriesFormGroup.get('endDateTime').valueChanges.pipe( |
||||
|
takeUntil(this.destroy$) |
||||
|
).subscribe(value => this.onEndDateTimeChange(value)); |
||||
|
} |
||||
|
|
||||
|
ngOnDestroy(): void { |
||||
|
this.destroy$.next(); |
||||
|
this.destroy$.complete(); |
||||
|
} |
||||
|
|
||||
|
delete(): void { |
||||
|
if (this.deleteTimeseriesFormGroup.valid) { |
||||
|
this.result = this.deleteTimeseriesFormGroup.value; |
||||
|
this.overlayRef.dispose(); |
||||
|
} else { |
||||
|
this.deleteTimeseriesFormGroup.markAllAsTouched(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
cancel(): void { |
||||
|
this.overlayRef.dispose(); |
||||
|
} |
||||
|
|
||||
|
isPeriodStrategy(): boolean { |
||||
|
return this.deleteTimeseriesFormGroup.get('strategy').value === TimeseriesDeleteStrategy.DELETE_ALL_DATA_FOR_TIME_PERIOD; |
||||
|
} |
||||
|
|
||||
|
isDeleteLatestStrategy(): boolean { |
||||
|
return this.deleteTimeseriesFormGroup.get('strategy').value === TimeseriesDeleteStrategy.DELETE_LATEST_VALUE; |
||||
|
} |
||||
|
|
||||
|
private onStartDateTimeChange(newStartDateTime: Date) { |
||||
|
if (newStartDateTime) { |
||||
|
const endDateTimeTs = this.deleteTimeseriesFormGroup.get('endDateTime').value.getTime(); |
||||
|
if (newStartDateTime.getTime() >= endDateTimeTs) { |
||||
|
this.deleteTimeseriesFormGroup.get('startDateTime') |
||||
|
.patchValue(new Date(endDateTimeTs - MINUTE), {onlySelf: true, emitEvent: false}); |
||||
|
} else { |
||||
|
this.deleteTimeseriesFormGroup.get('startDateTime') |
||||
|
.patchValue(newStartDateTime, {onlySelf: true, emitEvent: false}); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private onEndDateTimeChange(newEndDateTime: Date) { |
||||
|
if (newEndDateTime) { |
||||
|
const startDateTimeTs = this.deleteTimeseriesFormGroup.get('startDateTime').value.getTime(); |
||||
|
if (newEndDateTime.getTime() <= startDateTimeTs) { |
||||
|
this.deleteTimeseriesFormGroup.get('endDateTime') |
||||
|
.patchValue(new Date(startDateTimeTs + MINUTE), {onlySelf: true, emitEvent: false}); |
||||
|
} else { |
||||
|
this.deleteTimeseriesFormGroup.get('endDateTime') |
||||
|
.patchValue(newEndDateTime, {onlySelf: true, emitEvent: false}); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue