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