Maksym Tsymbarov
3 months ago
1092 changed files with 50034 additions and 26956 deletions
File diff suppressed because one or more lines are too long
@ -0,0 +1,85 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc; |
||||
|
|
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEventType; |
||||
|
import org.thingsboard.server.gen.edge.v1.DownlinkMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.EdgeVersion; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.edge.EdgeContextComponent; |
||||
|
import org.thingsboard.server.service.edge.EdgeMsgConstructorUtils; |
||||
|
import org.thingsboard.server.service.edge.rpc.utils.EdgeVersionUtils; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
|
||||
|
@Component |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
@ConditionalOnProperty(prefix = "edges", value = "enabled", havingValue = "true") |
||||
|
@TbCoreComponent |
||||
|
public class DownlinkMessageMapper { |
||||
|
|
||||
|
private final EdgeContextComponent ctx; |
||||
|
|
||||
|
public List<DownlinkMsg> convertToDownlinkMsgsPack(EdgeSessionState state, List<EdgeEvent> edgeEvents) { |
||||
|
List<DownlinkMsg> result = new ArrayList<>(); |
||||
|
List<EdgeEvent> filtered = EdgeMsgConstructorUtils.mergeAndFilterDownlinkDuplicates(edgeEvents); |
||||
|
for (EdgeEvent edgeEvent : filtered) { |
||||
|
log.trace("[{}][{}] converting edge event to downlink msg [{}]", state.getTenantId(), state.getEdgeId(), edgeEvent); |
||||
|
DownlinkMsg downlinkMsg = null; |
||||
|
try { |
||||
|
switch (edgeEvent.getAction()) { |
||||
|
case UPDATED, ADDED, DELETED, ASSIGNED_TO_EDGE, UNASSIGNED_FROM_EDGE, ALARM_ACK, ALARM_CLEAR, |
||||
|
ALARM_DELETE, CREDENTIALS_UPDATED, RELATION_ADD_OR_UPDATE, RELATION_DELETED, RPC_CALL, |
||||
|
ASSIGNED_TO_CUSTOMER, UNASSIGNED_FROM_CUSTOMER, ADDED_COMMENT, UPDATED_COMMENT, DELETED_COMMENT -> { |
||||
|
downlinkMsg = convertEntityEventToDownlink(state, edgeEvent); |
||||
|
if (downlinkMsg != null && downlinkMsg.getWidgetTypeUpdateMsgCount() > 0) { |
||||
|
log.trace("[{}][{}] widgetTypeUpdateMsg message processed, downlinkMsgId = {}", |
||||
|
state.getTenantId(), state.getEdgeId(), downlinkMsg.getDownlinkMsgId()); |
||||
|
} else { |
||||
|
log.trace("[{}][{}] entity message processed [{}]", state.getTenantId(), state.getEdgeId(), downlinkMsg); |
||||
|
} |
||||
|
} |
||||
|
case ATTRIBUTES_UPDATED, POST_ATTRIBUTES, ATTRIBUTES_DELETED, TIMESERIES_UPDATED -> |
||||
|
downlinkMsg = ctx.getTelemetryProcessor().convertTelemetryEventToDownlink(state.getEdge(), edgeEvent); |
||||
|
default -> log.warn("[{}][{}] Unsupported action type [{}]", state.getTenantId(), state.getEdgeId(), edgeEvent.getAction()); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.trace("[{}][{}] Exception during converting edge event to downlink msg", state.getTenantId(), state.getEdgeId(), e); |
||||
|
} |
||||
|
if (downlinkMsg != null) { |
||||
|
result.add(downlinkMsg); |
||||
|
} |
||||
|
} |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
protected DownlinkMsg convertEntityEventToDownlink(EdgeSessionState state, EdgeEvent edgeEvent) { |
||||
|
log.trace("[{}] Executing convertEntityEventToDownlink, edgeEvent [{}], action [{}]", edgeEvent.getTenantId(), edgeEvent, edgeEvent.getAction()); |
||||
|
if ((EdgeEventType.OAUTH2_CLIENT.equals(edgeEvent.getType()) || EdgeEventType.DOMAIN.equals(edgeEvent.getType())) && |
||||
|
(EdgeVersionUtils.isEdgeVersionOlderThan(state.getEdgeVersion(), EdgeVersion.V_3_8_0))) { |
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
return ctx.getProcessor(edgeEvent.getType()).convertEdgeEventToDownlink(edgeEvent, state.getEdgeVersion()); |
||||
|
} |
||||
|
} |
||||
@ -1,712 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2026 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.edge.rpc; |
|
||||
|
|
||||
import com.fasterxml.jackson.databind.node.ObjectNode; |
|
||||
import com.google.common.util.concurrent.FutureCallback; |
|
||||
import com.google.common.util.concurrent.Futures; |
|
||||
import io.grpc.Server; |
|
||||
import io.grpc.netty.shaded.io.grpc.netty.GrpcSslContexts; |
|
||||
import io.grpc.netty.shaded.io.grpc.netty.NettyServerBuilder; |
|
||||
import io.grpc.netty.shaded.io.netty.handler.ssl.SslContext; |
|
||||
import io.grpc.netty.shaded.io.netty.handler.ssl.SslContextBuilder; |
|
||||
import io.grpc.stub.StreamObserver; |
|
||||
import jakarta.annotation.Nullable; |
|
||||
import jakarta.annotation.PreDestroy; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.springframework.beans.factory.annotation.Autowired; |
|
||||
import org.springframework.beans.factory.annotation.Value; |
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
|
||||
import org.springframework.context.annotation.Lazy; |
|
||||
import org.springframework.stereotype.Service; |
|
||||
import org.thingsboard.common.util.JacksonUtil; |
|
||||
import org.thingsboard.common.util.ThingsBoardExecutors; |
|
||||
import org.thingsboard.rule.engine.api.AttributesSaveRequest; |
|
||||
import org.thingsboard.rule.engine.api.TimeseriesSaveRequest; |
|
||||
import org.thingsboard.server.cache.TbTransactionalCache; |
|
||||
import org.thingsboard.server.cluster.TbClusterService; |
|
||||
import org.thingsboard.server.common.data.AttributeScope; |
|
||||
import org.thingsboard.server.common.data.DataConstants; |
|
||||
import org.thingsboard.server.common.data.StringUtils; |
|
||||
import org.thingsboard.server.common.transport.config.ssl.PemSslCredentials; |
|
||||
import org.thingsboard.server.common.data.edge.Edge; |
|
||||
import org.thingsboard.server.common.data.edge.EdgeEvent; |
|
||||
import org.thingsboard.server.common.data.id.EdgeId; |
|
||||
import org.thingsboard.server.common.data.id.TenantId; |
|
||||
import org.thingsboard.server.common.data.kv.BooleanDataEntry; |
|
||||
import org.thingsboard.server.common.data.kv.LongDataEntry; |
|
||||
import org.thingsboard.server.common.data.msg.TbMsgType; |
|
||||
import org.thingsboard.server.common.data.notification.rule.trigger.EdgeConnectionTrigger; |
|
||||
import org.thingsboard.server.common.msg.TbMsg; |
|
||||
import org.thingsboard.server.common.msg.TbMsgDataType; |
|
||||
import org.thingsboard.server.common.msg.TbMsgMetaData; |
|
||||
import org.thingsboard.server.common.msg.edge.EdgeEventUpdateMsg; |
|
||||
import org.thingsboard.server.common.msg.edge.EdgeHighPriorityMsg; |
|
||||
import org.thingsboard.server.common.msg.edge.EdgeSessionMsg; |
|
||||
import org.thingsboard.server.common.msg.edge.FromEdgeSyncResponse; |
|
||||
import org.thingsboard.server.common.msg.edge.ToEdgeSyncRequest; |
|
||||
import org.thingsboard.server.gen.edge.v1.EdgeRpcServiceGrpc; |
|
||||
import org.thingsboard.server.gen.edge.v1.RequestMsg; |
|
||||
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
|
||||
import org.thingsboard.server.queue.discovery.TbServiceInfoProvider; |
|
||||
import org.thingsboard.server.queue.discovery.TopicService; |
|
||||
import org.thingsboard.server.queue.kafka.KafkaAdmin; |
|
||||
import org.thingsboard.server.queue.provider.TbCoreQueueFactory; |
|
||||
import org.thingsboard.server.queue.util.AfterStartUp; |
|
||||
import org.thingsboard.server.queue.util.TbCoreComponent; |
|
||||
import org.thingsboard.server.service.edge.EdgeContextComponent; |
|
||||
import org.thingsboard.server.service.telemetry.TelemetrySubscriptionService; |
|
||||
|
|
||||
import java.io.IOException; |
|
||||
import java.util.ArrayList; |
|
||||
import java.util.Collection; |
|
||||
import java.util.HashMap; |
|
||||
import java.util.List; |
|
||||
import java.util.Map; |
|
||||
import java.util.Optional; |
|
||||
import java.util.UUID; |
|
||||
import java.util.concurrent.ConcurrentHashMap; |
|
||||
import java.util.concurrent.ConcurrentMap; |
|
||||
import java.util.concurrent.ScheduledExecutorService; |
|
||||
import java.util.concurrent.ScheduledFuture; |
|
||||
import java.util.concurrent.TimeUnit; |
|
||||
import java.util.concurrent.locks.Lock; |
|
||||
import java.util.concurrent.locks.ReentrantLock; |
|
||||
import java.util.function.Consumer; |
|
||||
import java.util.function.Function; |
|
||||
|
|
||||
import static org.thingsboard.server.service.state.DefaultDeviceStateService.ACTIVITY_STATE; |
|
||||
import static org.thingsboard.server.service.state.DefaultDeviceStateService.LAST_CONNECT_TIME; |
|
||||
import static org.thingsboard.server.service.state.DefaultDeviceStateService.LAST_DISCONNECT_TIME; |
|
||||
|
|
||||
@Service |
|
||||
@Slf4j |
|
||||
@ConditionalOnProperty(prefix = "edges", value = "enabled", havingValue = "true") |
|
||||
@TbCoreComponent |
|
||||
public class EdgeGrpcService extends EdgeRpcServiceGrpc.EdgeRpcServiceImplBase implements EdgeRpcService { |
|
||||
|
|
||||
private final ConcurrentMap<EdgeId, EdgeGrpcSession> sessions = new ConcurrentHashMap<>(); |
|
||||
private final ConcurrentMap<UUID, EdgeGrpcSession> sessionsById = new ConcurrentHashMap<>(); |
|
||||
private final ConcurrentMap<EdgeId, Lock> sessionNewEventsLocks = new ConcurrentHashMap<>(); |
|
||||
private final Map<EdgeId, Boolean> sessionNewEvents = new HashMap<>(); |
|
||||
private final ConcurrentMap<EdgeId, ScheduledFuture<?>> sessionEdgeEventChecks = new ConcurrentHashMap<>(); |
|
||||
private final ConcurrentMap<UUID, Consumer<FromEdgeSyncResponse>> localSyncEdgeRequests = new ConcurrentHashMap<>(); |
|
||||
private final ConcurrentMap<EdgeId, Boolean> edgeEventsMigrationProcessed = new ConcurrentHashMap<>(); |
|
||||
private final List<EdgeGrpcSession> zombieSessions = new ArrayList<>(); |
|
||||
|
|
||||
@Value("${edges.rpc.port}") |
|
||||
private int rpcPort; |
|
||||
@Value("${edges.rpc.ssl.enabled}") |
|
||||
private boolean sslEnabled; |
|
||||
@Value("${edges.rpc.ssl.cert}") |
|
||||
private String certFileResource; |
|
||||
@Value("${edges.rpc.ssl.private_key}") |
|
||||
private String privateKeyResource; |
|
||||
@Value("${edges.rpc.ssl.key_password:}") |
|
||||
private String keyPassword; |
|
||||
@Value("${edges.state.persistToTelemetry:false}") |
|
||||
private boolean persistToTelemetry; |
|
||||
@Value("${edges.rpc.client_max_keep_alive_time_sec:1}") |
|
||||
private int clientMaxKeepAliveTimeSec; |
|
||||
@Value("${edges.rpc.max_inbound_message_size:4194304}") |
|
||||
private int maxInboundMessageSize; |
|
||||
@Value("${edges.rpc.keep_alive_time_sec:10}") |
|
||||
private int keepAliveTimeSec; |
|
||||
@Value("${edges.rpc.keep_alive_timeout_sec:5}") |
|
||||
private int keepAliveTimeoutSec; |
|
||||
@Value("${edges.scheduler_pool_size}") |
|
||||
private int schedulerPoolSize; |
|
||||
|
|
||||
@Value("${edges.send_scheduler_pool_size}") |
|
||||
private int sendSchedulerPoolSize; |
|
||||
|
|
||||
@Value("${edges.max_high_priority_queue_size_per_session:10000}") |
|
||||
private int maxHighPriorityQueueSizePerSession; |
|
||||
|
|
||||
@Autowired |
|
||||
@Lazy |
|
||||
private EdgeContextComponent ctx; |
|
||||
|
|
||||
@Autowired |
|
||||
private TbClusterService clusterService; |
|
||||
|
|
||||
@Autowired |
|
||||
private TbServiceInfoProvider serviceInfoProvider; |
|
||||
|
|
||||
@Autowired |
|
||||
private TbTransactionalCache<EdgeId, String> edgeIdServiceIdCache; |
|
||||
|
|
||||
@Autowired |
|
||||
private TopicService topicService; |
|
||||
|
|
||||
@Autowired |
|
||||
private TbCoreQueueFactory tbCoreQueueFactory; |
|
||||
|
|
||||
@Autowired |
|
||||
private Optional<KafkaAdmin> kafkaAdmin; |
|
||||
|
|
||||
private Server server; |
|
||||
|
|
||||
private ScheduledExecutorService edgeEventProcessingExecutorService; |
|
||||
|
|
||||
private ScheduledExecutorService sendDownlinkExecutorService; |
|
||||
|
|
||||
private ScheduledExecutorService executorService; |
|
||||
|
|
||||
@AfterStartUp(order = AfterStartUp.REGULAR_SERVICE) |
|
||||
public void onStartUp() { |
|
||||
log.info("Initializing Edge RPC service!"); |
|
||||
NettyServerBuilder builder = NettyServerBuilder.forPort(rpcPort) |
|
||||
.permitKeepAliveTime(clientMaxKeepAliveTimeSec, TimeUnit.SECONDS) |
|
||||
.keepAliveTime(keepAliveTimeSec, TimeUnit.SECONDS) |
|
||||
.keepAliveTimeout(keepAliveTimeoutSec, TimeUnit.SECONDS) |
|
||||
.permitKeepAliveWithoutCalls(true) |
|
||||
.maxInboundMessageSize(maxInboundMessageSize) |
|
||||
.addService(this); |
|
||||
if (sslEnabled) { |
|
||||
try { |
|
||||
setupSsl(builder); |
|
||||
} catch (Exception e) { |
|
||||
log.error("Unable to set up SSL context. Reason: " + e.getMessage(), e); |
|
||||
throw new RuntimeException("Unable to set up SSL context!", e); |
|
||||
} |
|
||||
} |
|
||||
server = builder.build(); |
|
||||
log.info("Going to start Edge RPC server using port: {}", rpcPort); |
|
||||
try { |
|
||||
server.start(); |
|
||||
} catch (IOException e) { |
|
||||
log.error("Failed to start Edge RPC server!", e); |
|
||||
throw new RuntimeException("Failed to start Edge RPC server!"); |
|
||||
} |
|
||||
this.edgeEventProcessingExecutorService = ThingsBoardExecutors.newScheduledThreadPool(schedulerPoolSize, "edge-event-check-scheduler"); |
|
||||
this.sendDownlinkExecutorService = ThingsBoardExecutors.newScheduledThreadPool(sendSchedulerPoolSize, "edge-send-scheduler"); |
|
||||
this.executorService = ThingsBoardExecutors.newSingleThreadScheduledExecutor("edge-service"); |
|
||||
this.executorService.scheduleAtFixedRate(this::cleanupZombieSessions, 60, 60, TimeUnit.SECONDS); |
|
||||
log.info("Edge RPC service initialized!"); |
|
||||
} |
|
||||
|
|
||||
/** |
|
||||
* Configures TLS for the Edge gRPC server. |
|
||||
* <p> |
|
||||
* Delegates PEM parsing and key management to {@link PemSslCredentials} — the same |
|
||||
* class used by MQTT, CoAP, and LwM2M transports — which supports: |
|
||||
* <ul> |
|
||||
* <li>Separate certificate and private key files (classic two-file setup)</li> |
|
||||
* <li>Combined PEM: certificate chain + private key in a single {@code cert} file |
|
||||
* ({@code private_key} left empty)</li> |
|
||||
* <li>Encrypted private keys (password supplied via {@code key_password})</li> |
|
||||
* </ul> |
|
||||
* Path resolution (for both {@code cert} and {@code private_key}) is handled by |
|
||||
* {@link org.thingsboard.server.common.data.ResourceUtils#getInputStream ResourceUtils}: |
|
||||
* absolute path → relative / working-dir → classpath → {@code classpath:} prefix. |
|
||||
*/ |
|
||||
void setupSsl(NettyServerBuilder builder) throws Exception { |
|
||||
PemSslCredentials credentials = new PemSslCredentials(); |
|
||||
credentials.setCertFile(certFileResource); |
|
||||
credentials.setKeyFile(StringUtils.isEmpty(privateKeyResource) ? null : privateKeyResource); |
|
||||
credentials.setKeyPassword(keyPassword); |
|
||||
credentials.init(false); |
|
||||
|
|
||||
SslContext sslContext = GrpcSslContexts.configure( |
|
||||
SslContextBuilder.forServer(credentials.createKeyManagerFactory())).build(); |
|
||||
builder.sslContext(sslContext); |
|
||||
} |
|
||||
|
|
||||
@PreDestroy |
|
||||
public void destroy() { |
|
||||
if (server != null) { |
|
||||
server.shutdownNow(); |
|
||||
} |
|
||||
for (Map.Entry<EdgeId, ScheduledFuture<?>> entry : sessionEdgeEventChecks.entrySet()) { |
|
||||
EdgeId edgeId = entry.getKey(); |
|
||||
ScheduledFuture<?> sessionEdgeEventCheck = entry.getValue(); |
|
||||
if (sessionEdgeEventCheck != null && !sessionEdgeEventCheck.isCancelled() && !sessionEdgeEventCheck.isDone()) { |
|
||||
sessionEdgeEventCheck.cancel(true); |
|
||||
sessionEdgeEventChecks.remove(edgeId); |
|
||||
} |
|
||||
} |
|
||||
if (edgeEventProcessingExecutorService != null) { |
|
||||
edgeEventProcessingExecutorService.shutdownNow(); |
|
||||
} |
|
||||
if (sendDownlinkExecutorService != null) { |
|
||||
sendDownlinkExecutorService.shutdownNow(); |
|
||||
} |
|
||||
if (executorService != null) { |
|
||||
executorService.shutdownNow(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public StreamObserver<RequestMsg> handleMsgs(StreamObserver<ResponseMsg> outputStream) { |
|
||||
EdgeGrpcSession session = createEdgeGrpcSession(outputStream); |
|
||||
return session.getInputStream(); |
|
||||
} |
|
||||
|
|
||||
private EdgeGrpcSession createEdgeGrpcSession(StreamObserver<ResponseMsg> outputStream) { |
|
||||
return kafkaAdmin.isPresent() |
|
||||
? new KafkaEdgeGrpcSession(ctx, topicService, tbCoreQueueFactory, kafkaAdmin.get(), outputStream, this::onEdgeConnect, this::onEdgeDisconnect, |
|
||||
sendDownlinkExecutorService, maxInboundMessageSize, maxHighPriorityQueueSizePerSession) |
|
||||
: new PostgresEdgeGrpcSession(ctx, outputStream, this::onEdgeConnect, this::onEdgeDisconnect, |
|
||||
sendDownlinkExecutorService, maxInboundMessageSize, maxHighPriorityQueueSizePerSession); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void onToEdgeSessionMsg(TenantId tenantId, EdgeSessionMsg msg) { |
|
||||
switch (msg.getMsgType()) { |
|
||||
case EDGE_HIGH_PRIORITY_TO_EDGE_SESSION_MSG -> { |
|
||||
EdgeHighPriorityMsg edgeHighPriorityMsg = (EdgeHighPriorityMsg) msg; |
|
||||
log.trace("[{}] edgeEventMsg [{}]", tenantId, msg); |
|
||||
onEdgeHighPriorityEvent(edgeHighPriorityMsg); |
|
||||
} |
|
||||
case EDGE_EVENT_UPDATE_TO_EDGE_SESSION_MSG -> { |
|
||||
EdgeEventUpdateMsg edgeEventUpdateMsg = (EdgeEventUpdateMsg) msg; |
|
||||
log.trace("[{}] onToEdgeEventUpdateMsg [{}]", tenantId, msg); |
|
||||
onEdgeEventUpdate(tenantId, edgeEventUpdateMsg.getEdgeId()); |
|
||||
} |
|
||||
case EDGE_SYNC_REQUEST_TO_EDGE_SESSION_MSG -> { |
|
||||
ToEdgeSyncRequest toEdgeSyncRequest = (ToEdgeSyncRequest) msg; |
|
||||
log.trace("[{}] toEdgeSyncRequest [{}]", tenantId, msg); |
|
||||
startSyncProcess(tenantId, toEdgeSyncRequest.getEdgeId(), toEdgeSyncRequest.getId(), toEdgeSyncRequest.getServiceId()); |
|
||||
} |
|
||||
case EDGE_SYNC_RESPONSE_FROM_EDGE_SESSION_MSG -> { |
|
||||
FromEdgeSyncResponse fromEdgeSyncResponse = (FromEdgeSyncResponse) msg; |
|
||||
log.trace("[{}] fromEdgeSyncResponse [{}]", tenantId, msg); |
|
||||
processSyncResponse(fromEdgeSyncResponse); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void updateEdge(TenantId tenantId, Edge edge) { |
|
||||
if (edge == null) { |
|
||||
log.warn("[{}] Edge is null - edge is removed and outdated notification is in process!", tenantId); |
|
||||
return; |
|
||||
} |
|
||||
EdgeGrpcSession session = sessions.get(edge.getId()); |
|
||||
if (session != null && session.isConnected()) { |
|
||||
log.debug("[{}] Updating configuration for edge [{}] [{}]", tenantId, edge.getName(), edge.getId()); |
|
||||
session.onConfigurationUpdate(edge); |
|
||||
} else { |
|
||||
log.debug("[{}] Session doesn't exist for edge [{}] [{}]", tenantId, edge.getName(), edge.getId()); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void deleteEdge(TenantId tenantId, EdgeId edgeId) { |
|
||||
EdgeGrpcSession session = sessions.get(edgeId); |
|
||||
if (session != null && session.isConnected()) { |
|
||||
log.info("[{}] Closing and removing session for edge [{}]", tenantId, edgeId); |
|
||||
destroySession(session); |
|
||||
session.cleanUp(); |
|
||||
sessions.remove(edgeId); |
|
||||
sessionsById.remove(session.getSessionId()); |
|
||||
final Lock newEventLock = sessionNewEventsLocks.computeIfAbsent(edgeId, id -> new ReentrantLock()); |
|
||||
newEventLock.lock(); |
|
||||
try { |
|
||||
sessionNewEvents.remove(edgeId); |
|
||||
} finally { |
|
||||
newEventLock.unlock(); |
|
||||
} |
|
||||
cancelScheduleEdgeEventsCheck(edgeId); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void onEdgeEventUpdate(TenantId tenantId, EdgeId edgeId) { |
|
||||
EdgeGrpcSession session = sessions.get(edgeId); |
|
||||
if (session != null && session.isConnected()) { |
|
||||
log.trace("[{}] onEdgeEventUpdate [{}]", tenantId, edgeId.getId()); |
|
||||
updateSessionEventsFlag(tenantId, edgeId); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void onEdgeHighPriorityEvent(EdgeHighPriorityMsg msg) { |
|
||||
TenantId tenantId = msg.getTenantId(); |
|
||||
EdgeEvent edgeEvent = msg.getEdgeEvent(); |
|
||||
EdgeId edgeId = edgeEvent.getEdgeId(); |
|
||||
EdgeGrpcSession session = sessions.get(edgeId); |
|
||||
if (session != null && session.isConnected()) { |
|
||||
log.trace("[{}] onEdgeEvent [{}]", tenantId, edgeId); |
|
||||
session.addEventToHighPriorityQueue(edgeEvent); |
|
||||
updateSessionEventsFlag(tenantId, edgeId); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void updateSessionEventsFlag(TenantId tenantId, EdgeId edgeId) { |
|
||||
final Lock newEventLock = sessionNewEventsLocks.computeIfAbsent(edgeId, id -> new ReentrantLock()); |
|
||||
newEventLock.lock(); |
|
||||
try { |
|
||||
if (Boolean.FALSE.equals(sessionNewEvents.get(edgeId))) { |
|
||||
log.trace("[{}] set session new events flag to true [{}]", tenantId, edgeId.getId()); |
|
||||
sessionNewEvents.put(edgeId, true); |
|
||||
} |
|
||||
} finally { |
|
||||
newEventLock.unlock(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void onEdgeConnect(EdgeId edgeId, EdgeGrpcSession edgeGrpcSession) { |
|
||||
Edge edge = edgeGrpcSession.getEdge(); |
|
||||
TenantId tenantId = edge.getTenantId(); |
|
||||
log.info("[{}][{}] edge [{}] connected successfully.", tenantId, edgeGrpcSession.getSessionId(), edgeId); |
|
||||
if (sessions.containsKey(edgeId)) { |
|
||||
EdgeGrpcSession existing = sessions.get(edgeId); |
|
||||
if (existing != null) { |
|
||||
log.info("[{}][{}] Replacing existing session [{}] for edge [{}]", tenantId, edgeGrpcSession.getSessionId(), existing.getSessionId(), edgeId); |
|
||||
destroySession(existing); |
|
||||
sessionsById.remove(existing.getSessionId()); |
|
||||
} |
|
||||
} |
|
||||
sessions.put(edgeId, edgeGrpcSession); |
|
||||
sessionsById.put(edgeGrpcSession.getSessionId(), edgeGrpcSession); |
|
||||
final Lock newEventLock = sessionNewEventsLocks.computeIfAbsent(edgeId, id -> new ReentrantLock()); |
|
||||
newEventLock.lock(); |
|
||||
try { |
|
||||
sessionNewEvents.put(edgeId, true); |
|
||||
} finally { |
|
||||
newEventLock.unlock(); |
|
||||
} |
|
||||
save(tenantId, edgeId, ACTIVITY_STATE, true); |
|
||||
long lastConnectTs = System.currentTimeMillis(); |
|
||||
save(tenantId, edgeId, LAST_CONNECT_TIME, lastConnectTs); |
|
||||
edgeIdServiceIdCache.put(edgeId, serviceInfoProvider.getServiceId()); |
|
||||
pushRuleEngineMessage(tenantId, edge, lastConnectTs, TbMsgType.CONNECT_EVENT); |
|
||||
cancelScheduleEdgeEventsCheck(edgeId); |
|
||||
edgeEventsMigrationProcessed.putIfAbsent(edgeId, Boolean.FALSE); |
|
||||
scheduleEdgeEventsCheck(edgeGrpcSession); |
|
||||
} |
|
||||
|
|
||||
private void startSyncProcess(TenantId tenantId, EdgeId edgeId, UUID requestId, String requestServiceId) { |
|
||||
EdgeGrpcSession session = sessions.get(edgeId); |
|
||||
if (session != null) { |
|
||||
if (session.isSyncInProgress()) { |
|
||||
clusterService.pushEdgeSyncResponseToCore(new FromEdgeSyncResponse(requestId, tenantId, edgeId, false, "Sync process is active at the moment"), requestServiceId); |
|
||||
} else { |
|
||||
boolean success = false; |
|
||||
if (session.isConnected()) { |
|
||||
session.startSyncProcess(true); |
|
||||
success = true; |
|
||||
} |
|
||||
clusterService.pushEdgeSyncResponseToCore(new FromEdgeSyncResponse(requestId, tenantId, edgeId, success, ""), requestServiceId); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void processSyncRequest(TenantId tenantId, EdgeId edgeId, Consumer<FromEdgeSyncResponse> responseConsumer) { |
|
||||
ToEdgeSyncRequest request = new ToEdgeSyncRequest(UUID.randomUUID(), tenantId, edgeId, serviceInfoProvider.getServiceId()); |
|
||||
|
|
||||
UUID requestId = request.getId(); |
|
||||
EdgeGrpcSession session = sessions.get(request.getEdgeId()); |
|
||||
if (session != null && session.isSyncInProgress()) { |
|
||||
responseConsumer.accept(new FromEdgeSyncResponse(requestId, request.getTenantId(), request.getEdgeId(), false, "Sync process is active at the moment")); |
|
||||
} else { |
|
||||
log.trace("[{}][{}] Processing sync edge request [{}], serviceId [{}]", request.getTenantId(), request.getId(), request.getEdgeId(), request.getServiceId()); |
|
||||
localSyncEdgeRequests.put(requestId, responseConsumer); |
|
||||
clusterService.pushEdgeSyncRequestToEdge(request); |
|
||||
scheduleSyncRequestTimeout(request, requestId); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void scheduleSyncRequestTimeout(ToEdgeSyncRequest request, UUID requestId) { |
|
||||
log.trace("[{}] scheduling sync edge request", requestId); |
|
||||
executorService.schedule(() -> { |
|
||||
log.trace("[{}] checking if sync edge request is not processed...", requestId); |
|
||||
Consumer<FromEdgeSyncResponse> consumer = localSyncEdgeRequests.remove(requestId); |
|
||||
if (consumer != null) { |
|
||||
log.trace("[{}] timeout for processing sync edge request.", requestId); |
|
||||
consumer.accept(new FromEdgeSyncResponse(requestId, request.getTenantId(), request.getEdgeId(), false, "Edge is not connected")); |
|
||||
} |
|
||||
}, 20, TimeUnit.SECONDS); |
|
||||
} |
|
||||
|
|
||||
private void processSyncResponse(FromEdgeSyncResponse response) { |
|
||||
log.trace("[{}] Received response from sync service: [{}]", response.getId(), response); |
|
||||
UUID requestId = response.getId(); |
|
||||
Consumer<FromEdgeSyncResponse> consumer = localSyncEdgeRequests.remove(requestId); |
|
||||
if (consumer != null) { |
|
||||
consumer.accept(response); |
|
||||
} else { |
|
||||
log.trace("[{}] Unknown or stale sync response received [{}]", requestId, response); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void scheduleEdgeEventsCheck(EdgeGrpcSession session) { |
|
||||
EdgeId edgeId = session.getEdge().getId(); |
|
||||
TenantId tenantId = session.getEdge().getTenantId(); |
|
||||
|
|
||||
cancelScheduleEdgeEventsCheck(edgeId); |
|
||||
|
|
||||
if (sessions.containsKey(edgeId)) { |
|
||||
ScheduledFuture<?> edgeEventCheckTask = edgeEventProcessingExecutorService.schedule(() -> { |
|
||||
try { |
|
||||
final Lock newEventLock = sessionNewEventsLocks.computeIfAbsent(edgeId, id -> new ReentrantLock()); |
|
||||
newEventLock.lock(); |
|
||||
try { |
|
||||
if (Boolean.TRUE.equals(sessionNewEvents.get(edgeId))) { |
|
||||
log.trace("[{}][{}] set session new events flag to false", tenantId, edgeId.getId()); |
|
||||
sessionNewEvents.put(edgeId, false); |
|
||||
session.processHighPriorityEvents(); |
|
||||
processEdgeEventMigrationIfNeeded(session, edgeId); |
|
||||
if (Boolean.TRUE.equals(edgeEventsMigrationProcessed.get(edgeId))) { |
|
||||
Futures.addCallback(session.processEdgeEvents(), new FutureCallback<>() { |
|
||||
@Override |
|
||||
public void onSuccess(Boolean newEventsAdded) { |
|
||||
if (Boolean.TRUE.equals(newEventsAdded)) { |
|
||||
log.trace("[{}][{}] new events added. set session new events flag to true", tenantId, edgeId.getId()); |
|
||||
sessionNewEvents.put(edgeId, true); |
|
||||
} |
|
||||
scheduleEdgeEventsCheck(session); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void onFailure(Throwable t) { |
|
||||
log.warn("[{}] Failed to process edge events for edge [{}]!", tenantId, session.getEdge().getId().getId(), t); |
|
||||
scheduleEdgeEventsCheck(session); |
|
||||
} |
|
||||
}, ctx.getGrpcCallbackExecutorService()); |
|
||||
} else { |
|
||||
scheduleEdgeEventsCheck(session); |
|
||||
} |
|
||||
} else { |
|
||||
scheduleEdgeEventsCheck(session); |
|
||||
} |
|
||||
} finally { |
|
||||
newEventLock.unlock(); |
|
||||
} |
|
||||
} catch (Exception e) { |
|
||||
log.warn("[{}] Failed to process edge events for edge [{}]!", tenantId, session.getEdge().getId().getId(), e); |
|
||||
} |
|
||||
}, ctx.getEdgeEventStorageSettings().getNoRecordsSleepInterval(), TimeUnit.MILLISECONDS); |
|
||||
sessionEdgeEventChecks.put(edgeId, edgeEventCheckTask); |
|
||||
log.trace("[{}] Check edge event scheduled for edge [{}]", tenantId, edgeId.getId()); |
|
||||
} else { |
|
||||
log.debug("[{}] Session was removed and edge event check schedule must not be started [{}]", |
|
||||
tenantId, edgeId.getId()); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void processEdgeEventMigrationIfNeeded(EdgeGrpcSession session, EdgeId edgeId) throws Exception { |
|
||||
boolean isMigrationProcessed = edgeEventsMigrationProcessed.getOrDefault(edgeId, Boolean.FALSE); |
|
||||
if (!isMigrationProcessed) { |
|
||||
log.info("Starting edge event migration for edge [{}]", edgeId.getId()); |
|
||||
Boolean eventsExist = session.migrateEdgeEvents().get(); |
|
||||
if (Boolean.TRUE.equals(eventsExist)) { |
|
||||
log.info("Migration still in progress for edge [{}]", edgeId.getId()); |
|
||||
sessionNewEvents.put(edgeId, true); |
|
||||
scheduleEdgeEventsCheck(session); |
|
||||
} else if (Boolean.FALSE.equals(eventsExist)) { |
|
||||
log.info("Migration completed for edge [{}]", edgeId.getId()); |
|
||||
edgeEventsMigrationProcessed.put(edgeId, true); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void cancelScheduleEdgeEventsCheck(EdgeId edgeId) { |
|
||||
log.trace("[{}] cancelling edge event check for edge", edgeId); |
|
||||
if (sessionEdgeEventChecks.containsKey(edgeId)) { |
|
||||
ScheduledFuture<?> sessionEdgeEventCheck = sessionEdgeEventChecks.get(edgeId); |
|
||||
if (sessionEdgeEventCheck != null && !sessionEdgeEventCheck.isCancelled() && !sessionEdgeEventCheck.isDone()) { |
|
||||
sessionEdgeEventCheck.cancel(true); |
|
||||
sessionEdgeEventChecks.remove(edgeId); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void onEdgeDisconnect(Edge edge, UUID sessionId) { |
|
||||
EdgeId edgeId = edge.getId(); |
|
||||
log.info("[{}][{}] edge disconnected!", edgeId, sessionId); |
|
||||
EdgeGrpcSession current = sessions.get(edgeId); |
|
||||
if (current != null && current.getSessionId().equals(sessionId)) { |
|
||||
EdgeGrpcSession toRemove = sessions.remove(edgeId); |
|
||||
final Lock newEventLock = sessionNewEventsLocks.computeIfAbsent(edgeId, id -> new ReentrantLock()); |
|
||||
newEventLock.lock(); |
|
||||
try { |
|
||||
sessionNewEvents.remove(edgeId); |
|
||||
} finally { |
|
||||
newEventLock.unlock(); |
|
||||
} |
|
||||
destroySession(toRemove); |
|
||||
sessionsById.remove(sessionId); |
|
||||
TenantId tenantId = toRemove.getEdge().getTenantId(); |
|
||||
save(tenantId, edgeId, ACTIVITY_STATE, false); |
|
||||
long lastDisconnectTs = System.currentTimeMillis(); |
|
||||
save(tenantId, edgeId, LAST_DISCONNECT_TIME, lastDisconnectTs); |
|
||||
pushRuleEngineMessage(toRemove.getEdge().getTenantId(), edge, lastDisconnectTs, TbMsgType.DISCONNECT_EVENT); |
|
||||
cancelScheduleEdgeEventsCheck(edgeId); |
|
||||
} else { |
|
||||
log.info("[{}] edge session [{}] is not current anymore. Attempting to destroy it by sessionId.", edgeId, sessionId); |
|
||||
EdgeGrpcSession stale = sessionsById.remove(sessionId); |
|
||||
if (stale != null) { |
|
||||
try { |
|
||||
destroySession(stale); |
|
||||
log.info("[{}][{}] Successfully destroyed stale session for edge [{}]", stale.getTenantId(), sessionId, edgeId); |
|
||||
} catch (Exception e) { |
|
||||
log.warn("[{}][{}] Failed to destroy stale session for edge [{}]", stale.getTenantId(), sessionId, edgeId, e); |
|
||||
} |
|
||||
} else { |
|
||||
log.debug("[{}] No session found by sessionId [{}] to destroy", edgeId, sessionId); |
|
||||
} |
|
||||
} |
|
||||
edgeIdServiceIdCache.evict(edgeId); |
|
||||
} |
|
||||
|
|
||||
private void destroySession(EdgeGrpcSession session) { |
|
||||
try (session) { |
|
||||
if (!session.destroy()) { |
|
||||
log.warn("[{}][{}] Session destroy failed for edge [{}] with session id [{}]. Adding to zombie queue for later cleanup.", |
|
||||
session.getTenantId(), session.getEdge().getId(), session.getEdge().getName(), session.getSessionId()); |
|
||||
zombieSessions.add(session); |
|
||||
} |
|
||||
} catch (Exception e) { |
|
||||
log.warn("[{}][{}] Exception during session destroy for edge [{}] with session id [{}]", |
|
||||
session.getTenantId(), session.getEdge().getId(), session.getEdge().getName(), session.getSessionId(), e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void save(TenantId tenantId, EdgeId edgeId, String key, long value) { |
|
||||
log.debug("[{}][{}] Updating long edge telemetry [{}] [{}]", tenantId, edgeId, key, value); |
|
||||
if (persistToTelemetry) { |
|
||||
ctx.getTsSubService().saveTimeseries(TimeseriesSaveRequest.builder() |
|
||||
.tenantId(tenantId) |
|
||||
.entityId(edgeId) |
|
||||
.entry(new LongDataEntry(key, value)) |
|
||||
.callback(new AttributeSaveCallback(tenantId, edgeId, key, value)) |
|
||||
.build()); |
|
||||
} else { |
|
||||
ctx.getTsSubService().saveAttributes(AttributesSaveRequest.builder() |
|
||||
.tenantId(tenantId) |
|
||||
.entityId(edgeId) |
|
||||
.scope(AttributeScope.SERVER_SCOPE) |
|
||||
.entry(new LongDataEntry(key, value)) |
|
||||
.callback(new AttributeSaveCallback(tenantId, edgeId, key, value)) |
|
||||
.build()); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void save(TenantId tenantId, EdgeId edgeId, String key, boolean value) { |
|
||||
log.debug("[{}][{}] Updating boolean edge telemetry [{}] [{}]", tenantId, edgeId, key, value); |
|
||||
if (persistToTelemetry) { |
|
||||
ctx.getTsSubService().saveTimeseries(TimeseriesSaveRequest.builder() |
|
||||
.tenantId(tenantId) |
|
||||
.entityId(edgeId) |
|
||||
.entry(new BooleanDataEntry(key, value)) |
|
||||
.callback(new AttributeSaveCallback(tenantId, edgeId, key, value)) |
|
||||
.build()); |
|
||||
} else { |
|
||||
ctx.getTsSubService().saveAttributes(AttributesSaveRequest.builder() |
|
||||
.tenantId(tenantId) |
|
||||
.entityId(edgeId) |
|
||||
.scope(AttributeScope.SERVER_SCOPE) |
|
||||
.entry(new BooleanDataEntry(key, value)) |
|
||||
.callback(new AttributeSaveCallback(tenantId, edgeId, key, value)) |
|
||||
.build()); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void pushRuleEngineMessage(TenantId tenantId, Edge edge, long ts, TbMsgType msgType) { |
|
||||
try { |
|
||||
EdgeId edgeId = edge.getId(); |
|
||||
ObjectNode edgeState = JacksonUtil.newObjectNode(); |
|
||||
boolean isConnected = TbMsgType.CONNECT_EVENT.equals(msgType); |
|
||||
if (isConnected) { |
|
||||
edgeState.put(ACTIVITY_STATE, true); |
|
||||
edgeState.put(LAST_CONNECT_TIME, ts); |
|
||||
} else { |
|
||||
edgeState.put(ACTIVITY_STATE, false); |
|
||||
edgeState.put(LAST_DISCONNECT_TIME, ts); |
|
||||
} |
|
||||
ctx.getRuleProcessor().process(EdgeConnectionTrigger.builder() |
|
||||
.tenantId(tenantId) |
|
||||
.customerId(edge.getCustomerId()) |
|
||||
.edgeId(edgeId) |
|
||||
.edgeName(edge.getName()) |
|
||||
.connected(isConnected).build()); |
|
||||
String data = JacksonUtil.toString(edgeState); |
|
||||
TbMsgMetaData md = new TbMsgMetaData(); |
|
||||
if (!persistToTelemetry) { |
|
||||
md.putValue(DataConstants.SCOPE, DataConstants.SERVER_SCOPE); |
|
||||
md.putValue("edgeName", edge.getName()); |
|
||||
md.putValue("edgeType", edge.getType()); |
|
||||
} |
|
||||
TbMsg tbMsg = TbMsg.newMsg() |
|
||||
.type(msgType) |
|
||||
.originator(edgeId) |
|
||||
.copyMetaData(md) |
|
||||
.dataType(TbMsgDataType.JSON) |
|
||||
.data(data) |
|
||||
.build(); |
|
||||
clusterService.pushMsgToRuleEngine(tenantId, edgeId, tbMsg, null); |
|
||||
} catch (Exception e) { |
|
||||
log.warn("[{}][{}] Failed to push {}", tenantId, edge.getId(), msgType, e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void cleanupZombieSessions() { |
|
||||
try { |
|
||||
tryToDestroyZombieSessions(getZombieSessions(sessions.values()), s -> sessions.remove(s.getEdge().getId())); |
|
||||
tryToDestroyZombieSessions(getZombieSessions(sessionsById.values()), s -> sessionsById.remove(s.getSessionId())); |
|
||||
|
|
||||
zombieSessions.removeIf(zombie -> { |
|
||||
if (zombie.destroy()) { |
|
||||
log.info("[{}][{}] Successfully cleaned up zombie session [{}] for edge [{}].", |
|
||||
zombie.getTenantId(), zombie.getEdge().getId(), zombie.getSessionId(), zombie.getEdge().getName()); |
|
||||
return true; |
|
||||
} else { |
|
||||
log.warn("[{}][{}] Failed to remove zombie session [{}] for edge [{}].", |
|
||||
zombie.getTenantId(), zombie.getEdge().getId(), zombie.getSessionId(), zombie.getEdge().getName()); |
|
||||
return false; |
|
||||
} |
|
||||
}); |
|
||||
} catch (Exception e) { |
|
||||
log.warn("Failed to cleanup kafka sessions", e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private List<EdgeGrpcSession> getZombieSessions(Collection<EdgeGrpcSession> sessions) { |
|
||||
List<EdgeGrpcSession> result = new ArrayList<>(); |
|
||||
for (EdgeGrpcSession session : sessions) { |
|
||||
if (isKafkaSessionAndZombie(session)) { |
|
||||
result.add(session); |
|
||||
} |
|
||||
} |
|
||||
return result; |
|
||||
} |
|
||||
|
|
||||
private void tryToDestroyZombieSessions(List<EdgeGrpcSession> sessionsToRemove, Function<EdgeGrpcSession, EdgeGrpcSession> removeFunc) { |
|
||||
for (EdgeGrpcSession toRemove : sessionsToRemove) { |
|
||||
log.info("[{}] Destroying session for edge because edge is not connected", toRemove.getEdge().getId()); |
|
||||
if (toRemove.destroy()) { |
|
||||
removeFunc.apply(toRemove); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private boolean isKafkaSessionAndZombie(EdgeGrpcSession session) { |
|
||||
if (session instanceof KafkaEdgeGrpcSession kafkaSession) { |
|
||||
log.debug("[{}] kafkaSession.isConnected() = {}, kafkaSession.getConsumer().getConsumer().isStopped() = {}", |
|
||||
kafkaSession.getEdge().getId(), |
|
||||
kafkaSession.isConnected(), |
|
||||
kafkaSession.getConsumer() != null ? kafkaSession.getConsumer().getConsumer() != null ? kafkaSession.getConsumer().getConsumer().isStopped() : null : null); |
|
||||
return !kafkaSession.isConnected() && |
|
||||
kafkaSession.getConsumer() != null && |
|
||||
kafkaSession.getConsumer().getConsumer() != null && |
|
||||
!kafkaSession.getConsumer().getConsumer().isStopped(); |
|
||||
} |
|
||||
return false; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
File diff suppressed because it is too large
@ -0,0 +1,238 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc; |
||||
|
|
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.notification.rule.trigger.EdgeCommunicationFailureTrigger; |
||||
|
import org.thingsboard.server.gen.edge.v1.AiModelUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.AlarmCommentUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.AlarmUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.ApiKeyUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.AssetProfileUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.AssetUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.AttributesRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.CalculatedFieldRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.CalculatedFieldUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DashboardUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DeviceCredentialsRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DeviceCredentialsUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DeviceProfileUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DeviceRpcCallMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DeviceUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.EntityDataProto; |
||||
|
import org.thingsboard.server.gen.edge.v1.EntityViewUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.EntityViewsRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RelationRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RelationUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.ResourceUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RuleChainMetadataRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RuleChainMetadataUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RuleChainUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.UplinkMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.UserCredentialsRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.UserCredentialsUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.UserUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.WidgetBundleTypesRequestMsg; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.edge.EdgeContextComponent; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
@ConditionalOnProperty(prefix = "edges", value = "enabled", havingValue = "true") |
||||
|
@TbCoreComponent |
||||
|
public class EdgeUplinkMessageDispatcher { |
||||
|
|
||||
|
private final EdgeContextComponent ctx; |
||||
|
|
||||
|
public ListenableFuture<List<Void>> processUplinkMsg(EdgeSessionState state, UplinkMsg uplinkMsg) { |
||||
|
Edge edge = state.getEdge(); |
||||
|
List<ListenableFuture<Void>> result = new ArrayList<>(); |
||||
|
try { |
||||
|
if (uplinkMsg.getDeviceProfileUpdateMsgCount() > 0) { |
||||
|
for (DeviceProfileUpdateMsg deviceProfileUpdateMsg : uplinkMsg.getDeviceProfileUpdateMsgList()) { |
||||
|
result.add(ctx.getDeviceProfileProcessor().processDeviceProfileMsgFromEdge(edge.getTenantId(), edge, deviceProfileUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getDeviceUpdateMsgCount() > 0) { |
||||
|
for (DeviceUpdateMsg deviceUpdateMsg : uplinkMsg.getDeviceUpdateMsgList()) { |
||||
|
result.add(ctx.getDeviceProcessor().processDeviceMsgFromEdge(edge.getTenantId(), edge, deviceUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getDeviceCredentialsUpdateMsgCount() > 0) { |
||||
|
for (DeviceCredentialsUpdateMsg deviceCredentialsUpdateMsg : uplinkMsg.getDeviceCredentialsUpdateMsgList()) { |
||||
|
result.add(ctx.getDeviceProcessor().processDeviceCredentialsMsgFromEdge(edge.getTenantId(), edge.getId(), deviceCredentialsUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getAssetProfileUpdateMsgCount() > 0) { |
||||
|
for (AssetProfileUpdateMsg assetProfileUpdateMsg : uplinkMsg.getAssetProfileUpdateMsgList()) { |
||||
|
result.add(ctx.getAssetProfileProcessor().processAssetProfileMsgFromEdge(edge.getTenantId(), edge, assetProfileUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getAssetUpdateMsgCount() > 0) { |
||||
|
for (AssetUpdateMsg assetUpdateMsg : uplinkMsg.getAssetUpdateMsgList()) { |
||||
|
result.add(ctx.getAssetProcessor().processAssetMsgFromEdge(edge.getTenantId(), edge, assetUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getRuleChainUpdateMsgCount() > 0) { |
||||
|
for (RuleChainUpdateMsg ruleChainUpdateMsg : uplinkMsg.getRuleChainUpdateMsgList()) { |
||||
|
result.add(ctx.getRuleChainProcessor().processRuleChainMsgFromEdge(edge.getTenantId(), edge, ruleChainUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getRuleChainMetadataUpdateMsgCount() > 0) { |
||||
|
for (RuleChainMetadataUpdateMsg ruleChainMetadataUpdateMsg : uplinkMsg.getRuleChainMetadataUpdateMsgList()) { |
||||
|
result.add(ctx.getRuleChainProcessor().processRuleChainMetadataMsgFromEdge(edge.getTenantId(), edge, ruleChainMetadataUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getEntityViewUpdateMsgCount() > 0) { |
||||
|
for (EntityViewUpdateMsg entityViewUpdateMsg : uplinkMsg.getEntityViewUpdateMsgList()) { |
||||
|
result.add(ctx.getEntityViewProcessor().processEntityViewMsgFromEdge(edge.getTenantId(), edge, entityViewUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getEntityDataCount() > 0) { |
||||
|
for (EntityDataProto entityData : uplinkMsg.getEntityDataList()) { |
||||
|
result.addAll(ctx.getTelemetryProcessor().processTelemetryMsg(edge.getTenantId(), entityData)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getAlarmUpdateMsgCount() > 0) { |
||||
|
for (AlarmUpdateMsg alarmUpdateMsg : uplinkMsg.getAlarmUpdateMsgList()) { |
||||
|
result.add(ctx.getAlarmProcessor().processAlarmMsgFromEdge(edge.getTenantId(), edge.getId(), alarmUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getAlarmCommentUpdateMsgCount() > 0) { |
||||
|
for (AlarmCommentUpdateMsg alarmCommentUpdateMsg : uplinkMsg.getAlarmCommentUpdateMsgList()) { |
||||
|
result.add(ctx.getAlarmCommentProcessor().processAlarmCommentMsgFromEdge(edge.getTenantId(), edge.getId(), alarmCommentUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getRelationUpdateMsgCount() > 0) { |
||||
|
for (RelationUpdateMsg relationUpdateMsg : uplinkMsg.getRelationUpdateMsgList()) { |
||||
|
result.add(ctx.getRelationProcessor().processRelationMsgFromEdge(edge.getTenantId(), edge, relationUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getDashboardUpdateMsgCount() > 0) { |
||||
|
for (DashboardUpdateMsg dashboardUpdateMsg : uplinkMsg.getDashboardUpdateMsgList()) { |
||||
|
result.add(ctx.getDashboardProcessor().processDashboardMsgFromEdge(edge.getTenantId(), edge, dashboardUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getResourceUpdateMsgCount() > 0) { |
||||
|
for (ResourceUpdateMsg resourceUpdateMsg : uplinkMsg.getResourceUpdateMsgList()) { |
||||
|
result.add(ctx.getResourceProcessor().processResourceMsgFromEdge(edge.getTenantId(), edge, resourceUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getRuleChainMetadataRequestMsgCount() > 0) { |
||||
|
for (RuleChainMetadataRequestMsg ruleChainMetadataRequestMsg : uplinkMsg.getRuleChainMetadataRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processRuleChainMetadataRequestMsg(edge.getTenantId(), edge, ruleChainMetadataRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getAttributesRequestMsgCount() > 0) { |
||||
|
for (AttributesRequestMsg attributesRequestMsg : uplinkMsg.getAttributesRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processAttributesRequestMsg(edge.getTenantId(), edge, attributesRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getRelationRequestMsgCount() > 0) { |
||||
|
for (RelationRequestMsg relationRequestMsg : uplinkMsg.getRelationRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processRelationRequestMsg(edge.getTenantId(), edge, relationRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getCalculatedFieldRequestMsgCount() > 0) { |
||||
|
for (CalculatedFieldRequestMsg calculatedFieldRequestMsg : uplinkMsg.getCalculatedFieldRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processCalculatedFieldRequestMsg(edge.getTenantId(), edge, calculatedFieldRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getUserCredentialsRequestMsgCount() > 0) { |
||||
|
for (UserCredentialsRequestMsg userCredentialsRequestMsg : uplinkMsg.getUserCredentialsRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processUserCredentialsRequestMsg(edge.getTenantId(), edge, userCredentialsRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getDeviceCredentialsRequestMsgCount() > 0) { |
||||
|
for (DeviceCredentialsRequestMsg deviceCredentialsRequestMsg : uplinkMsg.getDeviceCredentialsRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processDeviceCredentialsRequestMsg(edge.getTenantId(), edge, deviceCredentialsRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getDeviceRpcCallMsgCount() > 0) { |
||||
|
for (DeviceRpcCallMsg deviceRpcCallMsg : uplinkMsg.getDeviceRpcCallMsgList()) { |
||||
|
result.add(ctx.getDeviceProcessor().processDeviceRpcCallFromEdge(edge.getTenantId(), edge, deviceRpcCallMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getWidgetBundleTypesRequestMsgCount() > 0) { |
||||
|
for (WidgetBundleTypesRequestMsg widgetBundleTypesRequestMsg : uplinkMsg.getWidgetBundleTypesRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processWidgetBundleTypesRequestMsg(edge.getTenantId(), edge, widgetBundleTypesRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getEntityViewsRequestMsgCount() > 0) { |
||||
|
for (EntityViewsRequestMsg entityViewRequestMsg : uplinkMsg.getEntityViewsRequestMsgList()) { |
||||
|
result.add(ctx.getEdgeRequestsService().processEntityViewsRequestMsg(edge.getTenantId(), edge, entityViewRequestMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getCalculatedFieldUpdateMsgCount() > 0) { |
||||
|
for (CalculatedFieldUpdateMsg calculatedFieldUpdateMsg : uplinkMsg.getCalculatedFieldUpdateMsgList()) { |
||||
|
result.add(ctx.getCalculatedFieldProcessor().processCalculatedFieldMsgFromEdge(edge.getTenantId(), edge, calculatedFieldUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getAiModelUpdateMsgCount() > 0) { |
||||
|
for (AiModelUpdateMsg aiModelUpdateMsg : uplinkMsg.getAiModelUpdateMsgList()) { |
||||
|
result.add(ctx.getAiModelProcessor().processAiModelMsgFromEdge(edge.getTenantId(), edge, aiModelUpdateMsg)); |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getUserUpdateMsgCount() > 0) { |
||||
|
for (UserUpdateMsg userUpdateMsg : uplinkMsg.getUserUpdateMsgList()) { |
||||
|
state.lockSequenceDependency(); |
||||
|
try { |
||||
|
result.add(ctx.getUserProcessor().processUserMsgFromEdge(edge.getTenantId(), edge, userUpdateMsg)); |
||||
|
} finally { |
||||
|
state.unlockSequenceDependency(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getUserCredentialsUpdateMsgCount() > 0) { |
||||
|
for (UserCredentialsUpdateMsg userCredentialsUpdateMsg : uplinkMsg.getUserCredentialsUpdateMsgList()) { |
||||
|
state.lockSequenceDependency(); |
||||
|
try { |
||||
|
result.add(ctx.getUserProcessor().processUserCredentialsMsgFromEdge(edge.getTenantId(), edge, userCredentialsUpdateMsg)); |
||||
|
} finally { |
||||
|
state.unlockSequenceDependency(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
if (uplinkMsg.getApiKeyUpdateMsgCount() > 0) { |
||||
|
for (ApiKeyUpdateMsg apiKeyUpdateMsg : uplinkMsg.getApiKeyUpdateMsgList()) { |
||||
|
state.lockSequenceDependency(); |
||||
|
try { |
||||
|
result.add(ctx.getApiKeyProcessor().processApiKeyMsgFromEdge(edge.getTenantId(), edge, apiKeyUpdateMsg)); |
||||
|
} finally { |
||||
|
state.unlockSequenceDependency(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
String failureMsg = String.format("Can't process uplink msg [%s] from edge", uplinkMsg); |
||||
|
log.trace("[{}][{}] Can't process uplink msg [{}]", edge.getTenantId(), edge.getId(), uplinkMsg, e); |
||||
|
ctx.getRuleProcessor().process(EdgeCommunicationFailureTrigger.builder().tenantId(edge.getTenantId()).edgeId(edge.getId()) |
||||
|
.customerId(edge.getCustomerId()).edgeName(edge.getName()).failureMsg(failureMsg).error(e.getMessage()).build()); |
||||
|
return Futures.immediateFailedFuture(e); |
||||
|
} |
||||
|
return Futures.allAsList(result); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,131 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc; |
||||
|
|
||||
|
import io.grpc.Server; |
||||
|
import io.grpc.netty.shaded.io.grpc.netty.GrpcSslContexts; |
||||
|
import io.grpc.netty.shaded.io.grpc.netty.NettyServerBuilder; |
||||
|
import io.grpc.netty.shaded.io.netty.handler.ssl.SslContext; |
||||
|
import io.grpc.netty.shaded.io.netty.handler.ssl.SslContextBuilder; |
||||
|
import jakarta.annotation.PreDestroy; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.StringUtils; |
||||
|
import org.thingsboard.server.common.transport.config.ssl.PemSslCredentials; |
||||
|
import org.thingsboard.server.gen.edge.v1.EdgeRpcServiceGrpc.EdgeRpcServiceImplBase; |
||||
|
import org.thingsboard.server.queue.util.AfterStartUp; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
|
||||
|
import java.io.IOException; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
|
||||
|
@Component |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
@ConditionalOnProperty(prefix = "edges", value = "enabled", havingValue = "true") |
||||
|
@TbCoreComponent |
||||
|
public class GrpcServer { |
||||
|
|
||||
|
@Value("${edges.rpc.port}") |
||||
|
private int rpcPort; |
||||
|
@Value("${edges.rpc.ssl.enabled}") |
||||
|
private boolean sslEnabled; |
||||
|
@Value("${edges.rpc.ssl.cert}") |
||||
|
private String certFileResource; |
||||
|
@Value("${edges.rpc.ssl.private_key}") |
||||
|
private String privateKeyResource; |
||||
|
@Value("${edges.rpc.ssl.key_password:}") |
||||
|
private String keyPassword; |
||||
|
@Value("${edges.rpc.client_max_keep_alive_time_sec:1}") |
||||
|
private int clientMaxKeepAliveTimeSec; |
||||
|
@Value("${edges.rpc.max_inbound_message_size:4194304}") |
||||
|
private int maxInboundMessageSize; |
||||
|
@Value("${edges.rpc.keep_alive_time_sec:10}") |
||||
|
private int keepAliveTimeSec; |
||||
|
@Value("${edges.rpc.keep_alive_timeout_sec:5}") |
||||
|
private int keepAliveTimeoutSec; |
||||
|
|
||||
|
private final EdgeRpcServiceImplBase edgeGrpcService; |
||||
|
|
||||
|
private Server server; |
||||
|
|
||||
|
@AfterStartUp(order = AfterStartUp.REGULAR_SERVICE) |
||||
|
public void onStartUp() { |
||||
|
log.info("Initializing Edge RPC server!"); |
||||
|
NettyServerBuilder serverBuilder = NettyServerBuilder.forPort(rpcPort) |
||||
|
.permitKeepAliveTime(clientMaxKeepAliveTimeSec, TimeUnit.SECONDS) |
||||
|
.keepAliveTime(keepAliveTimeSec, TimeUnit.SECONDS) |
||||
|
.keepAliveTimeout(keepAliveTimeoutSec, TimeUnit.SECONDS) |
||||
|
.permitKeepAliveWithoutCalls(true) |
||||
|
.maxInboundMessageSize(maxInboundMessageSize) |
||||
|
.addService(edgeGrpcService); |
||||
|
|
||||
|
if (sslEnabled) { |
||||
|
setupSsl(serverBuilder); |
||||
|
} |
||||
|
server = serverBuilder.build(); |
||||
|
log.info("Going to start Edge RPC server using port: {}", rpcPort); |
||||
|
try { |
||||
|
server.start(); |
||||
|
} catch (IOException e) { |
||||
|
log.error("Failed to start Edge RPC server!", e); |
||||
|
throw new RuntimeException("Failed to start Edge RPC server!", e); |
||||
|
} |
||||
|
log.info("Edge RPC server initialized!"); |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* Configures TLS for the Edge gRPC server. |
||||
|
* <p> |
||||
|
* Delegates PEM parsing and key management to {@link PemSslCredentials} — the same |
||||
|
* class used by MQTT, CoAP, and LwM2M transports — which supports: |
||||
|
* <ul> |
||||
|
* <li>Separate certificate and private key files (classic two-file setup)</li> |
||||
|
* <li>Combined PEM: certificate chain + private key in a single {@code cert} file |
||||
|
* ({@code private_key} left empty)</li> |
||||
|
* <li>Encrypted private keys (password supplied via {@code key_password})</li> |
||||
|
* </ul> |
||||
|
* Path resolution (for both {@code cert} and {@code private_key}) is handled by |
||||
|
* {@link org.thingsboard.server.common.data.ResourceUtils#getInputStream ResourceUtils}: |
||||
|
* absolute path → relative / working-dir → classpath → {@code classpath:} prefix. |
||||
|
*/ |
||||
|
void setupSsl(NettyServerBuilder builder) { |
||||
|
try { |
||||
|
PemSslCredentials credentials = new PemSslCredentials(); |
||||
|
credentials.setCertFile(certFileResource); |
||||
|
credentials.setKeyFile(StringUtils.isEmpty(privateKeyResource) ? null : privateKeyResource); |
||||
|
credentials.setKeyPassword(keyPassword); |
||||
|
credentials.init(false); |
||||
|
|
||||
|
SslContext sslContext = GrpcSslContexts.configure( |
||||
|
SslContextBuilder.forServer(credentials.createKeyManagerFactory())).build(); |
||||
|
builder.sslContext(sslContext); |
||||
|
} catch (Exception e) { |
||||
|
log.error("Unable to set up SSL context. Reason: " + e.getMessage(), e); |
||||
|
throw new RuntimeException("Unable to set up SSL context!", e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@PreDestroy |
||||
|
public void destroy() { |
||||
|
if (server != null) { |
||||
|
server.shutdownNow(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -1,196 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2026 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.edge.rpc; |
|
||||
|
|
||||
import com.google.common.util.concurrent.Futures; |
|
||||
import com.google.common.util.concurrent.ListenableFuture; |
|
||||
import io.grpc.stub.StreamObserver; |
|
||||
import lombok.Getter; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.thingsboard.common.util.ThingsBoardThreadFactory; |
|
||||
import org.thingsboard.server.common.data.edge.Edge; |
|
||||
import org.thingsboard.server.common.data.edge.EdgeEvent; |
|
||||
import org.thingsboard.server.common.data.id.EdgeId; |
|
||||
import org.thingsboard.server.common.util.ProtoUtils; |
|
||||
import org.thingsboard.server.gen.edge.v1.DownlinkMsg; |
|
||||
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
|
||||
import org.thingsboard.server.gen.transport.TransportProtos.ToEdgeEventNotificationMsg; |
|
||||
import org.thingsboard.server.queue.TbQueueConsumer; |
|
||||
import org.thingsboard.server.queue.common.TbProtoQueueMsg; |
|
||||
import org.thingsboard.server.queue.common.consumer.QueueConsumerManager; |
|
||||
import org.thingsboard.server.queue.discovery.TopicService; |
|
||||
import org.thingsboard.server.queue.kafka.KafkaAdmin; |
|
||||
import org.thingsboard.server.queue.provider.TbCoreQueueFactory; |
|
||||
import org.thingsboard.server.service.edge.EdgeContextComponent; |
|
||||
|
|
||||
import java.util.ArrayList; |
|
||||
import java.util.List; |
|
||||
import java.util.UUID; |
|
||||
import java.util.concurrent.ExecutorService; |
|
||||
import java.util.concurrent.Executors; |
|
||||
import java.util.concurrent.ScheduledExecutorService; |
|
||||
import java.util.function.BiConsumer; |
|
||||
|
|
||||
@Slf4j |
|
||||
public class KafkaEdgeGrpcSession extends EdgeGrpcSession { |
|
||||
|
|
||||
private final TopicService topicService; |
|
||||
private final TbCoreQueueFactory tbCoreQueueFactory; |
|
||||
private final KafkaAdmin kafkaAdmin; |
|
||||
|
|
||||
private volatile boolean isHighPriorityProcessing; |
|
||||
|
|
||||
@Getter |
|
||||
private QueueConsumerManager<TbProtoQueueMsg<ToEdgeEventNotificationMsg>> consumer; |
|
||||
|
|
||||
private ExecutorService consumerExecutor; |
|
||||
|
|
||||
public KafkaEdgeGrpcSession(EdgeContextComponent ctx, TopicService topicService, TbCoreQueueFactory tbCoreQueueFactory, |
|
||||
KafkaAdmin kafkaAdmin, StreamObserver<ResponseMsg> outputStream, |
|
||||
BiConsumer<EdgeId, EdgeGrpcSession> sessionOpenListener, BiConsumer<Edge, UUID> sessionCloseListener, |
|
||||
ScheduledExecutorService sendDownlinkExecutorService, int maxInboundMessageSize, int maxHighPriorityQueueSizePerSession) { |
|
||||
super(ctx, outputStream, sessionOpenListener, sessionCloseListener, sendDownlinkExecutorService, maxInboundMessageSize, maxHighPriorityQueueSizePerSession); |
|
||||
this.topicService = topicService; |
|
||||
this.tbCoreQueueFactory = tbCoreQueueFactory; |
|
||||
this.kafkaAdmin = kafkaAdmin; |
|
||||
} |
|
||||
|
|
||||
private void processMsgs(List<TbProtoQueueMsg<ToEdgeEventNotificationMsg>> msgs, TbQueueConsumer<TbProtoQueueMsg<ToEdgeEventNotificationMsg>> consumer) { |
|
||||
log.trace("[{}][{}] starting processing edge events", tenantId, edge.getId()); |
|
||||
// Defensive backstop: the loop already gates polling on readiness; this only fires on the narrow race
|
|
||||
// where readiness flips during poll(), and that already-polled batch is dropped here (can't rewind).
|
|
||||
if (!isReadyToProcessGeneralEvents()) { |
|
||||
log.debug("[{}][{}] edge not connected, edge sync is not completed or high priority processing in progress, " + |
|
||||
"connected = {}, sync in progress = {}, high priority in progress = {}. Skipping iteration", |
|
||||
tenantId, edge.getId(), isConnected(), isSyncInProgress(), isHighPriorityProcessing); |
|
||||
return; |
|
||||
} |
|
||||
List<EdgeEvent> edgeEvents = new ArrayList<>(); |
|
||||
for (TbProtoQueueMsg<ToEdgeEventNotificationMsg> msg : msgs) { |
|
||||
EdgeEvent edgeEvent = ProtoUtils.fromProto(msg.getValue().getEdgeEventMsg()); |
|
||||
edgeEvents.add(edgeEvent); |
|
||||
} |
|
||||
List<DownlinkMsg> downlinkMsgsPack = convertToDownlinkMsgsPack(edgeEvents); |
|
||||
boolean isInterrupted = true; |
|
||||
try { |
|
||||
isInterrupted = sendDownlinkMsgsPack(downlinkMsgsPack).get(); |
|
||||
if (isInterrupted) { |
|
||||
log.debug("[{}][{}] Send downlink messages task was interrupted", tenantId, edge.getId()); |
|
||||
} |
|
||||
} catch (Exception e) { |
|
||||
log.error("[{}][{}] Failed to process downlink messages", tenantId, edge.getId(), e); |
|
||||
} |
|
||||
if (!isInterrupted) { |
|
||||
consumer.commit(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private boolean isReadyToProcessGeneralEvents() { |
|
||||
return isConnected() && !isSyncInProgress() && !isHighPriorityProcessing; |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public ListenableFuture<Boolean> migrateEdgeEvents() throws Exception { |
|
||||
return super.processEdgeEvents(); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public ListenableFuture<Boolean> processEdgeEvents() { |
|
||||
if (!isReadyToProcessGeneralEvents()) { |
|
||||
log.warn("[{}][{}] Session is not ready (connected={}, syncInProgress={}, highPriority={}), skip starting edge event consumer", |
|
||||
tenantId, edge != null ? edge.getId() : null, isConnected(), isSyncInProgress(), isHighPriorityProcessing); |
|
||||
return Futures.immediateFuture(Boolean.FALSE); |
|
||||
} |
|
||||
if (consumer == null || (consumer.getConsumer() != null && consumer.getConsumer().isStopped())) { |
|
||||
try { |
|
||||
if (consumerExecutor != null && !consumerExecutor.isShutdown()) { |
|
||||
try { |
|
||||
consumerExecutor.shutdown(); |
|
||||
awaitConsumerTermination(); |
|
||||
} catch (Exception e) { |
|
||||
log.warn("[{}][{}] Failed to shutdown previous consumer executor", tenantId, edge.getId(), e); |
|
||||
} |
|
||||
} |
|
||||
this.consumerExecutor = Executors.newSingleThreadExecutor(ThingsBoardThreadFactory.forName("edge-event-consumer")); |
|
||||
this.consumer = QueueConsumerManager.<TbProtoQueueMsg<ToEdgeEventNotificationMsg>>builder() |
|
||||
.name("TB Edge events [" + edge.getId() + "]") |
|
||||
.msgPackProcessor(this::processMsgs) |
|
||||
.pollInterval(ctx.getEdgeEventStorageSettings().getNoRecordsSleepInterval()) |
|
||||
.consumerCreator(() -> tbCoreQueueFactory.createEdgeEventMsgConsumer(tenantId, edge.getId())) |
|
||||
.consumerExecutor(consumerExecutor) |
|
||||
.threadPrefix("edge-events-" + edge.getId()) |
|
||||
.readinessCheck(this::isReadyToProcessGeneralEvents) |
|
||||
.build(); |
|
||||
consumer.subscribe(); |
|
||||
consumer.launch(); |
|
||||
} catch (Exception e) { |
|
||||
destroy(); |
|
||||
log.warn("[{}][{}] Failed to start edge event consumer", sessionId, edge.getId(), e); |
|
||||
} |
|
||||
} |
|
||||
return Futures.immediateFuture(Boolean.FALSE); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void processHighPriorityEvents() { |
|
||||
isHighPriorityProcessing = true; |
|
||||
try { |
|
||||
super.processHighPriorityEvents(); |
|
||||
} finally { |
|
||||
isHighPriorityProcessing = false; |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public boolean destroy() { |
|
||||
try { |
|
||||
if (consumer != null) { |
|
||||
log.info("[{}][{}] Stopping edge event consumer...", tenantId, edge != null ? edge.getId() : null); |
|
||||
consumer.stop(); |
|
||||
} |
|
||||
} catch (Exception e) { |
|
||||
log.warn("[{}][{}] Failed to stop edge event consumer", tenantId, edge.getId(), e); |
|
||||
return false; |
|
||||
} |
|
||||
consumer = null; |
|
||||
try { |
|
||||
if (consumerExecutor != null && !consumerExecutor.isShutdown()) { |
|
||||
consumerExecutor.shutdown(); |
|
||||
awaitConsumerTermination(); |
|
||||
} |
|
||||
} catch (Exception e) { |
|
||||
log.warn("[{}][{}] Failed to shutdown edge event consumer executor", tenantId, edge.getId(), e); |
|
||||
return false; |
|
||||
} |
|
||||
return true; |
|
||||
} |
|
||||
|
|
||||
private void awaitConsumerTermination() { |
|
||||
try { |
|
||||
consumerExecutor.awaitTermination(5, java.util.concurrent.TimeUnit.SECONDS); |
|
||||
} catch (InterruptedException ie) { |
|
||||
log.warn("[{}][{}] Interrupted while awaiting consumer executor termination", tenantId, edge.getId()); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public void cleanUp() { |
|
||||
String topic = topicService.buildEdgeEventNotificationsTopicPartitionInfo(tenantId, edge.getId()).getTopic(); |
|
||||
kafkaAdmin.deleteTopic(topic); |
|
||||
kafkaAdmin.deleteConsumerGroup(topic); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -1,46 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2026 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.edge.rpc; |
|
||||
|
|
||||
import com.google.common.util.concurrent.Futures; |
|
||||
import com.google.common.util.concurrent.ListenableFuture; |
|
||||
import io.grpc.stub.StreamObserver; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.thingsboard.server.common.data.edge.Edge; |
|
||||
import org.thingsboard.server.common.data.id.EdgeId; |
|
||||
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
|
||||
import org.thingsboard.server.service.edge.EdgeContextComponent; |
|
||||
|
|
||||
import java.util.UUID; |
|
||||
import java.util.concurrent.ScheduledExecutorService; |
|
||||
import java.util.function.BiConsumer; |
|
||||
|
|
||||
@Slf4j |
|
||||
public class PostgresEdgeGrpcSession extends EdgeGrpcSession { |
|
||||
|
|
||||
PostgresEdgeGrpcSession(EdgeContextComponent ctx, StreamObserver<ResponseMsg> outputStream, |
|
||||
BiConsumer<EdgeId, EdgeGrpcSession> sessionOpenListener, |
|
||||
BiConsumer<Edge, UUID> sessionCloseListener, ScheduledExecutorService sendDownlinkExecutorService, |
|
||||
int maxInboundMessageSize, int maxHighPriorityQueueSizePerSession) { |
|
||||
super(ctx, outputStream, sessionOpenListener, sessionCloseListener, sendDownlinkExecutorService, maxInboundMessageSize, maxHighPriorityQueueSizePerSession); |
|
||||
} |
|
||||
|
|
||||
@Override |
|
||||
public ListenableFuture<Boolean> migrateEdgeEvents() { |
|
||||
return Futures.immediateFuture(Boolean.FALSE); |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,23 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.processor; |
||||
|
|
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
|
||||
|
public interface EdgeEventsDispatcher { |
||||
|
|
||||
|
ListenableFuture<Boolean> processNewEvents() throws Exception; |
||||
|
} |
||||
@ -0,0 +1,218 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.processor; |
||||
|
|
||||
|
import com.google.common.util.concurrent.FutureCallback; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import com.google.common.util.concurrent.SettableFuture; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.checkerframework.checker.nullness.qual.Nullable; |
||||
|
import org.springframework.data.util.Pair; |
||||
|
import org.thingsboard.server.common.data.AttributeScope; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.AttributesSaveResult; |
||||
|
import org.thingsboard.server.common.data.kv.BaseAttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.LongDataEntry; |
||||
|
import org.thingsboard.server.common.data.page.PageData; |
||||
|
import org.thingsboard.server.common.data.page.TimePageLink; |
||||
|
import org.thingsboard.server.service.edge.EdgeContextComponent; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.fetch.GeneralEdgeEventFetcher; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeSession; |
||||
|
|
||||
|
import java.util.Arrays; |
||||
|
import java.util.List; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
|
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
public class PostgresGeneralEdgeEventsDispatcher implements EdgeEventsDispatcher { |
||||
|
|
||||
|
private static final String QUEUE_START_TS_ATTR_KEY = "queueStartTs"; |
||||
|
private static final String QUEUE_START_SEQ_ID_ATTR_KEY = "queueStartSeqId"; |
||||
|
|
||||
|
private final EdgeSession session; |
||||
|
private final EdgeContextComponent ctx; |
||||
|
|
||||
|
private Long newStartTs; |
||||
|
private Long previousStartTs; |
||||
|
private Long newStartSeqId; |
||||
|
private Long previousStartSeqId; |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Boolean> processNewEvents() throws Exception { |
||||
|
EdgeSessionState state = session.getState(); |
||||
|
TenantId tenantId = state.getTenantId(); |
||||
|
EdgeId edgeId = state.getEdgeId(); |
||||
|
UUID sessionId = state.getSessionId(); |
||||
|
|
||||
|
SettableFuture<Boolean> result = SettableFuture.create(); |
||||
|
if (state.isConnected() && !state.isSyncInProgress()) { |
||||
|
setLocalPrevStartTsAndSeqId(tenantId, edgeId, sessionId); |
||||
|
GeneralEdgeEventFetcher fetcher = initGeneralEdgeEventFetcher(); |
||||
|
log.trace("[{}][{}] starting processing edge events, previousStartTs = {}, previousStartSeqId = {}", |
||||
|
tenantId, edgeId, previousStartTs, previousStartSeqId); |
||||
|
|
||||
|
Futures.addCallback(session.fetchAndSendEdgeEvents(fetcher), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(@Nullable Pair<Long, Long> newStartTsAndSeqId) { |
||||
|
if (newStartTsAndSeqId != null) { |
||||
|
Futures.addCallback(updateQueueStartTsAndSeqId(tenantId, edgeId, newStartTsAndSeqId), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(@Nullable AttributesSaveResult saveResult) { |
||||
|
log.debug("[{}][{}] queue offset was updated [{}]", tenantId, edgeId, newStartTsAndSeqId); |
||||
|
boolean newEventsAvailable; |
||||
|
if (fetcher.isSeqIdNewCycleStarted()) { |
||||
|
newEventsAvailable = isNewEdgeEventsAvailable(tenantId, edgeId, sessionId); |
||||
|
} else { |
||||
|
newEventsAvailable = isSeqIdStartedNewCycle(tenantId, edgeId, sessionId); |
||||
|
if (!newEventsAvailable) { |
||||
|
newEventsAvailable = isNewEdgeEventsAvailable(tenantId, edgeId, sessionId); |
||||
|
} |
||||
|
} |
||||
|
result.set(newEventsAvailable); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.error("[{}][{}] Failed to update queue offset [{}]", tenantId, edgeId, newStartTsAndSeqId, t); |
||||
|
result.setException(t); |
||||
|
} |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} else { |
||||
|
log.trace("[{}][{}] newStartTsAndSeqId is null. Skipping iteration without db update", tenantId, edgeId); |
||||
|
result.set(Boolean.FALSE); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.error("[{}][{}] Failed to process events", tenantId, edgeId, t); |
||||
|
result.setException(t); |
||||
|
} |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} else { |
||||
|
if (state.isSyncInProgress()) { |
||||
|
log.trace("[{}][{}] edge sync is not completed yet. Skipping iteration", tenantId, edgeId); |
||||
|
result.set(Boolean.TRUE); |
||||
|
} else { |
||||
|
log.trace("[{}][{}] edge is not connected. Skipping iteration", tenantId, edgeId); |
||||
|
result.set(null); |
||||
|
} |
||||
|
} |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
private boolean isSeqIdStartedNewCycle(TenantId tenantId, EdgeId edgeId, UUID sessionId) { |
||||
|
try { |
||||
|
log.trace("[{}][{}][{}] Checking if seq id started new cycle", tenantId, edgeId, sessionId); |
||||
|
TimePageLink pageLink = new TimePageLink(ctx.getEdgeEventStorageSettings().getMaxReadRecordsCount(), 0, null, null, newStartTs, System.currentTimeMillis()); |
||||
|
PageData<EdgeEvent> edgeEvents = ctx.getEdgeEventService() |
||||
|
.findEdgeEvents(tenantId, edgeId, 0L, previousStartSeqId == 0 ? null : previousStartSeqId - 1, pageLink); |
||||
|
|
||||
|
boolean result = !edgeEvents.getData().isEmpty(); |
||||
|
log.trace("[{}][{}][{}] Result of check if seq id started new cycle, result = {}", tenantId, edgeId, sessionId, result); |
||||
|
return result; |
||||
|
} catch (Exception e) { |
||||
|
log.error("[{}][{}][{}] Failed to execute isSeqIdStartedNewCycle", tenantId, edgeId, sessionId, e); |
||||
|
} |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
private boolean isNewEdgeEventsAvailable(TenantId tenantId, EdgeId edgeId, UUID sessionId) { |
||||
|
try { |
||||
|
log.trace("[{}][{}][{}] Checking if new edge events available", tenantId, edgeId, sessionId); |
||||
|
TimePageLink pageLink = new TimePageLink(ctx.getEdgeEventStorageSettings().getMaxReadRecordsCount(), 0, null, null, newStartTs, System.currentTimeMillis()); |
||||
|
PageData<EdgeEvent> edgeEvents = ctx.getEdgeEventService().findEdgeEvents(tenantId, edgeId, newStartSeqId, null, pageLink); |
||||
|
boolean available = !edgeEvents.getData().isEmpty() || session.hasHighPriorityEvents(); |
||||
|
log.trace("[{}][{}][{}] Result of check if new edge events available, result = {}", tenantId, edgeId, sessionId, available); |
||||
|
return available; |
||||
|
} catch (Exception e) { |
||||
|
log.error("[{}][{}][{}] Failed to execute isNewEdgeEventsAvailable", tenantId, edgeId, sessionId, e); |
||||
|
} |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
private ListenableFuture<AttributesSaveResult> updateQueueStartTsAndSeqId(TenantId tenantId, EdgeId edgeId, Pair<Long, Long> pair) { |
||||
|
newStartTs = pair.getFirst(); |
||||
|
newStartSeqId = pair.getSecond(); |
||||
|
log.trace("[{}] updateQueueStartTsAndSeqId [{}][{}][{}]", session.getState().getSessionId(), edgeId, newStartTs, newStartSeqId); |
||||
|
List<AttributeKvEntry> attributes = List.of( |
||||
|
new BaseAttributeKvEntry(new LongDataEntry(QUEUE_START_TS_ATTR_KEY, newStartTs), System.currentTimeMillis()), |
||||
|
new BaseAttributeKvEntry(new LongDataEntry(QUEUE_START_SEQ_ID_ATTR_KEY, newStartSeqId), System.currentTimeMillis()) |
||||
|
); |
||||
|
return ctx.getAttributesService().save(tenantId, edgeId, AttributeScope.SERVER_SCOPE, attributes); |
||||
|
} |
||||
|
|
||||
|
private GeneralEdgeEventFetcher initGeneralEdgeEventFetcher() { |
||||
|
return new GeneralEdgeEventFetcher( |
||||
|
previousStartTs, |
||||
|
previousStartSeqId, |
||||
|
false, |
||||
|
Integer.toUnsignedLong(ctx.getEdgeEventStorageSettings().getMaxReadRecordsCount()), |
||||
|
ctx.getEdgeEventService(), |
||||
|
ctx.getEdgeEventStorageSettings().getMisorderingCompensationMillis()); |
||||
|
} |
||||
|
|
||||
|
private void setLocalPrevStartTsAndSeqId(TenantId tenantId, EdgeId edgeId, UUID sessionId) throws InterruptedException, ExecutionException { |
||||
|
Pair<Long, Long> startTsAndSeqId = getQueueStartTsAndSeqId(tenantId, edgeId, sessionId).get(); |
||||
|
previousStartTs = startTsAndSeqId.getFirst(); |
||||
|
previousStartSeqId = startTsAndSeqId.getSecond(); |
||||
|
} |
||||
|
|
||||
|
private ListenableFuture<Pair<Long, Long>> getQueueStartTsAndSeqId(TenantId tenantId, EdgeId edgeId, UUID sessionId) { |
||||
|
ListenableFuture<List<AttributeKvEntry>> future = ctx.getAttributesService() |
||||
|
.find(tenantId, edgeId, AttributeScope.SERVER_SCOPE, Arrays.asList(QUEUE_START_TS_ATTR_KEY, QUEUE_START_SEQ_ID_ATTR_KEY)); |
||||
|
|
||||
|
return Futures.transform(future, attributeKvEntries -> { |
||||
|
long startTs = 0L; |
||||
|
long startSeqId = 0L; |
||||
|
for (AttributeKvEntry attributeKvEntry : attributeKvEntries) { |
||||
|
if (QUEUE_START_TS_ATTR_KEY.equals(attributeKvEntry.getKey())) { |
||||
|
startTs = attributeKvEntry.getLongValue().isPresent() ? attributeKvEntry.getLongValue().get() : 0L; |
||||
|
} |
||||
|
if (QUEUE_START_SEQ_ID_ATTR_KEY.equals(attributeKvEntry.getKey())) { |
||||
|
startSeqId = attributeKvEntry.getLongValue().isPresent() ? attributeKvEntry.getLongValue().get() : 0L; |
||||
|
} |
||||
|
} |
||||
|
if (startSeqId == 0L) { |
||||
|
startSeqId = findStartSeqIdFromOldestEventIfAny(tenantId, edgeId, sessionId); |
||||
|
} |
||||
|
return Pair.of(startTs, startSeqId); |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} |
||||
|
|
||||
|
private long findStartSeqIdFromOldestEventIfAny(TenantId tenantId, EdgeId edgeId, UUID sessionId) { |
||||
|
long startSeqId = 0L; |
||||
|
try { |
||||
|
TimePageLink pageLink = new TimePageLink(1, 0, null, null, null, null); |
||||
|
PageData<EdgeEvent> edgeEvents = ctx.getEdgeEventService() |
||||
|
.findEdgeEvents(tenantId, edgeId, null, null, pageLink); |
||||
|
if (!edgeEvents.getData().isEmpty()) { |
||||
|
startSeqId = edgeEvents.getData().get(0).getSeqId() - 1; |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("[{}][{}][{}] Failed to execute findStartSeqIdFromOldestEventIfAny", tenantId, edgeId, sessionId, e); |
||||
|
} |
||||
|
return startSeqId; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,392 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.service; |
||||
|
|
||||
|
import com.fasterxml.jackson.databind.node.ObjectNode; |
||||
|
import io.grpc.stub.StreamObserver; |
||||
|
import jakarta.annotation.PostConstruct; |
||||
|
import jakarta.annotation.PreDestroy; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.context.ApplicationContext; |
||||
|
import org.springframework.context.annotation.Lazy; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.common.util.JacksonUtil; |
||||
|
import org.thingsboard.common.util.ThingsBoardExecutors; |
||||
|
import org.thingsboard.rule.engine.api.AttributesSaveRequest; |
||||
|
import org.thingsboard.rule.engine.api.TimeseriesSaveRequest; |
||||
|
import org.thingsboard.server.cache.TbTransactionalCache; |
||||
|
import org.thingsboard.server.cluster.TbClusterService; |
||||
|
import org.thingsboard.server.common.data.AttributeScope; |
||||
|
import org.thingsboard.server.common.data.DataConstants; |
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.BooleanDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.LongDataEntry; |
||||
|
import org.thingsboard.server.common.data.msg.TbMsgType; |
||||
|
import org.thingsboard.server.common.data.notification.rule.trigger.EdgeConnectionTrigger; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbMsgDataType; |
||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData; |
||||
|
import org.thingsboard.server.common.msg.edge.EdgeEventUpdateMsg; |
||||
|
import org.thingsboard.server.common.msg.edge.EdgeHighPriorityMsg; |
||||
|
import org.thingsboard.server.common.msg.edge.EdgeSessionMsg; |
||||
|
import org.thingsboard.server.common.msg.edge.FromEdgeSyncResponse; |
||||
|
import org.thingsboard.server.common.msg.edge.ToEdgeSyncRequest; |
||||
|
import org.thingsboard.server.gen.edge.v1.EdgeRpcServiceGrpc; |
||||
|
import org.thingsboard.server.gen.edge.v1.RequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
||||
|
import org.thingsboard.server.queue.discovery.TbServiceInfoProvider; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.edge.EdgeContextComponent; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeRpcService; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeAttributeSaveCallback; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeSessionsHolder; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.manager.EdgeGrpcSessionManager; |
||||
|
import org.thingsboard.server.service.telemetry.TelemetrySubscriptionService; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||
|
import java.util.concurrent.ConcurrentMap; |
||||
|
import java.util.concurrent.ExecutorService; |
||||
|
import java.util.concurrent.ScheduledExecutorService; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
import java.util.function.Consumer; |
||||
|
|
||||
|
import static org.thingsboard.server.service.state.DefaultDeviceStateService.ACTIVITY_STATE; |
||||
|
import static org.thingsboard.server.service.state.DefaultDeviceStateService.LAST_CONNECT_TIME; |
||||
|
import static org.thingsboard.server.service.state.DefaultDeviceStateService.LAST_DISCONNECT_TIME; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
@ConditionalOnProperty(prefix = "edges", value = "enabled", havingValue = "true") |
||||
|
@TbCoreComponent |
||||
|
public class EdgeGrpcService extends EdgeRpcServiceGrpc.EdgeRpcServiceImplBase implements EdgeRpcService { |
||||
|
|
||||
|
@Value("${edges.send_scheduler_pool_size}") |
||||
|
private int sendSchedulerPoolSize; |
||||
|
|
||||
|
@Value("${edges.state.persistToTelemetry:false}") |
||||
|
private boolean persistToTelemetry; |
||||
|
|
||||
|
@Autowired |
||||
|
@Lazy |
||||
|
private EdgeContextComponent ctx; |
||||
|
|
||||
|
private final EdgeSessionsHolder sessions; |
||||
|
private final ApplicationContext applicationContext; |
||||
|
private final TbClusterService clusterService; |
||||
|
private final TbServiceInfoProvider serviceInfoProvider; |
||||
|
private final TelemetrySubscriptionService tsSubService; |
||||
|
private final TbTransactionalCache<EdgeId, String> edgeIdServiceIdCache; |
||||
|
|
||||
|
private final ConcurrentMap<UUID, Consumer<FromEdgeSyncResponse>> localSyncEdgeRequests = new ConcurrentHashMap<>(); |
||||
|
private ScheduledExecutorService executorService; |
||||
|
private ScheduledExecutorService sendDownlinkExecutorService; |
||||
|
|
||||
|
@PostConstruct |
||||
|
public void onStartUp() { |
||||
|
this.executorService = ThingsBoardExecutors.newSingleThreadScheduledExecutor("edge-service"); |
||||
|
this.sendDownlinkExecutorService = ThingsBoardExecutors.newScheduledThreadPool(sendSchedulerPoolSize, "edge-send-scheduler"); |
||||
|
} |
||||
|
|
||||
|
@PreDestroy |
||||
|
private void preDestroy() { |
||||
|
shutdownExecutorSafely(executorService); |
||||
|
shutdownExecutorSafely(sendDownlinkExecutorService); |
||||
|
sessions.forEach(EdgeGrpcSessionManager::onEdgeDisconnect); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public StreamObserver<RequestMsg> handleMsgs(StreamObserver<ResponseMsg> outputStream) { |
||||
|
EdgeGrpcSessionManager sessionManager = applicationContext.getBean(EdgeGrpcSessionManager.class); |
||||
|
return sessionManager.initInputStream(outputStream, this::onEdgeConnect, this::onEdgeDisconnect, sendDownlinkExecutorService); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onToEdgeSessionMsg(TenantId tenantId, EdgeSessionMsg msg) { |
||||
|
switch (msg.getMsgType()) { |
||||
|
case EDGE_HIGH_PRIORITY_TO_EDGE_SESSION_MSG -> { |
||||
|
EdgeHighPriorityMsg edgeHighPriorityMsg = (EdgeHighPriorityMsg) msg; |
||||
|
log.trace("[{}] edgeEventMsg [{}]", tenantId, msg); |
||||
|
onEdgeHighPriorityEvent(edgeHighPriorityMsg); |
||||
|
} |
||||
|
case EDGE_EVENT_UPDATE_TO_EDGE_SESSION_MSG -> { |
||||
|
EdgeEventUpdateMsg edgeEventUpdateMsg = (EdgeEventUpdateMsg) msg; |
||||
|
log.trace("[{}] onToEdgeEventUpdateMsg [{}]", tenantId, msg); |
||||
|
onEdgeEventUpdate(tenantId, edgeEventUpdateMsg.getEdgeId()); |
||||
|
} |
||||
|
case EDGE_SYNC_REQUEST_TO_EDGE_SESSION_MSG -> { |
||||
|
ToEdgeSyncRequest toEdgeSyncRequest = (ToEdgeSyncRequest) msg; |
||||
|
log.trace("[{}] toEdgeSyncRequest [{}]", tenantId, msg); |
||||
|
startSyncProcess(tenantId, toEdgeSyncRequest.getEdgeId(), toEdgeSyncRequest.getId(), toEdgeSyncRequest.getServiceId()); |
||||
|
} |
||||
|
case EDGE_SYNC_RESPONSE_FROM_EDGE_SESSION_MSG -> { |
||||
|
FromEdgeSyncResponse fromEdgeSyncResponse = (FromEdgeSyncResponse) msg; |
||||
|
log.trace("[{}] fromEdgeSyncResponse [{}]", tenantId, msg); |
||||
|
processSyncResponse(fromEdgeSyncResponse); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void updateEdge(TenantId tenantId, Edge edge) { |
||||
|
if (edge == null) { |
||||
|
log.warn("[{}] Edge is null - edge is removed and outdated notification is in process!", tenantId); |
||||
|
return; |
||||
|
} |
||||
|
EdgeGrpcSessionManager session = sessions.getByEdgeId(edge.getId()); |
||||
|
if (session != null && session.getState().isConnected()) { |
||||
|
log.debug("[{}] Updating configuration for edge [{}] [{}]", tenantId, edge.getName(), edge.getId()); |
||||
|
session.onConfigurationUpdate(edge); |
||||
|
} else { |
||||
|
log.debug("[{}] Session doesn't exist for edge [{}] [{}]", tenantId, edge.getName(), edge.getId()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void deleteEdge(TenantId tenantId, EdgeId edgeId) { |
||||
|
EdgeGrpcSessionManager session = sessions.getByEdgeId(edgeId); |
||||
|
if (session != null && session.getState().isConnected()) { |
||||
|
log.info("[{}] Closing and removing session for edge [{}]", tenantId, edgeId); |
||||
|
session.destroyAndMarkAsZombieIfFailed(); |
||||
|
session.onEdgeRemoval(); |
||||
|
sessions.remove(session); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void processSyncRequest(TenantId tenantId, EdgeId edgeId, Consumer<FromEdgeSyncResponse> responseConsumer) { |
||||
|
ToEdgeSyncRequest request = new ToEdgeSyncRequest(UUID.randomUUID(), tenantId, edgeId, serviceInfoProvider.getServiceId()); |
||||
|
|
||||
|
UUID requestId = request.getId(); |
||||
|
EdgeGrpcSessionManager session = sessions.getByEdgeId(request.getEdgeId()); |
||||
|
if (session != null && session.getState().isSyncInProgress()) { |
||||
|
responseConsumer.accept(new FromEdgeSyncResponse(requestId, request.getTenantId(), request.getEdgeId(), false, "Sync process is active at the moment")); |
||||
|
} else { |
||||
|
log.trace("[{}][{}] Processing sync edge request [{}], serviceId [{}]", request.getTenantId(), request.getId(), request.getEdgeId(), request.getServiceId()); |
||||
|
localSyncEdgeRequests.put(requestId, responseConsumer); |
||||
|
clusterService.pushEdgeSyncRequestToEdge(request); |
||||
|
scheduleSyncRequestTimeout(request, requestId); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void onEdgeConnect(EdgeId edgeId, EdgeGrpcSessionManager edgeSession) { |
||||
|
EdgeSessionState state = edgeSession.getState(); |
||||
|
Edge edge = state.getEdge(); |
||||
|
TenantId tenantId = state.getTenantId(); |
||||
|
log.info("[{}][{}] edge [{}] connected successfully.", tenantId, state.getSessionId(), edgeId); |
||||
|
if (sessions.hasByEdgeId(edgeId)) { |
||||
|
EdgeGrpcSessionManager existingSession = sessions.getByEdgeId(edgeId); |
||||
|
if (existingSession != null) { |
||||
|
UUID sessionId = existingSession.getState().getSessionId(); |
||||
|
log.info("[{}][{}] Replacing existing session [{}] for edge [{}]", tenantId, state.getSessionId(), sessionId, edgeId); |
||||
|
existingSession.destroyAndMarkAsZombieIfFailed(); |
||||
|
sessions.removeBySessionId(sessionId); |
||||
|
} |
||||
|
} |
||||
|
sessions.put(edgeSession); |
||||
|
save(tenantId, edgeId, ACTIVITY_STATE, true); |
||||
|
long lastConnectTs = System.currentTimeMillis(); |
||||
|
save(tenantId, edgeId, LAST_CONNECT_TIME, lastConnectTs); |
||||
|
edgeIdServiceIdCache.put(edgeId, serviceInfoProvider.getServiceId()); |
||||
|
pushRuleEngineMessage(tenantId, edge, lastConnectTs, TbMsgType.CONNECT_EVENT); |
||||
|
edgeSession.onEdgeConnect(); |
||||
|
} |
||||
|
|
||||
|
private void onEdgeDisconnect(Edge edge, UUID sessionId) { |
||||
|
TenantId tenantId = edge.getTenantId(); |
||||
|
EdgeId edgeId = edge.getId(); |
||||
|
log.info("[{}][{}] edge disconnected!", edgeId, sessionId); |
||||
|
EdgeGrpcSessionManager current = sessions.getByEdgeId(edgeId); |
||||
|
if (current != null && current.getState().getSessionId().equals(sessionId)) { |
||||
|
EdgeGrpcSessionManager toRemove = sessions.removeByEdgeId(edgeId); |
||||
|
if (toRemove != null) { |
||||
|
toRemove.onEdgeDisconnect(); |
||||
|
toRemove.destroyAndMarkAsZombieIfFailed(); |
||||
|
} |
||||
|
sessions.removeBySessionId(sessionId); |
||||
|
save(tenantId, edgeId, ACTIVITY_STATE, false); |
||||
|
long lastDisconnectTs = System.currentTimeMillis(); |
||||
|
save(tenantId, edgeId, LAST_DISCONNECT_TIME, lastDisconnectTs); |
||||
|
pushRuleEngineMessage(tenantId, edge, lastDisconnectTs, TbMsgType.DISCONNECT_EVENT); |
||||
|
} else { |
||||
|
log.info("[{}] edge session [{}] is not current anymore. Attempting to destroy it by sessionId.", edgeId, sessionId); |
||||
|
EdgeGrpcSessionManager stale = sessions.removeBySessionId(sessionId); |
||||
|
if (stale != null) { |
||||
|
try { |
||||
|
stale.destroyAndMarkAsZombieIfFailed(); |
||||
|
log.info("[{}][{}] Successfully destroyed stale session for edge [{}]", stale.getState().getTenantId(), sessionId, edgeId); |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}][{}] Failed to destroy stale session for edge [{}]", stale.getState().getTenantId(), sessionId, edgeId, e); |
||||
|
} |
||||
|
} else { |
||||
|
log.debug("[{}] No session found by sessionId [{}] to destroy", edgeId, sessionId); |
||||
|
} |
||||
|
} |
||||
|
edgeIdServiceIdCache.evict(edgeId); |
||||
|
} |
||||
|
|
||||
|
private void scheduleSyncRequestTimeout(ToEdgeSyncRequest request, UUID requestId) { |
||||
|
log.trace("[{}] scheduling sync edge request", requestId); |
||||
|
executorService.schedule(() -> { |
||||
|
log.trace("[{}] checking if sync edge request is not processed...", requestId); |
||||
|
Consumer<FromEdgeSyncResponse> consumer = localSyncEdgeRequests.remove(requestId); |
||||
|
if (consumer != null) { |
||||
|
log.trace("[{}] timeout for processing sync edge request.", requestId); |
||||
|
consumer.accept(new FromEdgeSyncResponse(requestId, request.getTenantId(), request.getEdgeId(), false, "Edge is not connected")); |
||||
|
} |
||||
|
}, 20, TimeUnit.SECONDS); |
||||
|
} |
||||
|
|
||||
|
private void onEdgeHighPriorityEvent(EdgeHighPriorityMsg msg) { |
||||
|
TenantId tenantId = msg.getTenantId(); |
||||
|
EdgeEvent edgeEvent = msg.getEdgeEvent(); |
||||
|
EdgeId edgeId = edgeEvent.getEdgeId(); |
||||
|
EdgeGrpcSessionManager session = sessions.getByEdgeId(edgeId); |
||||
|
if (session != null && session.getState().isConnected()) { |
||||
|
log.trace("[{}] onEdgeEvent [{}]", tenantId, edgeId); |
||||
|
session.addEventToHighPriorityQueue(edgeEvent); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void onEdgeEventUpdate(TenantId tenantId, EdgeId edgeId) { |
||||
|
EdgeGrpcSessionManager session = sessions.getByEdgeId(edgeId); |
||||
|
if (session != null && session.getState().isConnected()) { |
||||
|
log.trace("[{}] onEdgeEventUpdate [{}]", tenantId, edgeId.getId()); |
||||
|
session.onEdgeEventUpdate(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void startSyncProcess(TenantId tenantId, EdgeId edgeId, UUID requestId, String requestServiceId) { |
||||
|
EdgeGrpcSessionManager session = sessions.getByEdgeId(edgeId); |
||||
|
if (session != null) { |
||||
|
EdgeSessionState sessionState = session.getState(); |
||||
|
if (sessionState.isSyncInProgress()) { |
||||
|
clusterService.pushEdgeSyncResponseToCore(new FromEdgeSyncResponse(requestId, tenantId, edgeId, false, "Sync process is active at the moment"), requestServiceId); |
||||
|
} else { |
||||
|
boolean success = false; |
||||
|
if (sessionState.isConnected()) { |
||||
|
session.startSyncProcess(true); |
||||
|
success = true; |
||||
|
} |
||||
|
clusterService.pushEdgeSyncResponseToCore(new FromEdgeSyncResponse(requestId, tenantId, edgeId, success, ""), requestServiceId); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void processSyncResponse(FromEdgeSyncResponse response) { |
||||
|
log.trace("[{}] Received response from sync service: [{}]", response.getId(), response); |
||||
|
UUID requestId = response.getId(); |
||||
|
Consumer<FromEdgeSyncResponse> consumer = localSyncEdgeRequests.remove(requestId); |
||||
|
if (consumer != null) { |
||||
|
consumer.accept(response); |
||||
|
} else { |
||||
|
log.trace("[{}] Unknown or stale sync response received [{}]", requestId, response); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void pushRuleEngineMessage(TenantId tenantId, Edge edge, long ts, TbMsgType msgType) { |
||||
|
try { |
||||
|
EdgeId edgeId = edge.getId(); |
||||
|
ObjectNode edgeState = JacksonUtil.newObjectNode(); |
||||
|
boolean isConnected = TbMsgType.CONNECT_EVENT.equals(msgType); |
||||
|
if (isConnected) { |
||||
|
edgeState.put(ACTIVITY_STATE, true); |
||||
|
edgeState.put(LAST_CONNECT_TIME, ts); |
||||
|
} else { |
||||
|
edgeState.put(ACTIVITY_STATE, false); |
||||
|
edgeState.put(LAST_DISCONNECT_TIME, ts); |
||||
|
} |
||||
|
ctx.getRuleProcessor().process(EdgeConnectionTrigger.builder() |
||||
|
.tenantId(tenantId) |
||||
|
.customerId(edge.getCustomerId()) |
||||
|
.edgeId(edgeId) |
||||
|
.edgeName(edge.getName()) |
||||
|
.connected(isConnected).build()); |
||||
|
String data = JacksonUtil.toString(edgeState); |
||||
|
TbMsgMetaData md = new TbMsgMetaData(); |
||||
|
if (!persistToTelemetry) { |
||||
|
md.putValue(DataConstants.SCOPE, DataConstants.SERVER_SCOPE); |
||||
|
md.putValue("edgeName", edge.getName()); |
||||
|
md.putValue("edgeType", edge.getType()); |
||||
|
} |
||||
|
TbMsg tbMsg = TbMsg.newMsg() |
||||
|
.type(msgType) |
||||
|
.originator(edgeId) |
||||
|
.copyMetaData(md) |
||||
|
.dataType(TbMsgDataType.JSON) |
||||
|
.data(data) |
||||
|
.build(); |
||||
|
clusterService.pushMsgToRuleEngine(tenantId, edgeId, tbMsg, null); |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}][{}] Failed to push {}", tenantId, edge.getId(), msgType, e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void save(TenantId tenantId, EdgeId edgeId, String key, long value) { |
||||
|
log.debug("[{}][{}] Updating long edge telemetry [{}] [{}]", tenantId, edgeId, key, value); |
||||
|
if (persistToTelemetry) { |
||||
|
tsSubService.saveTimeseries(TimeseriesSaveRequest.builder() |
||||
|
.tenantId(tenantId) |
||||
|
.entityId(edgeId) |
||||
|
.entry(new LongDataEntry(key, value)) |
||||
|
.callback(new EdgeAttributeSaveCallback(tenantId, edgeId, key, value)) |
||||
|
.build()); |
||||
|
} else { |
||||
|
tsSubService.saveAttributes(AttributesSaveRequest.builder() |
||||
|
.tenantId(tenantId) |
||||
|
.entityId(edgeId) |
||||
|
.scope(AttributeScope.SERVER_SCOPE) |
||||
|
.entry(new LongDataEntry(key, value)) |
||||
|
.callback(new EdgeAttributeSaveCallback(tenantId, edgeId, key, value)) |
||||
|
.build()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void save(TenantId tenantId, EdgeId edgeId, String key, boolean value) { |
||||
|
log.debug("[{}][{}] Updating boolean edge telemetry [{}] [{}]", tenantId, edgeId, key, value); |
||||
|
if (persistToTelemetry) { |
||||
|
tsSubService.saveTimeseries(TimeseriesSaveRequest.builder() |
||||
|
.tenantId(tenantId) |
||||
|
.entityId(edgeId) |
||||
|
.entry(new BooleanDataEntry(key, value)) |
||||
|
.callback(new EdgeAttributeSaveCallback(tenantId, edgeId, key, value)) |
||||
|
.build()); |
||||
|
} else { |
||||
|
tsSubService.saveAttributes(AttributesSaveRequest.builder() |
||||
|
.tenantId(tenantId) |
||||
|
.entityId(edgeId) |
||||
|
.scope(AttributeScope.SERVER_SCOPE) |
||||
|
.entry(new BooleanDataEntry(key, value)) |
||||
|
.callback(new EdgeAttributeSaveCallback(tenantId, edgeId, key, value)) |
||||
|
.build()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void shutdownExecutorSafely(ExecutorService e) { |
||||
|
if (e != null && !e.isShutdown()) { |
||||
|
e.shutdown(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,131 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session; |
||||
|
|
||||
|
import jakarta.annotation.PostConstruct; |
||||
|
import jakarta.annotation.PreDestroy; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.common.util.ThingsBoardExecutors; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.manager.EdgeGrpcSessionManager; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.manager.KafkaBasedEdgeGrpcSessionManager; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.Collection; |
||||
|
import java.util.List; |
||||
|
import java.util.Queue; |
||||
|
import java.util.concurrent.ConcurrentLinkedQueue; |
||||
|
import java.util.concurrent.ScheduledExecutorService; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
import java.util.function.Function; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@ConditionalOnProperty(prefix = "edges", value = "enabled", havingValue = "true") |
||||
|
@TbCoreComponent |
||||
|
public class DefaultZombieSessionCleanupService implements ZombieSessionCleanupService { |
||||
|
|
||||
|
@Autowired |
||||
|
private EdgeSessionsHolder edgeSessionsHolder; |
||||
|
|
||||
|
private final Queue<EdgeGrpcSessionManager> zombieSessions; |
||||
|
private final ScheduledExecutorService zombieSessionsExecutorService; |
||||
|
|
||||
|
public DefaultZombieSessionCleanupService() { |
||||
|
this.zombieSessions = new ConcurrentLinkedQueue<>(); |
||||
|
this.zombieSessionsExecutorService = ThingsBoardExecutors.newSingleThreadScheduledExecutor("zombie-sessions"); |
||||
|
} |
||||
|
|
||||
|
@PostConstruct |
||||
|
public void init() { |
||||
|
this.zombieSessionsExecutorService.scheduleAtFixedRate(this::cleanupZombieSessions, 30, 60, TimeUnit.SECONDS); |
||||
|
} |
||||
|
|
||||
|
@PreDestroy |
||||
|
public void destroy() { |
||||
|
if (zombieSessionsExecutorService != null && !zombieSessionsExecutorService.isShutdown()) { |
||||
|
zombieSessionsExecutorService.shutdown(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void add(EdgeGrpcSessionManager session) { |
||||
|
zombieSessions.add(session); |
||||
|
} |
||||
|
|
||||
|
private void cleanupZombieSessions() { |
||||
|
try { |
||||
|
tryToDestroyZombieSessions(getZombieSessions(edgeSessionsHolder.getSessions().values()), |
||||
|
s -> edgeSessionsHolder.removeByEdgeId(s.getState().getEdge().getId())); |
||||
|
|
||||
|
tryToDestroyZombieSessions(getZombieSessions(edgeSessionsHolder.getSessionsById().values()), |
||||
|
s -> edgeSessionsHolder.removeBySessionId(s.getState().getSessionId())); |
||||
|
|
||||
|
zombieSessions.removeIf(zombie -> { |
||||
|
EdgeSessionState state = zombie.getState(); |
||||
|
if (zombie.destroy()) { |
||||
|
log.info("[{}][{}] Successfully cleaned up zombie session [{}] for edge [{}].", |
||||
|
state.getTenantId(), state.getEdgeId(), state.getSessionId(), state.getEdge().getName()); |
||||
|
return true; |
||||
|
} else { |
||||
|
log.warn("[{}][{}] Failed to remove zombie session [{}] for edge [{}].", |
||||
|
state.getTenantId(), state.getEdgeId(), state.getSessionId(), state.getEdge().getName()); |
||||
|
return false; |
||||
|
} |
||||
|
}); |
||||
|
} catch (Exception e) { |
||||
|
log.warn("Failed to cleanup zombie sessions", e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private List<EdgeGrpcSessionManager> getZombieSessions(Collection<EdgeGrpcSessionManager> sessions) { |
||||
|
List<EdgeGrpcSessionManager> result = new ArrayList<>(); |
||||
|
for (EdgeGrpcSessionManager session : sessions) { |
||||
|
if (isKafkaSessionAndZombie(session)) { |
||||
|
result.add(session); |
||||
|
} |
||||
|
} |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
private boolean isKafkaSessionAndZombie(EdgeGrpcSessionManager session) { |
||||
|
if (session instanceof KafkaBasedEdgeGrpcSessionManager kafkaSession) { |
||||
|
EdgeSessionState sessionState = kafkaSession.getState(); |
||||
|
log.debug("[{}] kafkaSession.isConnected() = {}, kafkaSession.getConsumer().getConsumer().isStopped() = {}", |
||||
|
sessionState.getEdgeId(), |
||||
|
sessionState.isConnected(), |
||||
|
kafkaSession.getConsumer() != null ? kafkaSession.getConsumer().getConsumer() != null ? kafkaSession.getConsumer().getConsumer().isStopped() : null : null); |
||||
|
return !sessionState.isConnected() && |
||||
|
kafkaSession.getConsumer() != null && |
||||
|
kafkaSession.getConsumer().getConsumer() != null && |
||||
|
!kafkaSession.getConsumer().getConsumer().isStopped(); |
||||
|
} |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
private void tryToDestroyZombieSessions(List<EdgeGrpcSessionManager> sessionsToRemove, Function<EdgeGrpcSessionManager, EdgeGrpcSessionManager> removeFunc) { |
||||
|
for (EdgeGrpcSessionManager toRemove : sessionsToRemove) { |
||||
|
log.info("[{}] Destroying session for edge because edge is not connected", toRemove.getState().getEdge().getId()); |
||||
|
if (toRemove.destroy()) { |
||||
|
removeFunc.apply(toRemove); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,621 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session; |
||||
|
|
||||
|
import com.datastax.oss.driver.api.core.uuid.Uuids; |
||||
|
import com.google.common.util.concurrent.FutureCallback; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import com.google.common.util.concurrent.SettableFuture; |
||||
|
import io.grpc.stub.StreamObserver; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.checkerframework.checker.nullness.qual.Nullable; |
||||
|
import org.springframework.data.util.Pair; |
||||
|
import org.thingsboard.rule.engine.api.AttributesSaveRequest; |
||||
|
import org.thingsboard.server.common.data.AttributeScope; |
||||
|
import org.thingsboard.server.common.data.DataConstants; |
||||
|
import org.thingsboard.server.common.data.EdgeUtils; |
||||
|
import org.thingsboard.server.common.data.StringUtils; |
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.BaseAttributeKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.BooleanDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.StringDataEntry; |
||||
|
import org.thingsboard.server.common.data.limit.LimitedApi; |
||||
|
import org.thingsboard.server.common.data.notification.rule.trigger.EdgeCommunicationFailureTrigger; |
||||
|
import org.thingsboard.server.common.data.page.PageData; |
||||
|
import org.thingsboard.server.common.data.page.PageLink; |
||||
|
import org.thingsboard.server.common.msg.edge.EdgeEventUpdateMsg; |
||||
|
import org.thingsboard.server.dao.edge.stats.EdgeStatsKey; |
||||
|
import org.thingsboard.server.gen.edge.v1.ConnectRequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.ConnectResponseCode; |
||||
|
import org.thingsboard.server.gen.edge.v1.ConnectResponseMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DownlinkMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.DownlinkResponseMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.EdgeConfiguration; |
||||
|
import org.thingsboard.server.gen.edge.v1.RequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RequestMsgType; |
||||
|
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.SyncCompletedMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.UplinkMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.UplinkResponseMsg; |
||||
|
import org.thingsboard.server.service.edge.EdgeContextComponent; |
||||
|
import org.thingsboard.server.service.edge.EdgeMsgConstructorUtils; |
||||
|
import org.thingsboard.server.service.edge.rpc.DownlinkMessageMapper; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSyncCursor; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeUplinkMessageDispatcher; |
||||
|
import org.thingsboard.server.service.edge.rpc.fetch.EdgeEventFetcher; |
||||
|
import org.thingsboard.server.service.edge.rpc.fetch.GeneralEdgeEventFetcher; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.manager.EdgeGrpcSessionManager; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.Collections; |
||||
|
import java.util.List; |
||||
|
import java.util.Optional; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ConcurrentLinkedQueue; |
||||
|
import java.util.concurrent.ScheduledExecutorService; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
import java.util.concurrent.locks.Lock; |
||||
|
import java.util.concurrent.locks.ReentrantLock; |
||||
|
import java.util.function.BiConsumer; |
||||
|
|
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
public class EdgeGrpcSession implements EdgeSession { |
||||
|
|
||||
|
private static final int MAX_DOWNLINK_ATTEMPTS = 3; |
||||
|
private static final String RATE_LIMIT_REACHED = "Rate limit reached"; |
||||
|
|
||||
|
private final EdgeGrpcSessionManager parentManagerRef; |
||||
|
private final EdgeContextComponent ctx; |
||||
|
private final StreamObserver<ResponseMsg> outputStream; |
||||
|
private final DownlinkMessageMapper downlinkMessageMapper; |
||||
|
private final EdgeUplinkMessageDispatcher uplinkMessageDispatcher; |
||||
|
private final BiConsumer<EdgeId, EdgeGrpcSessionManager> sessionOpenListener; |
||||
|
private final BiConsumer<Edge, UUID> sessionCloseListener; |
||||
|
private final ScheduledExecutorService sendDownlinkExecutorService; |
||||
|
private final int maxInboundMessageSize; |
||||
|
private final int maxHighPriorityQueueSizePerSession; |
||||
|
|
||||
|
private final EdgeSessionState state = new EdgeSessionState(); |
||||
|
private final Lock downlinkMsgLock = new ReentrantLock(); |
||||
|
private final ConcurrentLinkedQueue<EdgeEvent> highPriorityQueue = new ConcurrentLinkedQueue<>(); |
||||
|
|
||||
|
private int clientMaxInboundMessageSize; |
||||
|
|
||||
|
@Override |
||||
|
public EdgeSessionState getState() { |
||||
|
return state; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public StreamObserver<RequestMsg> initInputStream() { |
||||
|
return new StreamObserver<>() { |
||||
|
@Override |
||||
|
public void onNext(RequestMsg requestMsg) { |
||||
|
if (!state.isConnected() && requestMsg.getMsgType().equals(RequestMsgType.CONNECT_RPC_MESSAGE)) { |
||||
|
ConnectResponseMsg responseMsg = processConnect(requestMsg.getConnectRequestMsg()); |
||||
|
outputStream.onNext(ResponseMsg.newBuilder() |
||||
|
.setConnectResponseMsg(responseMsg) |
||||
|
.build()); |
||||
|
if (ConnectResponseCode.ACCEPTED != responseMsg.getResponseCode()) { |
||||
|
outputStream.onError(new RuntimeException(responseMsg.getErrorMsg())); |
||||
|
} else { |
||||
|
if (requestMsg.getConnectRequestMsg().hasMaxInboundMessageSize()) { |
||||
|
log.debug("[{}][{}] Client max inbound message size: {}", |
||||
|
getTenantId(), getSessionId(), requestMsg.getConnectRequestMsg().getMaxInboundMessageSize()); |
||||
|
clientMaxInboundMessageSize = requestMsg.getConnectRequestMsg().getMaxInboundMessageSize(); |
||||
|
} |
||||
|
state.setConnected(true); |
||||
|
} |
||||
|
} |
||||
|
if (state.isConnected()) { |
||||
|
if (requestMsg.getMsgType().equals(RequestMsgType.SYNC_REQUEST_RPC_MESSAGE)) { |
||||
|
if (requestMsg.hasSyncRequestMsg()) { |
||||
|
boolean fullSync = false; |
||||
|
if (requestMsg.getSyncRequestMsg().hasFullSync()) { |
||||
|
fullSync = requestMsg.getSyncRequestMsg().getFullSync(); |
||||
|
} |
||||
|
startSyncProcess(fullSync); |
||||
|
} else { |
||||
|
state.finishSync(); |
||||
|
saveSyncInProgressAsAttribute(false); |
||||
|
} |
||||
|
} |
||||
|
if (requestMsg.getMsgType().equals(RequestMsgType.UPLINK_RPC_MESSAGE)) { |
||||
|
if (requestMsg.hasUplinkMsg()) { |
||||
|
onUplinkMsg(requestMsg.getUplinkMsg()); |
||||
|
} |
||||
|
if (requestMsg.hasDownlinkResponseMsg()) { |
||||
|
onDownlinkResponse(requestMsg.getDownlinkResponseMsg()); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onError(Throwable t) { |
||||
|
log.trace("[{}][{}] Stream was terminated due to error:", getTenantId(), getSessionId(), t); |
||||
|
closeSession(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onCompleted() { |
||||
|
log.info("[{}][{}] Stream was closed and completed successfully!", getTenantId(), getSessionId()); |
||||
|
closeSession(); |
||||
|
} |
||||
|
|
||||
|
private void closeSession() { |
||||
|
state.setConnected(false); |
||||
|
if (state.getEdge() != null) { |
||||
|
try { |
||||
|
sessionCloseListener.accept(state.getEdge(), getSessionId()); |
||||
|
} catch (Exception ignored) { |
||||
|
} |
||||
|
} |
||||
|
try { |
||||
|
outputStream.onCompleted(); |
||||
|
} catch (Exception ignored) { |
||||
|
} |
||||
|
} |
||||
|
}; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void startSyncProcess(boolean fullSync) { |
||||
|
if (state.tryStartSync()) { |
||||
|
saveSyncInProgressAsAttribute(true); |
||||
|
log.info("[{}][{}][{}] Staring edge sync process", getTenantId(), getEdgeId(), getSessionId()); |
||||
|
interruptGeneralProcessingOnSync(); |
||||
|
doSync(new EdgeSyncCursor(ctx, state.getEdge(), fullSync)); |
||||
|
} else { |
||||
|
log.info("[{}][{}][{}] Sync is already started, skipping starting it now", getTenantId(), getEdgeId(), getSessionId()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void sendDownlinkMsg(ResponseMsg responseMsg) { |
||||
|
if (state.isConnected()) { |
||||
|
String responseMsgStr = StringUtils.truncate(responseMsg.toString(), 10000); |
||||
|
log.trace("[{}][{}] Sending downlink msg [{}]", getTenantId(), getEdgeId(), responseMsgStr); |
||||
|
downlinkMsgLock.lock(); |
||||
|
String downlinkMsgStr = responseMsg.hasDownlinkMsg() ? String.valueOf(responseMsg.getDownlinkMsg().getDownlinkMsgId()) : responseMsgStr; |
||||
|
try { |
||||
|
outputStream.onNext(responseMsg); |
||||
|
} catch (Exception e) { |
||||
|
log.trace("[{}][{}] Failed to send downlink message [{}]", getTenantId(), getEdgeId(), downlinkMsgStr, e); |
||||
|
state.setConnected(false); |
||||
|
sessionCloseListener.accept(state.getEdge(), getSessionId()); |
||||
|
} finally { |
||||
|
downlinkMsgLock.unlock(); |
||||
|
} |
||||
|
log.trace("[{}][{}] downlink msg successfully sent [{}]", getTenantId(), getEdgeId(), downlinkMsgStr); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void addHighPriorityEvent(EdgeEvent edgeEvent) { |
||||
|
while (highPriorityQueue.size() > maxHighPriorityQueueSizePerSession) { |
||||
|
EdgeEvent oldestHighPriority = highPriorityQueue.poll(); |
||||
|
if (oldestHighPriority != null) { |
||||
|
log.warn("[{}][{}][{}] High priority queue is full. Removing oldest high priority event from queue {}", |
||||
|
state.getTenantId(), state.getEdgeId(), getSessionId(), oldestHighPriority); |
||||
|
} |
||||
|
} |
||||
|
highPriorityQueue.add(edgeEvent); |
||||
|
ctx.getStatsCounterService().ifPresent(statsCounterService -> |
||||
|
statsCounterService.recordEvent(EdgeStatsKey.DOWNLINK_MSGS_ADDED, state.getEdge().getTenantId(), edgeEvent.getEdgeId(), 1)); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void processHighPriorityEvents() { |
||||
|
try { |
||||
|
if (state.isConnected() && !state.isSyncInProgress()) { |
||||
|
if (highPriorityQueue.isEmpty()) { |
||||
|
return; |
||||
|
} |
||||
|
List<EdgeEvent> highPriorityEvents = new ArrayList<>(); |
||||
|
EdgeEvent event; |
||||
|
while ((event = highPriorityQueue.poll()) != null) { |
||||
|
highPriorityEvents.add(event); |
||||
|
} |
||||
|
log.trace("[{}][{}] Sending high priority events {}", getTenantId(), getEdgeId(), highPriorityEvents.size()); |
||||
|
List<DownlinkMsg> downlinkMsgsPack = downlinkMessageMapper.convertToDownlinkMsgsPack(state, highPriorityEvents); |
||||
|
sendDownlinkMsgsPack(downlinkMsgsPack).get(); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("[{}] Failed to process high priority events", getEdgeId(), e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean hasHighPriorityEvents() { |
||||
|
return !highPriorityQueue.isEmpty(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Pair<Long, Long>> fetchAndSendEdgeEvents(EdgeEventFetcher fetcher) { |
||||
|
SettableFuture<Pair<Long, Long>> result = SettableFuture.create(); |
||||
|
PageLink pageLink = fetcher.getPageLink(ctx.getEdgeEventStorageSettings().getMaxReadRecordsCount()); |
||||
|
fetchAndSendEdgeEvents(fetcher, pageLink, result); |
||||
|
return result; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public ListenableFuture<Boolean> sendDownlinkMsgsPack(List<DownlinkMsg> downlinkMsgsPack) { |
||||
|
interruptPreviousSendDownlinkMsgsTask(); |
||||
|
|
||||
|
state.setSendDownlinkMsgsFuture(SettableFuture.create()); |
||||
|
state.getPendingMsgsMap().clear(); |
||||
|
|
||||
|
downlinkMsgsPack.forEach(msg -> state.getPendingMsgsMap().put(msg.getDownlinkMsgId(), msg)); |
||||
|
scheduleDownlinkMsgsPackSend(1); |
||||
|
|
||||
|
return state.getSendDownlinkMsgsFuture(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void close() { |
||||
|
log.debug("[{}][{}] Closing session", getTenantId(), getSessionId()); |
||||
|
state.setConnected(false); |
||||
|
try { |
||||
|
outputStream.onCompleted(); |
||||
|
} catch (Exception e) { |
||||
|
log.debug("[{}][{}] Failed to close output stream: {}", getTenantId(), getSessionId(), e.getMessage()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void onUplinkMsg(UplinkMsg uplinkMsg) { |
||||
|
if (isRateLimitViolated(uplinkMsg)) { |
||||
|
return; |
||||
|
} |
||||
|
Futures.addCallback(uplinkMessageDispatcher.processUplinkMsg(state, uplinkMsg), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(@Nullable List<Void> result) { |
||||
|
sendResponseMessage(uplinkMsg.getUplinkMsgId(), true, null); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
String errorMsg = EdgeUtils.createErrorMsgFromRootCauseAndStackTrace(t); |
||||
|
sendResponseMessage(uplinkMsg.getUplinkMsgId(), false, errorMsg); |
||||
|
} |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} |
||||
|
|
||||
|
private boolean isRateLimitViolated(UplinkMsg uplinkMsg) { |
||||
|
if (!ctx.getRateLimitService().checkRateLimit(LimitedApi.EDGE_UPLINK_MESSAGES, getTenantId()) || |
||||
|
!ctx.getRateLimitService().checkRateLimit(LimitedApi.EDGE_UPLINK_MESSAGES_PER_EDGE, getTenantId(), getEdgeId())) { |
||||
|
String errorMsg = String.format("Failed to process uplink message. %s", RATE_LIMIT_REACHED); |
||||
|
sendResponseMessage(uplinkMsg.getUplinkMsgId(), false, errorMsg); |
||||
|
return true; |
||||
|
} |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
private void sendResponseMessage(int uplinkMsgId, boolean success, String errorMsg) { |
||||
|
UplinkResponseMsg.Builder responseBuilder = UplinkResponseMsg.newBuilder() |
||||
|
.setUplinkMsgId(uplinkMsgId) |
||||
|
.setSuccess(success); |
||||
|
if (errorMsg != null) { |
||||
|
responseBuilder.setErrorMsg(errorMsg); |
||||
|
} |
||||
|
sendDownlinkMsg(ResponseMsg.newBuilder() |
||||
|
.setUplinkResponseMsg(responseBuilder.build()) |
||||
|
.build()); |
||||
|
} |
||||
|
|
||||
|
private void onDownlinkResponse(DownlinkResponseMsg msg) { |
||||
|
try { |
||||
|
if (msg.getSuccess()) { |
||||
|
state.getPendingMsgsMap().remove(msg.getDownlinkMsgId()); |
||||
|
ctx.getStatsCounterService().ifPresent(statsCounterService -> statsCounterService.recordEvent(EdgeStatsKey.DOWNLINK_MSGS_PUSHED, getTenantId(), getEdgeId(), 1)); |
||||
|
log.debug("[{}][{}][{}] Msg has been processed successfully! Msg Id: [{}], Msg: {}", getTenantId(), getEdgeId(), getSessionId(), msg.getDownlinkMsgId(), msg); |
||||
|
} else { |
||||
|
log.debug("[{}][{}][{}] Msg processing failed! Msg Id: [{}], Error msg: {}", getTenantId(), getEdgeId(), getSessionId(), msg.getDownlinkMsgId(), msg.getErrorMsg()); |
||||
|
DownlinkMsg downlinkMsg = state.getPendingMsgsMap().get(msg.getDownlinkMsgId()); |
||||
|
// if NOT timeseries or attributes failures - ack failed downlink
|
||||
|
if (downlinkMsg != null && downlinkMsg.getEntityDataCount() == 0) { |
||||
|
state.getPendingMsgsMap().remove(msg.getDownlinkMsgId()); |
||||
|
} |
||||
|
} |
||||
|
if (state.getPendingMsgsMap().isEmpty()) { |
||||
|
log.debug("[{}][{}][{}] Pending msgs map is empty. Stopping current iteration", getTenantId(), getEdgeId(), getSessionId()); |
||||
|
stopCurrentSendDownlinkMsgsTask(false); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("[{}][{}] Can't process downlink response message [{}]", getTenantId(), getEdgeId(), msg, e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void interruptGeneralProcessingOnSync() { |
||||
|
log.debug("[{}][{}][{}] Sync process started. General processing interrupted!", getTenantId(), getEdgeId(), getSessionId()); |
||||
|
stopCurrentSendDownlinkMsgsTask(true); |
||||
|
} |
||||
|
|
||||
|
private void doSync(EdgeSyncCursor cursor) { |
||||
|
if (cursor.hasNext()) { |
||||
|
EdgeEventFetcher next = cursor.getNext(); |
||||
|
log.debug("[{}][{}] starting sync process, cursor current idx = {}, class = {}", |
||||
|
getTenantId(), getEdgeId(), cursor.getCurrentIdx(), next.getClass().getSimpleName()); |
||||
|
ListenableFuture<Pair<Long, Long>> future = fetchAndSendEdgeEvents(next); |
||||
|
Futures.addCallback(future, new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(@Nullable Pair<Long, Long> result) { |
||||
|
doSync(cursor); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.error("[{}][{}] Exception during sync process", getTenantId(), getEdgeId(), t); |
||||
|
} |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} else { |
||||
|
log.info("[{}][{}] sync process completed", getTenantId(), getEdgeId()); |
||||
|
DownlinkMsg syncCompleteDownlinkMsg = DownlinkMsg.newBuilder() |
||||
|
.setDownlinkMsgId(EdgeUtils.nextPositiveInt()) |
||||
|
.setSyncCompletedMsg(SyncCompletedMsg.newBuilder().build()) |
||||
|
.build(); |
||||
|
Futures.addCallback(sendDownlinkMsgsPack(Collections.singletonList(syncCompleteDownlinkMsg)), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(Boolean isInterrupted) { |
||||
|
markSyncCompletedSendEdgeEventUpdate(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.error("[{}][{}] Exception during sending sync complete", getTenantId(), getEdgeId(), t); |
||||
|
markSyncCompletedSendEdgeEventUpdate(); |
||||
|
} |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void markSyncCompletedSendEdgeEventUpdate() { |
||||
|
state.finishSync(); |
||||
|
saveSyncInProgressAsAttribute(false); |
||||
|
ctx.getClusterService().onEdgeEventUpdate(new EdgeEventUpdateMsg(getTenantId(), getEdgeId())); |
||||
|
} |
||||
|
|
||||
|
private void fetchAndSendEdgeEvents(EdgeEventFetcher fetcher, PageLink pageLink, SettableFuture<Pair<Long, Long>> result) { |
||||
|
Edge edge = state.getEdge(); |
||||
|
TenantId tenantId = getTenantId(); |
||||
|
try { |
||||
|
log.trace("[{}] Start processing edge events, fetcher = {}, pageLink = {}", edge.getId(), fetcher.getClass().getSimpleName(), pageLink); |
||||
|
processHighPriorityEvents(); |
||||
|
PageData<EdgeEvent> pageData = fetcher.fetchEdgeEvents(edge.getTenantId(), edge, pageLink); |
||||
|
if (state.isConnected() && !pageData.getData().isEmpty()) { |
||||
|
if (fetcher instanceof GeneralEdgeEventFetcher) { |
||||
|
long queueSize = pageData.getTotalElements() - ((long) pageLink.getPageSize() * pageLink.getPage()); |
||||
|
ctx.getStatsCounterService().ifPresent(statsCounterService -> |
||||
|
statsCounterService.recordEvent(EdgeStatsKey.DOWNLINK_MSGS_LAG, tenantId, edge.getId(), queueSize)); |
||||
|
} |
||||
|
log.trace("[{}][{}][{}] event(s) are going to be processed.", tenantId, edge.getId(), pageData.getData().size()); |
||||
|
List<DownlinkMsg> downlinkMsgsPack = downlinkMessageMapper.convertToDownlinkMsgsPack(state, pageData.getData()); |
||||
|
Futures.addCallback(sendDownlinkMsgsPack(downlinkMsgsPack), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(@Nullable Boolean isInterrupted) { |
||||
|
if (Boolean.TRUE.equals(isInterrupted)) { |
||||
|
log.debug("[{}][{}][{}] Send downlink messages task was interrupted", tenantId, edge.getId(), getSessionId()); |
||||
|
result.set(null); |
||||
|
} else { |
||||
|
if (state.isConnected() && pageData.hasNext()) { |
||||
|
fetchAndSendEdgeEvents(fetcher, pageLink.nextPageLink(), result); |
||||
|
} else { |
||||
|
EdgeEvent latestEdgeEvent = pageData.getData().get(pageData.getData().size() - 1); |
||||
|
UUID idOffset = latestEdgeEvent.getUuidId(); |
||||
|
if (idOffset != null) { |
||||
|
Long newStartTs = Uuids.unixTimestamp(idOffset); |
||||
|
long newStartSeqId = latestEdgeEvent.getSeqId(); |
||||
|
result.set(Pair.of(newStartTs, newStartSeqId)); |
||||
|
} else { |
||||
|
result.set(null); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.error("[{}] Failed to send downlink msgs pack", edge.getId(), t); |
||||
|
result.setException(t); |
||||
|
} |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} else { |
||||
|
log.trace("[{}] no event(s) found. Stop processing edge events, fetcher = {}, pageLink = {}", edge.getId(), fetcher.getClass().getSimpleName(), pageLink); |
||||
|
result.set(null); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("[{}] Failed to fetch edge events", edge.getId(), e); |
||||
|
result.setException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void scheduleDownlinkMsgsPackSend(int attempt) { |
||||
|
Runnable sendDownlinkMsgsTask = () -> { |
||||
|
try { |
||||
|
if (!state.isConnected()) { |
||||
|
stopCurrentSendDownlinkMsgsTask(true); |
||||
|
return; |
||||
|
} |
||||
|
if (!state.getPendingMsgsMap().values().isEmpty()) { |
||||
|
Edge edge = state.getEdge(); |
||||
|
List<DownlinkMsg> copy = new ArrayList<>(state.getPendingMsgsMap().values()); |
||||
|
if (attempt > 1) { |
||||
|
String error = "Failed to deliver the batch"; |
||||
|
String failureMsg = String.format("{%s} (size: {%s})", error, copy.size()); |
||||
|
if (attempt == 2) { |
||||
|
// Send a failure notification only on the second attempt.
|
||||
|
// This ensures that failure alerts are sent just once to avoid redundant notifications.
|
||||
|
ctx.getRuleProcessor().process(EdgeCommunicationFailureTrigger.builder().tenantId(getTenantId()) |
||||
|
.edgeId(getEdgeId()).customerId(edge.getCustomerId()).edgeName(edge.getName()).failureMsg(failureMsg).error(error).build()); |
||||
|
} |
||||
|
ctx.getStatsCounterService().ifPresent(statsCounterService -> |
||||
|
statsCounterService.recordEvent(EdgeStatsKey.DOWNLINK_MSGS_TMP_FAILED, edge.getTenantId(), getEdgeId(), 1)); |
||||
|
log.warn("[{}][{}] {} on attempt {}", getTenantId(), getEdgeId(), failureMsg, attempt); |
||||
|
log.debug("[{}][{}] entities in failed batch: {}", getTenantId(), getEdgeId(), copy); |
||||
|
} |
||||
|
log.trace("[{}][{}][{}] downlink msg(s) are going to be send.", getTenantId(), getEdgeId(), copy.size()); |
||||
|
for (DownlinkMsg downlinkMsg : copy) { |
||||
|
if (clientMaxInboundMessageSize != 0 && downlinkMsg.getSerializedSize() > clientMaxInboundMessageSize) { |
||||
|
String error = String.format("Client max inbound message size %s is exceeded. Please increase value of CLOUD_RPC_MAX_INBOUND_MESSAGE_SIZE " + |
||||
|
"env variable on the edge and restart it.", clientMaxInboundMessageSize); |
||||
|
String message = String.format("Downlink msg size %s exceeds client max inbound message size %s. " + |
||||
|
"Please increase value of CLOUD_RPC_MAX_INBOUND_MESSAGE_SIZE env variable on the edge and restart it.", downlinkMsg.getSerializedSize(), clientMaxInboundMessageSize); |
||||
|
log.error("[{}][{}][{}] {} Message {}", getTenantId(), getEdgeId(), getSessionId(), message, downlinkMsg); |
||||
|
ctx.getRuleProcessor().process(EdgeCommunicationFailureTrigger.builder().tenantId(getTenantId()) |
||||
|
.edgeId(getEdgeId()).customerId(edge.getCustomerId()).edgeName(edge.getName()).failureMsg(message).error(error).build()); |
||||
|
ctx.getStatsCounterService().ifPresent(statsCounterService -> |
||||
|
statsCounterService.recordEvent(EdgeStatsKey.DOWNLINK_MSGS_PERMANENTLY_FAILED, edge.getTenantId(), getEdgeId(), 1)); |
||||
|
state.getPendingMsgsMap().remove(downlinkMsg.getDownlinkMsgId()); |
||||
|
} else { |
||||
|
sendDownlinkMsg(ResponseMsg.newBuilder() |
||||
|
.setDownlinkMsg(downlinkMsg) |
||||
|
.build()); |
||||
|
} |
||||
|
} |
||||
|
if (attempt < MAX_DOWNLINK_ATTEMPTS) { |
||||
|
scheduleDownlinkMsgsPackSend(attempt + 1); |
||||
|
} else { |
||||
|
String failureMsg = String.format("Failed to deliver messages: %s", copy); |
||||
|
log.warn("[{}][{}] Failed to deliver the batch after {} attempts. Next messages are going to be discarded {}", |
||||
|
getTenantId(), getEdgeId(), MAX_DOWNLINK_ATTEMPTS, copy); |
||||
|
ctx.getRuleProcessor().process(EdgeCommunicationFailureTrigger.builder().tenantId(getTenantId()).edgeId(getEdgeId()) |
||||
|
.customerId(edge.getCustomerId()).edgeName(edge.getName()).failureMsg(failureMsg) |
||||
|
.error("Failed to deliver messages after " + MAX_DOWNLINK_ATTEMPTS + " attempts").build()); |
||||
|
ctx.getStatsCounterService().ifPresent(statsCounterService -> |
||||
|
statsCounterService.recordEvent(EdgeStatsKey.DOWNLINK_MSGS_PERMANENTLY_FAILED, edge.getTenantId(), getEdgeId(), copy.size())); |
||||
|
stopCurrentSendDownlinkMsgsTask(false); |
||||
|
} |
||||
|
} else { |
||||
|
stopCurrentSendDownlinkMsgsTask(false); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}][{}] Failed to send downlink msgs. Error msg {}", getTenantId(), getEdgeId(), e.getMessage(), e); |
||||
|
stopCurrentSendDownlinkMsgsTask(true); |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
if (attempt == 1) { |
||||
|
sendDownlinkExecutorService.submit(sendDownlinkMsgsTask); |
||||
|
} else { |
||||
|
state.setScheduledSendDownlinkTask( |
||||
|
sendDownlinkExecutorService.schedule( |
||||
|
sendDownlinkMsgsTask, |
||||
|
ctx.getEdgeEventStorageSettings().getSleepIntervalBetweenBatches(), |
||||
|
TimeUnit.MILLISECONDS) |
||||
|
); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void interruptPreviousSendDownlinkMsgsTask() { |
||||
|
if (state.getSendDownlinkMsgsFuture() != null && !state.getSendDownlinkMsgsFuture().isDone() |
||||
|
|| state.getScheduledSendDownlinkTask() != null && !state.getScheduledSendDownlinkTask().isCancelled()) { |
||||
|
log.debug("[{}][{}][{}] Previous send downlink future was not properly completed, stopping it now!", |
||||
|
getTenantId(), getEdgeId(), getSessionId()); |
||||
|
stopCurrentSendDownlinkMsgsTask(true); |
||||
|
} else { |
||||
|
log.trace("[{}][{}][{}] Previous send downlink future is not active", |
||||
|
getTenantId(), getEdgeId(), getSessionId()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void stopCurrentSendDownlinkMsgsTask(Boolean isInterrupted) { |
||||
|
if (state.getSendDownlinkMsgsFuture() != null && !state.getSendDownlinkMsgsFuture().isDone()) { |
||||
|
state.getSendDownlinkMsgsFuture().set(isInterrupted); |
||||
|
} |
||||
|
if (state.getScheduledSendDownlinkTask() != null) { |
||||
|
state.getScheduledSendDownlinkTask().cancel(true); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private ConnectResponseMsg processConnect(ConnectRequestMsg request) { |
||||
|
log.trace("[{}] processConnect [{}]", getSessionId(), request); |
||||
|
Optional<Edge> optional = ctx.getEdgeService().findEdgeByRoutingKey(TenantId.SYS_TENANT_ID, request.getEdgeRoutingKey()); |
||||
|
if (optional.isPresent()) { |
||||
|
Edge edge = optional.get(); |
||||
|
TenantId tenantId = edge.getTenantId(); |
||||
|
state.setEdge(edge); |
||||
|
try { |
||||
|
if (edge.getSecret().equals(request.getEdgeSecret())) { |
||||
|
sessionOpenListener.accept(edge.getId(), parentManagerRef); |
||||
|
state.setEdgeVersion(request.getEdgeVersion()); |
||||
|
processSaveEdgeVersionAsAttribute(request.getEdgeVersion().name()); |
||||
|
return ConnectResponseMsg.newBuilder() |
||||
|
.setResponseCode(ConnectResponseCode.ACCEPTED) |
||||
|
.setErrorMsg("") |
||||
|
.setConfiguration(EdgeMsgConstructorUtils.constructEdgeConfiguration(edge)) |
||||
|
.setMaxInboundMessageSize(maxInboundMessageSize) |
||||
|
.build(); |
||||
|
} |
||||
|
String error = "Failed to validate the edge!"; |
||||
|
String failureMsg = String.format("%s Provided request secret: %s", error, request.getEdgeSecret()); |
||||
|
ctx.getRuleProcessor().process(EdgeCommunicationFailureTrigger.builder().tenantId(tenantId).edgeId(edge.getId()) |
||||
|
.customerId(edge.getCustomerId()).edgeName(edge.getName()).failureMsg(failureMsg).error(error).build()); |
||||
|
return ConnectResponseMsg.newBuilder() |
||||
|
.setResponseCode(ConnectResponseCode.BAD_CREDENTIALS) |
||||
|
.setErrorMsg(failureMsg) |
||||
|
.setConfiguration(EdgeConfiguration.getDefaultInstance()).build(); |
||||
|
} catch (Exception e) { |
||||
|
String failureMsg = "Failed to process edge connection!"; |
||||
|
ctx.getRuleProcessor().process(EdgeCommunicationFailureTrigger.builder().tenantId(tenantId).edgeId(edge.getId()) |
||||
|
.customerId(edge.getCustomerId()).edgeName(edge.getName()).failureMsg(failureMsg).error(e.getMessage()).build()); |
||||
|
log.error(failureMsg, e); |
||||
|
return ConnectResponseMsg.newBuilder() |
||||
|
.setResponseCode(ConnectResponseCode.SERVER_UNAVAILABLE) |
||||
|
.setErrorMsg(failureMsg) |
||||
|
.setConfiguration(EdgeConfiguration.getDefaultInstance()).build(); |
||||
|
} |
||||
|
} |
||||
|
return ConnectResponseMsg.newBuilder() |
||||
|
.setResponseCode(ConnectResponseCode.BAD_CREDENTIALS) |
||||
|
.setErrorMsg("Failed to find the edge! Routing key: " + request.getEdgeRoutingKey()) |
||||
|
.setConfiguration(EdgeConfiguration.getDefaultInstance()).build(); |
||||
|
} |
||||
|
|
||||
|
private void processSaveEdgeVersionAsAttribute(String edgeVersion) { |
||||
|
AttributeKvEntry attributeKvEntry = new BaseAttributeKvEntry(new StringDataEntry(DataConstants.EDGE_VERSION_ATTR_KEY, edgeVersion), System.currentTimeMillis()); |
||||
|
ctx.getAttributesService().save(getTenantId(), getEdgeId(), AttributeScope.SERVER_SCOPE, attributeKvEntry); |
||||
|
} |
||||
|
|
||||
|
private void saveSyncInProgressAsAttribute(Boolean value) { |
||||
|
ctx.getTsSubService().saveAttributes(AttributesSaveRequest.builder() |
||||
|
.tenantId(getTenantId()) |
||||
|
.entityId(getEdgeId()) |
||||
|
.scope(AttributeScope.SERVER_SCOPE) |
||||
|
.entry(new BooleanDataEntry(DataConstants.EDGE_SYNC_IN_PROGRESS_ATTR_KEY, value)) |
||||
|
.callback(new EdgeAttributeSaveCallback(getTenantId(), getEdgeId(), DataConstants.EDGE_SYNC_IN_PROGRESS_ATTR_KEY, value)) |
||||
|
.build()); |
||||
|
} |
||||
|
|
||||
|
private TenantId getTenantId() { |
||||
|
return state.getTenantId(); |
||||
|
} |
||||
|
|
||||
|
private EdgeId getEdgeId() { |
||||
|
return state.getEdgeId(); |
||||
|
} |
||||
|
|
||||
|
private UUID getSessionId() { |
||||
|
return state.getSessionId(); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,34 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session; |
||||
|
|
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.manager.EdgeGrpcSessionManager; |
||||
|
|
||||
|
public abstract class EdgeGrpcSessionDelegate implements EdgeGrpcSessionManager { |
||||
|
|
||||
|
protected abstract EdgeSession getSession(); |
||||
|
|
||||
|
@Override |
||||
|
public void addEventToHighPriorityQueue(EdgeEvent edgeEvent) { |
||||
|
getSession().addHighPriorityEvent(edgeEvent); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void startSyncProcess(boolean fullSync) { |
||||
|
getSession().startSyncProcess(fullSync); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,43 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session; |
||||
|
|
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import io.grpc.stub.StreamObserver; |
||||
|
import org.springframework.data.util.Pair; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.gen.edge.v1.DownlinkMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.fetch.EdgeEventFetcher; |
||||
|
|
||||
|
import java.io.Closeable; |
||||
|
import java.util.List; |
||||
|
|
||||
|
public interface EdgeSession extends Closeable { |
||||
|
|
||||
|
StreamObserver<RequestMsg> initInputStream(); |
||||
|
EdgeSessionState getState(); |
||||
|
void startSyncProcess(boolean fullSync); |
||||
|
void sendDownlinkMsg(ResponseMsg responseMsg); |
||||
|
void addHighPriorityEvent(EdgeEvent edgeEvent); |
||||
|
void processHighPriorityEvents(); |
||||
|
boolean hasHighPriorityEvents(); |
||||
|
ListenableFuture<Pair<Long, Long>> fetchAndSendEdgeEvents(EdgeEventFetcher fetcher); |
||||
|
ListenableFuture<Boolean> sendDownlinkMsgsPack(List<DownlinkMsg> downlinkMsgsPack); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,83 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session; |
||||
|
|
||||
|
import lombok.Data; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.queue.util.TbCoreComponent; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.manager.EdgeGrpcSessionManager; |
||||
|
|
||||
|
import java.util.HashSet; |
||||
|
import java.util.Set; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||
|
import java.util.concurrent.ConcurrentMap; |
||||
|
import java.util.function.Consumer; |
||||
|
|
||||
|
@Data |
||||
|
@Slf4j |
||||
|
@Component |
||||
|
@ConditionalOnProperty(prefix = "edges", value = "enabled", havingValue = "true") |
||||
|
@TbCoreComponent |
||||
|
public class EdgeSessionsHolder { |
||||
|
|
||||
|
private final ConcurrentMap<EdgeId, EdgeGrpcSessionManager> sessions = new ConcurrentHashMap<>(); |
||||
|
private final ConcurrentMap<UUID, EdgeGrpcSessionManager> sessionsById = new ConcurrentHashMap<>(); |
||||
|
|
||||
|
public void forEach(Consumer<EdgeGrpcSessionManager> consumer) { |
||||
|
Set<EdgeGrpcSessionManager> unique = new HashSet<>(sessions.values()); |
||||
|
unique.addAll(sessionsById.values()); |
||||
|
|
||||
|
unique.forEach(consumer); |
||||
|
} |
||||
|
|
||||
|
public void put(EdgeGrpcSessionManager session) { |
||||
|
UUID sessionId = session.getState().getSessionId(); |
||||
|
EdgeId edgeId = session.getState().getEdgeId(); |
||||
|
sessionsById.put(sessionId, session); |
||||
|
sessions.put(edgeId, session); |
||||
|
} |
||||
|
|
||||
|
public EdgeGrpcSessionManager getByEdgeId(EdgeId id) { |
||||
|
return sessions.get(id); |
||||
|
} |
||||
|
|
||||
|
public boolean hasByEdgeId(EdgeId id) { |
||||
|
return sessions.containsKey(id); |
||||
|
} |
||||
|
|
||||
|
public EdgeGrpcSessionManager removeByEdgeId(EdgeId id) { |
||||
|
return sessions.remove(id); |
||||
|
} |
||||
|
|
||||
|
public EdgeGrpcSessionManager removeBySessionId(UUID sessionId) { |
||||
|
return sessionsById.remove(sessionId); |
||||
|
} |
||||
|
|
||||
|
public void remove(EdgeGrpcSessionManager session) { |
||||
|
if (session == null) { |
||||
|
log.warn("Can't remove session from holder because it's null"); |
||||
|
return; |
||||
|
} |
||||
|
EdgeSessionState sessionState = session.getState(); |
||||
|
removeByEdgeId(sessionState.getEdgeId()); |
||||
|
removeBySessionId(sessionState.getSessionId()); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,23 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session; |
||||
|
|
||||
|
import org.thingsboard.server.service.edge.rpc.session.manager.EdgeGrpcSessionManager; |
||||
|
|
||||
|
public interface ZombieSessionCleanupService { |
||||
|
|
||||
|
void add(EdgeGrpcSessionManager session); |
||||
|
} |
||||
@ -0,0 +1,149 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session.manager; |
||||
|
|
||||
|
import io.grpc.stub.StreamObserver; |
||||
|
import lombok.Getter; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.context.annotation.Lazy; |
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.id.CustomerId; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.gen.edge.v1.EdgeUpdateMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.RequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
||||
|
import org.thingsboard.server.service.edge.EdgeContextComponent; |
||||
|
import org.thingsboard.server.service.edge.EdgeMsgConstructorUtils; |
||||
|
import org.thingsboard.server.service.edge.rpc.DownlinkMessageMapper; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeUplinkMessageDispatcher; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeGrpcSession; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeGrpcSessionDelegate; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeSession; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.ZombieSessionCleanupService; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ScheduledExecutorService; |
||||
|
import java.util.concurrent.locks.Lock; |
||||
|
import java.util.concurrent.locks.ReentrantLock; |
||||
|
import java.util.function.BiConsumer; |
||||
|
|
||||
|
@Slf4j |
||||
|
public abstract class AbstractEdgeGrpcSessionManager extends EdgeGrpcSessionDelegate implements EdgeGrpcSessionManager { |
||||
|
|
||||
|
@Value("${edges.max_high_priority_queue_size_per_session:10000}") |
||||
|
protected int maxHighPriorityQueueSizePerSession; |
||||
|
|
||||
|
@Value("${edges.rpc.max_inbound_message_size:4194304}") |
||||
|
protected int maxInboundMessageSize; |
||||
|
|
||||
|
@Autowired |
||||
|
protected ZombieSessionCleanupService zombieSessionCleanupService; |
||||
|
|
||||
|
@Autowired |
||||
|
@Lazy |
||||
|
protected EdgeContextComponent ctx; |
||||
|
|
||||
|
@Autowired |
||||
|
protected DownlinkMessageMapper downlinkMessageMapper; |
||||
|
|
||||
|
@Autowired |
||||
|
protected EdgeUplinkMessageDispatcher uplinkMessageDispatcher; |
||||
|
|
||||
|
@Getter |
||||
|
protected EdgeSession session; |
||||
|
|
||||
|
private final Lock initLock = new ReentrantLock(); |
||||
|
|
||||
|
@Override |
||||
|
public StreamObserver<RequestMsg> initInputStream(StreamObserver<ResponseMsg> outputStream, |
||||
|
BiConsumer<EdgeId, EdgeGrpcSessionManager> sessionOpenListener, |
||||
|
BiConsumer<Edge, UUID> sessionCloseListener, |
||||
|
ScheduledExecutorService sendDownlinkExecutorService) { |
||||
|
if (session != null) { |
||||
|
throw new IllegalStateException("Session already initialized"); |
||||
|
} |
||||
|
initLock.lock(); |
||||
|
try { |
||||
|
this.session = new EdgeGrpcSession( |
||||
|
this, ctx, outputStream, downlinkMessageMapper, uplinkMessageDispatcher, |
||||
|
sessionOpenListener, sessionCloseListener, |
||||
|
sendDownlinkExecutorService, maxInboundMessageSize, maxHighPriorityQueueSizePerSession); |
||||
|
return session.initInputStream(); |
||||
|
} finally { |
||||
|
initLock.unlock(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public EdgeSessionState getState() { |
||||
|
if (session == null) { |
||||
|
return null; |
||||
|
} |
||||
|
return session.getState(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onConfigurationUpdate(Edge edge) { |
||||
|
EdgeSessionState state = getState(); |
||||
|
if (state == null) { |
||||
|
log.warn("Session state is null, skipping configuration update for edge [{}]", edge.getId()); |
||||
|
return; |
||||
|
} |
||||
|
CustomerId stateCustomerId = state.getEdge().getCustomerId(); |
||||
|
state.setEdge(edge); |
||||
|
if (stateCustomerId != null && !stateCustomerId.equals(edge.getCustomerId())) { |
||||
|
// do not send edge configuration message on customer update
|
||||
|
// message send by separate flow from assign_to or unassing_from customer
|
||||
|
return; |
||||
|
} |
||||
|
EdgeUpdateMsg edgeConfig = EdgeUpdateMsg.newBuilder() |
||||
|
.setConfiguration(EdgeMsgConstructorUtils.constructEdgeConfiguration(edge)).build(); |
||||
|
|
||||
|
ResponseMsg edgeConfigMsg = ResponseMsg.newBuilder() |
||||
|
.setEdgeUpdateMsg(edgeConfig) |
||||
|
.build(); |
||||
|
|
||||
|
session.sendDownlinkMsg(edgeConfigMsg); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void destroyAndMarkAsZombieIfFailed() { |
||||
|
EdgeSessionState state = getState(); |
||||
|
final var finalSession = session; |
||||
|
|
||||
|
try (finalSession) { |
||||
|
if (!destroy()) { |
||||
|
if (state != null) { |
||||
|
log.warn("[{}][{}] Session destroy failed for edge [{}] with session id [{}]. Adding to zombie queue for later cleanup.", |
||||
|
state.getTenantId(), state.getEdgeId(), state.getEdge().getName(), state.getSessionId()); |
||||
|
} else { |
||||
|
log.warn("Session destroy failed (state is null). Adding to zombie queue for later cleanup."); |
||||
|
} |
||||
|
zombieSessionCleanupService.add(this); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
if (state != null) { |
||||
|
log.warn("[{}][{}] Exception during session destroy for edge [{}] with session id [{}]", |
||||
|
state.getTenantId(), state.getEdgeId(), state.getEdge().getName(), state.getSessionId(), e); |
||||
|
} else { |
||||
|
log.warn("Exception during session destroy (state is null)", e); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,47 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session.manager; |
||||
|
|
||||
|
import io.grpc.stub.StreamObserver; |
||||
|
import org.thingsboard.server.common.data.edge.Edge; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.gen.edge.v1.RequestMsg; |
||||
|
import org.thingsboard.server.gen.edge.v1.ResponseMsg; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
|
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ScheduledExecutorService; |
||||
|
import java.util.function.BiConsumer; |
||||
|
|
||||
|
public interface EdgeGrpcSessionManager { |
||||
|
|
||||
|
StreamObserver<RequestMsg> initInputStream(StreamObserver<ResponseMsg> outputStream, |
||||
|
BiConsumer<EdgeId, EdgeGrpcSessionManager> sessionOpenListener, |
||||
|
BiConsumer<Edge, UUID> sessionCloseListener, |
||||
|
ScheduledExecutorService sendDownlinkExecutorService); |
||||
|
EdgeSessionState getState(); |
||||
|
void addEventToHighPriorityQueue(EdgeEvent edgeEvent); |
||||
|
void startSyncProcess(boolean fullSync); |
||||
|
void onEdgeConnect(); |
||||
|
void onEdgeEventUpdate(); |
||||
|
void onConfigurationUpdate(Edge edge); |
||||
|
void onEdgeDisconnect(); |
||||
|
void onEdgeRemoval(); |
||||
|
void destroyAndMarkAsZombieIfFailed(); |
||||
|
boolean destroy(); |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,313 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session.manager; |
||||
|
|
||||
|
import lombok.Getter; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.config.ConfigurableBeanFactory; |
||||
|
import org.springframework.context.annotation.Scope; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.common.util.ThingsBoardThreadFactory; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.util.ProtoUtils; |
||||
|
import org.thingsboard.server.gen.edge.v1.DownlinkMsg; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos.ToEdgeEventNotificationMsg; |
||||
|
import org.thingsboard.server.queue.TbQueueConsumer; |
||||
|
import org.thingsboard.server.queue.common.TbProtoQueueMsg; |
||||
|
import org.thingsboard.server.queue.common.consumer.QueueConsumerManager; |
||||
|
import org.thingsboard.server.queue.discovery.TopicService; |
||||
|
import org.thingsboard.server.queue.kafka.KafkaAdmin; |
||||
|
import org.thingsboard.server.queue.provider.TbCoreQueueFactory; |
||||
|
import org.thingsboard.server.queue.util.TbKafkaComponent; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.processor.PostgresGeneralEdgeEventsDispatcher; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeSessionsHolder; |
||||
|
|
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
import java.util.concurrent.ExecutorService; |
||||
|
import java.util.concurrent.Executors; |
||||
|
import java.util.concurrent.Future; |
||||
|
import java.util.concurrent.ScheduledFuture; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
import java.util.concurrent.atomic.AtomicReference; |
||||
|
import java.util.concurrent.locks.Lock; |
||||
|
import java.util.concurrent.locks.ReentrantLock; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@TbKafkaComponent |
||||
|
@Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE) |
||||
|
@RequiredArgsConstructor |
||||
|
public class KafkaBasedEdgeGrpcSessionManager extends AbstractEdgeGrpcSessionManager { |
||||
|
|
||||
|
private static final int NO_INITIAL_DELAY_VALUE = 0; |
||||
|
|
||||
|
private final TbCoreQueueFactory tbCoreQueueFactory; |
||||
|
private final TopicService topicService; |
||||
|
private final KafkaAdmin kafkaAdmin; |
||||
|
private final EdgeSessionsHolder sessions; |
||||
|
|
||||
|
private final AtomicReference<ScheduledFuture<?>> initMigrationAndProcessingFutureRef = new AtomicReference<>(); |
||||
|
private final AtomicReference<ScheduledFuture<?>> highPriorityProcessingFutureRef = new AtomicReference<>(); |
||||
|
private final Lock initLock = new ReentrantLock(); |
||||
|
private volatile boolean migrationProcessed; |
||||
|
private volatile boolean isHighPriorityProcessing; |
||||
|
@Getter |
||||
|
private QueueConsumerManager<TbProtoQueueMsg<ToEdgeEventNotificationMsg>> consumer; |
||||
|
private PostgresGeneralEdgeEventsDispatcher psqlGeneralEdgeEventsDispatcher; |
||||
|
private ExecutorService consumerExecutor; |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeConnect() { |
||||
|
scheduleMigrationAndProcessing(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeEventUpdate() {} |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeDisconnect() {} |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeRemoval() { |
||||
|
EdgeSessionState state = getState(); |
||||
|
String topic = topicService |
||||
|
.buildEdgeEventNotificationsTopicPartitionInfo(state.getTenantId(), state.getEdgeId()) |
||||
|
.getTopic(); |
||||
|
kafkaAdmin.deleteTopic(topic); |
||||
|
kafkaAdmin.deleteConsumerGroup(topic); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean destroy() { |
||||
|
cancelHighPriorityProcessing(); |
||||
|
EdgeSessionState state = getState(); |
||||
|
try { |
||||
|
if (consumer != null) { |
||||
|
log.info("[{}][{}] Stopping edge event consumer...", state.getTenantId(), state.getEdgeId()); |
||||
|
consumer.stop(); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}][{}] Failed to stop edge event consumer", state.getTenantId(), state.getEdgeId(), e); |
||||
|
return false; |
||||
|
} |
||||
|
consumer = null; |
||||
|
try { |
||||
|
if (consumerExecutor != null && !consumerExecutor.isShutdown()) { |
||||
|
consumerExecutor.shutdown(); |
||||
|
awaitConsumerTermination(); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}][{}] Failed to shutdown edge event consumer executor", state.getTenantId(), state.getEdgeId(), e); |
||||
|
return false; |
||||
|
} |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
private void scheduleMigrationAndProcessing() { |
||||
|
cancelMigrationAndProcessingInit(); |
||||
|
EdgeSessionState state = getSession().getState(); |
||||
|
TenantId tenantId = state.getTenantId(); |
||||
|
EdgeId edgeId = state.getEdgeId(); |
||||
|
|
||||
|
if (!sessions.hasByEdgeId(edgeId)) { |
||||
|
log.debug("[{}] Session was removed and edge event consumer must not be started [{}]", |
||||
|
tenantId, edgeId.getId()); |
||||
|
return; |
||||
|
} |
||||
|
ScheduledFuture<?> initMigrationAndProcessingTask = ctx.getEdgeEventProcessingExecutorService().schedule(() -> { |
||||
|
log.info("[{}] Edge event processing task started for edge [{}]", tenantId, edgeId); |
||||
|
initLock.lock(); |
||||
|
try { |
||||
|
processRemainingPostgresEventsIfRequired(edgeId); |
||||
|
if (migrationProcessed) { |
||||
|
initAndLaunchConsumerOrRetry(tenantId, edgeId, state); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}] Failed to process edge events for edge [{}]!", tenantId, edgeId, e); |
||||
|
} finally { |
||||
|
initLock.unlock(); |
||||
|
} |
||||
|
}, ctx.getEdgeEventStorageSettings().getNoRecordsSleepInterval(), TimeUnit.MILLISECONDS); |
||||
|
|
||||
|
initMigrationAndProcessingFutureRef.set(initMigrationAndProcessingTask); |
||||
|
} |
||||
|
|
||||
|
private void initAndLaunchConsumerOrRetry(TenantId tenantId, EdgeId edgeId, EdgeSessionState state) { |
||||
|
if (initAndLaunchConsumer(tenantId, edgeId, state)) { |
||||
|
launchProcessingOfHighPriorityEvents(tenantId, edgeId); |
||||
|
} else { |
||||
|
scheduleMigrationAndProcessing(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void launchProcessingOfHighPriorityEvents(TenantId tenantId, EdgeId edgeId) { |
||||
|
cancelHighPriorityProcessing(); |
||||
|
ScheduledFuture<?> highPriorityProcessingTask = ctx.getEdgeEventProcessingExecutorService().scheduleAtFixedRate(() -> { |
||||
|
try { |
||||
|
isHighPriorityProcessing = true; |
||||
|
session.processHighPriorityEvents(); |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}] Failed to process edge events for edge [{}]!", tenantId, edgeId, e); |
||||
|
} finally { |
||||
|
isHighPriorityProcessing = false; |
||||
|
} |
||||
|
}, NO_INITIAL_DELAY_VALUE, ctx.getEdgeEventStorageSettings().getNoRecordsSleepInterval(), TimeUnit.MILLISECONDS); |
||||
|
highPriorityProcessingFutureRef.set(highPriorityProcessingTask); |
||||
|
} |
||||
|
|
||||
|
private boolean initAndLaunchConsumer(TenantId tenantId, EdgeId edgeId, EdgeSessionState state) { |
||||
|
if (!isSessionReady(state)) { |
||||
|
log.warn("[{}][{}] Session is not ready (connected={}, syncInProgress={}, highPriority={}), skip starting edge event consumer", |
||||
|
tenantId, edgeId, state.isConnected(), state.isSyncInProgress(), isHighPriorityProcessing); |
||||
|
return false; |
||||
|
} |
||||
|
if (isConsumerInitializationRequired()) { |
||||
|
initConsumerAndExecutor(tenantId, edgeId, state); |
||||
|
} |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
private void initConsumerAndExecutor(TenantId tenantId, EdgeId edgeId, EdgeSessionState state) { |
||||
|
try { |
||||
|
if (consumerExecutor != null && !consumerExecutor.isShutdown()) { |
||||
|
try { |
||||
|
consumerExecutor.shutdown(); |
||||
|
awaitConsumerTermination(); |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}][{}] Failed to shutdown previous consumer executor", tenantId, edgeId, e); |
||||
|
} |
||||
|
} |
||||
|
this.consumerExecutor = Executors.newSingleThreadExecutor(ThingsBoardThreadFactory.forName("edge-event-consumer")); |
||||
|
this.consumer = QueueConsumerManager.<TbProtoQueueMsg<ToEdgeEventNotificationMsg>>builder() |
||||
|
.name("TB Edge events [" + edgeId + "]") |
||||
|
.msgPackProcessor(this::processMsgs) |
||||
|
.pollInterval(ctx.getEdgeEventStorageSettings().getNoRecordsSleepInterval()) |
||||
|
.consumerCreator(() -> tbCoreQueueFactory.createEdgeEventMsgConsumer(tenantId, edgeId)) |
||||
|
.consumerExecutor(consumerExecutor) |
||||
|
.threadPrefix("edge-events-" + edgeId) |
||||
|
.readinessCheck(this::isReadyToProcessGeneralEvents) |
||||
|
.build(); |
||||
|
consumer.subscribe(); |
||||
|
consumer.launch(); |
||||
|
} catch (Exception e) { |
||||
|
destroy(); |
||||
|
log.warn("[{}][{}] Failed to start edge event consumer", state.getSessionId(), edgeId, e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean isConsumerInitializationRequired() { |
||||
|
return consumer == null || (consumer.getConsumer() != null && consumer.getConsumer().isStopped()); |
||||
|
} |
||||
|
|
||||
|
private void processMsgs(List<TbProtoQueueMsg<ToEdgeEventNotificationMsg>> msgs, TbQueueConsumer<TbProtoQueueMsg<ToEdgeEventNotificationMsg>> consumer) { |
||||
|
EdgeSessionState state = getSession().getState(); |
||||
|
TenantId tenantId = state.getTenantId(); |
||||
|
EdgeId edgeId = state.getEdgeId(); |
||||
|
|
||||
|
log.trace("[{}][{}] starting processing edge events", tenantId, edgeId); |
||||
|
// Defensive backstop: the loop already gates polling on readiness; this only fires on the narrow race
|
||||
|
// where readiness flips during poll(), and that already-polled batch is dropped here (can't rewind).
|
||||
|
if (!isSessionReady(state)) { |
||||
|
log.debug("[{}][{}] edge not connected, edge sync is not completed or high priority processing in progress, " + |
||||
|
"connected = {}, sync in progress = {}, high priority in progress = {}. Skipping iteration", |
||||
|
tenantId, edgeId, state.isConnected(), state.isSyncInProgress(), isHighPriorityProcessing); |
||||
|
return; |
||||
|
} |
||||
|
List<EdgeEvent> edgeEvents = new ArrayList<>(); |
||||
|
for (TbProtoQueueMsg<ToEdgeEventNotificationMsg> msg : msgs) { |
||||
|
EdgeEvent edgeEvent = ProtoUtils.fromProto(msg.getValue().getEdgeEventMsg()); |
||||
|
edgeEvents.add(edgeEvent); |
||||
|
} |
||||
|
List<DownlinkMsg> downlinkMsgsPack = downlinkMessageMapper.convertToDownlinkMsgsPack(state, edgeEvents); |
||||
|
boolean isInterrupted = true; |
||||
|
try { |
||||
|
isInterrupted = session.sendDownlinkMsgsPack(downlinkMsgsPack).get(); |
||||
|
if (isInterrupted) { |
||||
|
log.debug("[{}][{}] Send downlink messages task was interrupted", tenantId, edgeId); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.error("[{}][{}] Failed to process downlink messages", tenantId, edgeId, e); |
||||
|
} |
||||
|
if (!isInterrupted) { |
||||
|
consumer.commit(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean isReadyToProcessGeneralEvents() { |
||||
|
return isSessionReady(getState()); |
||||
|
} |
||||
|
|
||||
|
/** Single source of truth for "the session may process general edge events" - see {@link #isReadyToProcessGeneralEvents}. */ |
||||
|
private boolean isSessionReady(EdgeSessionState state) { |
||||
|
return state != null && state.isConnected() && !state.isSyncInProgress() && !isHighPriorityProcessing; |
||||
|
} |
||||
|
|
||||
|
private void cancelMigrationAndProcessingInit() { |
||||
|
EdgeSessionState state = getState(); |
||||
|
log.trace("[{}] cancelling edge migration & processing init for edge", state != null ? state.getEdgeId() : "unknown"); |
||||
|
cancelFuture(initMigrationAndProcessingFutureRef); |
||||
|
} |
||||
|
|
||||
|
private void cancelHighPriorityProcessing() { |
||||
|
EdgeSessionState state = getState(); |
||||
|
log.trace("[{}] cancelling high priority processing task for edge", state != null ? state.getEdgeId() : "unknown"); |
||||
|
cancelFuture(highPriorityProcessingFutureRef); |
||||
|
} |
||||
|
|
||||
|
private void cancelFuture(AtomicReference<? extends Future<?>> futureReference) { |
||||
|
Future<?> f = futureReference.getAndSet(null); |
||||
|
if (f != null && !f.isCancelled() && !f.isDone()) { |
||||
|
f.cancel(true); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void processRemainingPostgresEventsIfRequired(EdgeId edgeId) throws Exception { |
||||
|
log.debug("Processing remaining postgres edge events for edge [{}]", edgeId); |
||||
|
|
||||
|
if (psqlGeneralEdgeEventsDispatcher == null) { |
||||
|
psqlGeneralEdgeEventsDispatcher = new PostgresGeneralEdgeEventsDispatcher(session, ctx); |
||||
|
} |
||||
|
if (migrationProcessed) { |
||||
|
return; |
||||
|
} |
||||
|
Boolean hasMorePostgresEvents = psqlGeneralEdgeEventsDispatcher.processNewEvents().get(); |
||||
|
if (Boolean.TRUE.equals(hasMorePostgresEvents)) { |
||||
|
log.info("Migration still in progress for edge [{}]", edgeId.getId()); |
||||
|
scheduleMigrationAndProcessing(); |
||||
|
} else if (Boolean.FALSE.equals(hasMorePostgresEvents)) { |
||||
|
log.info("Migration completed for edge [{}]", edgeId.getId()); |
||||
|
migrationProcessed = true; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void awaitConsumerTermination() { |
||||
|
EdgeSessionState state = getState(); |
||||
|
try { |
||||
|
if (!consumerExecutor.awaitTermination(10, java.util.concurrent.TimeUnit.SECONDS)) { |
||||
|
consumerExecutor.shutdownNow(); |
||||
|
} |
||||
|
} catch (InterruptedException ie) { |
||||
|
log.warn("[{}][{}] Interrupted while awaiting consumer executor termination", state.getTenantId(), state.getEdgeId()); |
||||
|
Thread.currentThread().interrupt(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,183 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.edge.rpc.session.manager; |
||||
|
|
||||
|
import com.google.common.util.concurrent.FutureCallback; |
||||
|
import com.google.common.util.concurrent.Futures; |
||||
|
import com.google.common.util.concurrent.ListenableFuture; |
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.springframework.beans.factory.config.ConfigurableBeanFactory; |
||||
|
import org.springframework.boot.autoconfigure.condition.ConditionalOnExpression; |
||||
|
import org.springframework.context.annotation.Scope; |
||||
|
import org.springframework.stereotype.Service; |
||||
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
||||
|
import org.thingsboard.server.common.data.id.EdgeId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.service.edge.rpc.EdgeSessionState; |
||||
|
import org.thingsboard.server.service.edge.rpc.processor.PostgresGeneralEdgeEventsDispatcher; |
||||
|
import org.thingsboard.server.service.edge.rpc.session.EdgeSessionsHolder; |
||||
|
|
||||
|
import java.util.concurrent.ScheduledFuture; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
import java.util.concurrent.atomic.AtomicReference; |
||||
|
import java.util.concurrent.locks.Lock; |
||||
|
import java.util.concurrent.locks.ReentrantLock; |
||||
|
|
||||
|
@Service |
||||
|
@Slf4j |
||||
|
@ConditionalOnExpression("'${queue.type:null}'!='kafka'") |
||||
|
@Scope(ConfigurableBeanFactory.SCOPE_PROTOTYPE) |
||||
|
@RequiredArgsConstructor |
||||
|
public class PostgresBasedEdgeGrpcSessionManager extends AbstractEdgeGrpcSessionManager { |
||||
|
|
||||
|
private final EdgeSessionsHolder edgeSessions; |
||||
|
|
||||
|
private final Lock newEventsLock = new ReentrantLock(); |
||||
|
private final AtomicReference<ScheduledFuture<?>> edgeEventCheckFutureRef = new AtomicReference<>(); |
||||
|
private PostgresGeneralEdgeEventsDispatcher generalEdgeEventsDispatcher; |
||||
|
private volatile boolean hasNewEvents; |
||||
|
|
||||
|
@Override |
||||
|
public void addEventToHighPriorityQueue(EdgeEvent edgeEvent) { |
||||
|
super.addEventToHighPriorityQueue(edgeEvent); |
||||
|
markHasNewEvents(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeConnect() { |
||||
|
markHasNewEvents(); |
||||
|
scheduleEdgeEventsCheck(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeEventUpdate() { |
||||
|
markHasNewEvents(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeDisconnect() { |
||||
|
markHasNoEvents(); |
||||
|
cancelScheduleEdgeEventsCheck(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onEdgeRemoval() { |
||||
|
markHasNoEvents(); |
||||
|
cancelScheduleEdgeEventsCheck(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public boolean destroy() { |
||||
|
markHasNoEvents(); |
||||
|
cancelScheduleEdgeEventsCheck(); |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
private void scheduleEdgeEventsCheck() { |
||||
|
cancelScheduleEdgeEventsCheck(); |
||||
|
EdgeId edgeId = getState().getEdgeId(); |
||||
|
TenantId tenantId = getState().getTenantId(); |
||||
|
|
||||
|
if (!edgeSessions.hasByEdgeId(edgeId)) { |
||||
|
log.debug("[{}] Session was removed and edge event check schedule must not be started [{}]", |
||||
|
tenantId, edgeId.getId()); |
||||
|
return; |
||||
|
} |
||||
|
ScheduledFuture<?> edgeEventCheckTask = ctx.getEdgeEventProcessingExecutorService().schedule(() -> { |
||||
|
try { |
||||
|
newEventsLock.lock(); |
||||
|
try { |
||||
|
if (!hasNewEvents) { |
||||
|
scheduleEdgeEventsCheck(); |
||||
|
return; |
||||
|
} |
||||
|
log.trace("[{}][{}] set session new events flag to false", tenantId, edgeId.getId()); |
||||
|
hasNewEvents = false; |
||||
|
session.processHighPriorityEvents(); |
||||
|
processEdgeEventsWithCallback(tenantId, edgeId); |
||||
|
} finally { |
||||
|
newEventsLock.unlock(); |
||||
|
} |
||||
|
} catch (Exception e) { |
||||
|
log.warn("[{}] Failed to process edge events for edge [{}]!", tenantId, edgeId, e); |
||||
|
} |
||||
|
}, ctx.getEdgeEventStorageSettings().getNoRecordsSleepInterval(), TimeUnit.MILLISECONDS); |
||||
|
|
||||
|
edgeEventCheckFutureRef.set(edgeEventCheckTask); |
||||
|
log.trace("[{}] Check edge event scheduled for edge [{}]", tenantId, edgeId.getId()); |
||||
|
} |
||||
|
|
||||
|
private void processEdgeEventsWithCallback(TenantId tenantId, EdgeId edgeId) throws Exception { |
||||
|
Futures.addCallback(processEdgeEvents(), new FutureCallback<>() { |
||||
|
@Override |
||||
|
public void onSuccess(Boolean newEventsAdded) { |
||||
|
if (Boolean.TRUE.equals(newEventsAdded)) { |
||||
|
log.trace("[{}][{}] new events added. set session new events flag to true", tenantId, edgeId.getId()); |
||||
|
hasNewEvents = true; |
||||
|
} |
||||
|
scheduleEdgeEventsCheck(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onFailure(Throwable t) { |
||||
|
log.warn("[{}] Failed to process edge events for edge [{}]!", tenantId, edgeId, t); |
||||
|
scheduleEdgeEventsCheck(); |
||||
|
} |
||||
|
}, ctx.getGrpcCallbackExecutorService()); |
||||
|
} |
||||
|
|
||||
|
public ListenableFuture<Boolean> processEdgeEvents() throws Exception { |
||||
|
if (generalEdgeEventsDispatcher == null) { |
||||
|
generalEdgeEventsDispatcher = new PostgresGeneralEdgeEventsDispatcher(session, ctx); |
||||
|
} |
||||
|
return generalEdgeEventsDispatcher.processNewEvents(); |
||||
|
} |
||||
|
|
||||
|
private void cancelScheduleEdgeEventsCheck() { |
||||
|
EdgeSessionState state = getState(); |
||||
|
log.trace("[{}] cancelling edge event check for edge", state != null ? state.getEdgeId() : "unknown"); |
||||
|
|
||||
|
ScheduledFuture<?> sf = edgeEventCheckFutureRef.getAndSet(null); |
||||
|
if (sf != null && !sf.isCancelled() && !sf.isDone()) { |
||||
|
sf.cancel(true); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void markHasNoEvents() { |
||||
|
markHasEvents(false); |
||||
|
} |
||||
|
|
||||
|
private void markHasNewEvents() { |
||||
|
markHasEvents(true); |
||||
|
} |
||||
|
|
||||
|
private void markHasEvents(boolean newEventsPresent) { |
||||
|
newEventsLock.lock(); |
||||
|
try { |
||||
|
if (hasNewEvents != newEventsPresent) { |
||||
|
EdgeSessionState state = getState(); |
||||
|
if (state != null) { |
||||
|
log.trace("[{}] set session new events flag to {} [{}]", state.getTenantId(), newEventsPresent, state.getEdgeId()); |
||||
|
} |
||||
|
hasNewEvents = newEventsPresent; |
||||
|
} |
||||
|
} finally { |
||||
|
newEventsLock.unlock(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,19 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.iot_hub; |
||||
|
|
||||
|
public record InstallReport(String tenantHash, String userHash, String tbVersion, String edition) { |
||||
|
} |
||||
@ -0,0 +1,35 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.service.queue.processing.TbRuleEngineSubmitStrategy; |
||||
|
|
||||
|
public interface TbMsgPackProcessingContextFactory { |
||||
|
|
||||
|
TbMsgPackProcessingContext create(String queueName, TbRuleEngineSubmitStrategy submitStrategy, boolean skipTimeouts); |
||||
|
|
||||
|
@Component |
||||
|
class DefaultTbMsgPackProcessingContextFactory implements TbMsgPackProcessingContextFactory { |
||||
|
|
||||
|
@Override |
||||
|
public TbMsgPackProcessingContext create(String queueName, TbRuleEngineSubmitStrategy submitStrategy, boolean skipTimeouts) { |
||||
|
return new TbMsgPackProcessingContext(queueName, submitStrategy, skipTimeouts); |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,71 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.controller; |
||||
|
|
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.junit.After; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.thingsboard.server.common.data.pat.ApiKey; |
||||
|
import org.thingsboard.server.common.data.pat.ApiKeyInfo; |
||||
|
import org.thingsboard.server.dao.service.DaoSqlTest; |
||||
|
|
||||
|
import java.net.URI; |
||||
|
import java.net.URISyntaxException; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; |
||||
|
|
||||
|
@Slf4j |
||||
|
@DaoSqlTest |
||||
|
public class ApiKeyWebSocketApiTest extends WebSocketApiTest { |
||||
|
|
||||
|
private ApiKey apiKey; |
||||
|
|
||||
|
@Before |
||||
|
public void setUpApiKey() { |
||||
|
ApiKeyInfo apiKeyInfo = new ApiKeyInfo(); |
||||
|
apiKeyInfo.setDescription("WS test API key"); |
||||
|
apiKeyInfo.setEnabled(true); |
||||
|
apiKeyInfo.setUserId(tenantAdminUserId); |
||||
|
apiKey = doPost("/api/apiKey", apiKeyInfo, ApiKey.class); |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void tearDownApiKey() throws Exception { |
||||
|
loginTenantAdmin(); |
||||
|
doDelete("/api/apiKey/" + apiKey.getId()).andExpect(status().isOk()); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
protected TbTestWebSocketClient buildAndConnectWebSocketClient() throws URISyntaxException, InterruptedException { |
||||
|
return buildAndConnectWebSocketClientWithApiKey(apiKey.getValue()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testInvalidApiKeyAuthCmd_connectionClosed() throws Exception { |
||||
|
TbTestWebSocketClient client = new TbTestWebSocketClient(new URI(WS_URL + wsPort + "/api/ws")); |
||||
|
assertThat(client.connectBlocking(TIMEOUT, TimeUnit.SECONDS)).isTrue(); |
||||
|
try { |
||||
|
client.authenticateWithApiKey("invalid-key"); |
||||
|
assertThat(client.waitForClose()).isTrue(); |
||||
|
} finally { |
||||
|
client.close(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,46 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.iot_hub; |
||||
|
|
||||
|
import org.junit.jupiter.api.Test; |
||||
|
|
||||
|
import java.security.MessageDigest; |
||||
|
import java.nio.charset.StandardCharsets; |
||||
|
import java.util.HexFormat; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.junit.jupiter.api.Assertions.assertEquals; |
||||
|
|
||||
|
class InstallReportTest { |
||||
|
|
||||
|
private static String sha256(String s) throws Exception { |
||||
|
MessageDigest d = MessageDigest.getInstance("SHA-256"); |
||||
|
return HexFormat.of().formatHex(d.digest(s.getBytes(StandardCharsets.UTF_8))); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void buildInstallReport_hashesMatchSaltedFormula() throws Exception { |
||||
|
UUID tenantId = UUID.randomUUID(); |
||||
|
UUID userId = UUID.randomUUID(); |
||||
|
|
||||
|
InstallReport r = DefaultIotHubService.buildInstallReport(tenantId, userId, "4.2.0", "CE"); |
||||
|
|
||||
|
assertEquals(sha256(tenantId.toString() + tenantId), r.tenantHash()); |
||||
|
assertEquals(sha256(tenantId.toString() + userId), r.userHash()); |
||||
|
assertEquals("4.2.0", r.tbVersion()); |
||||
|
assertEquals("CE", r.edition()); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,281 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.ruleengine; |
||||
|
|
||||
|
import com.google.common.util.concurrent.MoreExecutors; |
||||
|
import org.junit.jupiter.api.AfterEach; |
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.junit.jupiter.api.extension.ExtendWith; |
||||
|
import org.mockito.InOrder; |
||||
|
import org.mockito.Mock; |
||||
|
import org.mockito.junit.jupiter.MockitoExtension; |
||||
|
import org.thingsboard.common.util.ThingsBoardExecutors; |
||||
|
import org.thingsboard.common.util.ThingsBoardThreadFactory; |
||||
|
import org.thingsboard.server.actors.ActorSystemContext; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.msg.TbMsgType; |
||||
|
import org.thingsboard.server.common.data.queue.ProcessingStrategy; |
||||
|
import org.thingsboard.server.common.data.queue.ProcessingStrategyType; |
||||
|
import org.thingsboard.server.common.data.queue.Queue; |
||||
|
import org.thingsboard.server.common.data.queue.SubmitStrategy; |
||||
|
import org.thingsboard.server.common.data.queue.SubmitStrategyType; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbMsgMetaData; |
||||
|
import org.thingsboard.server.common.msg.queue.ServiceType; |
||||
|
import org.thingsboard.server.common.msg.queue.TopicPartitionInfo; |
||||
|
import org.thingsboard.server.common.stats.StatsFactory; |
||||
|
import org.thingsboard.server.gen.transport.TransportProtos; |
||||
|
import org.thingsboard.server.queue.TbQueueAdmin; |
||||
|
import org.thingsboard.server.queue.TbQueueConsumer; |
||||
|
import org.thingsboard.server.queue.TbQueueMsg; |
||||
|
import org.thingsboard.server.queue.common.DefaultTbQueueMsgHeaders; |
||||
|
import org.thingsboard.server.queue.common.TbProtoQueueMsg; |
||||
|
import org.thingsboard.server.queue.discovery.PartitionService; |
||||
|
import org.thingsboard.server.queue.discovery.QueueKey; |
||||
|
import org.thingsboard.server.queue.discovery.TbServiceInfoProvider; |
||||
|
import org.thingsboard.server.queue.memory.DefaultInMemoryStorage; |
||||
|
import org.thingsboard.server.queue.memory.InMemoryStorage; |
||||
|
import org.thingsboard.server.queue.memory.InMemoryTbQueueConsumer; |
||||
|
import org.thingsboard.server.queue.provider.TbQueueProducerProvider; |
||||
|
import org.thingsboard.server.queue.provider.TbRuleEngineQueueFactory; |
||||
|
import org.thingsboard.server.service.queue.TbMsgPackProcessingContext; |
||||
|
import org.thingsboard.server.service.queue.TbMsgPackProcessingContextFactory; |
||||
|
import org.thingsboard.server.service.queue.processing.TbRuleEngineProcessingStrategyFactory; |
||||
|
import org.thingsboard.server.service.queue.processing.TbRuleEngineSubmitStrategy; |
||||
|
import org.thingsboard.server.service.queue.processing.TbRuleEngineSubmitStrategyFactory; |
||||
|
import org.thingsboard.server.service.stats.RuleEngineStatisticsService; |
||||
|
|
||||
|
import java.time.Duration; |
||||
|
import java.util.List; |
||||
|
import java.util.Set; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||
|
import java.util.concurrent.ExecutorService; |
||||
|
import java.util.concurrent.Executors; |
||||
|
import java.util.concurrent.ScheduledExecutorService; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
import java.util.concurrent.atomic.AtomicInteger; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.awaitility.Awaitility.await; |
||||
|
import static org.mockito.ArgumentMatchers.any; |
||||
|
import static org.mockito.ArgumentMatchers.anyLong; |
||||
|
import static org.mockito.ArgumentMatchers.eq; |
||||
|
import static org.mockito.ArgumentMatchers.isNull; |
||||
|
import static org.mockito.BDDMockito.given; |
||||
|
import static org.mockito.Mockito.doAnswer; |
||||
|
import static org.mockito.Mockito.inOrder; |
||||
|
import static org.mockito.Mockito.spy; |
||||
|
import static org.mockito.Mockito.when; |
||||
|
|
||||
|
@ExtendWith(MockitoExtension.class) |
||||
|
class RuleEngineConsumerLoopTest { |
||||
|
|
||||
|
TenantId tenantId = TenantId.fromUUID(UUID.randomUUID()); |
||||
|
DeviceId deviceId = new DeviceId(UUID.randomUUID()); |
||||
|
|
||||
|
InMemoryStorage storage; |
||||
|
|
||||
|
@Mock |
||||
|
ActorSystemContext actorContext; |
||||
|
@Mock |
||||
|
StatsFactory statsFactory; |
||||
|
@Mock |
||||
|
TbRuleEngineQueueFactory queueFactory; |
||||
|
@Mock |
||||
|
RuleEngineStatisticsService statisticsService; |
||||
|
@Mock |
||||
|
TbServiceInfoProvider serviceInfoProvider; |
||||
|
@Mock |
||||
|
PartitionService partitionService; |
||||
|
@Mock |
||||
|
TbQueueProducerProvider producerProvider; |
||||
|
@Mock |
||||
|
TbQueueAdmin queueAdmin; |
||||
|
@Mock |
||||
|
TbMsgPackProcessingContextFactory packProcessingContextFactory; |
||||
|
@Mock |
||||
|
TbMsgPackProcessingContext packCtx; |
||||
|
|
||||
|
Queue mainQueue; |
||||
|
|
||||
|
TbQueueConsumer<TbProtoQueueMsg<TransportProtos.ToRuleEngineMsg>> consumer; |
||||
|
|
||||
|
TbRuleEngineConsumerContext ruleEngineConsumerContext; |
||||
|
TbRuleEngineQueueConsumerManager consumerManager; |
||||
|
|
||||
|
ExecutorService consumersExecutor; |
||||
|
ScheduledExecutorService scheduler; |
||||
|
ExecutorService mgmtExecutor; |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setup() throws InterruptedException { |
||||
|
consumersExecutor = Executors.newCachedThreadPool(ThingsBoardThreadFactory.forName("tb-rule-engine-consumer")); |
||||
|
scheduler = ThingsBoardExecutors.newSingleThreadScheduledExecutor("tb-rule-engine-consumer-scheduler"); |
||||
|
mgmtExecutor = ThingsBoardExecutors.newWorkStealingPool(1, "tb-rule-engine-mgmt"); |
||||
|
|
||||
|
mainQueue = new Queue(); |
||||
|
mainQueue.setTenantId(TenantId.SYS_TENANT_ID); |
||||
|
mainQueue.setName("Main"); |
||||
|
mainQueue.setTopic("tb_rule_engine.main"); |
||||
|
mainQueue.setPollInterval(25); |
||||
|
mainQueue.setPartitions(1); |
||||
|
mainQueue.setConsumerPerPartition(false); |
||||
|
mainQueue.setPackProcessingTimeout(2000L); |
||||
|
|
||||
|
var submitStrategy = new SubmitStrategy(); |
||||
|
submitStrategy.setType(SubmitStrategyType.BURST); |
||||
|
submitStrategy.setBatchSize(1000); |
||||
|
mainQueue.setSubmitStrategy(submitStrategy); |
||||
|
|
||||
|
var processingStrategy = new ProcessingStrategy(); |
||||
|
processingStrategy.setType(ProcessingStrategyType.SKIP_ALL_FAILURES); |
||||
|
processingStrategy.setRetries(3); |
||||
|
processingStrategy.setFailurePercentage(0.0); |
||||
|
processingStrategy.setPauseBetweenRetries(3); |
||||
|
processingStrategy.setMaxPauseBetweenRetries(3); |
||||
|
mainQueue.setProcessingStrategy(processingStrategy); |
||||
|
|
||||
|
storage = new DefaultInMemoryStorage(); |
||||
|
|
||||
|
consumer = spy(new InMemoryTbQueueConsumer<>(storage, mainQueue.getTopic())); |
||||
|
given(queueFactory.createToRuleEngineMsgConsumer(eq(mainQueue), isNull())).willReturn(consumer); |
||||
|
|
||||
|
ruleEngineConsumerContext = new TbRuleEngineConsumerContext( |
||||
|
actorContext, statsFactory, new TbRuleEngineSubmitStrategyFactory(), new TbRuleEngineProcessingStrategyFactory(), |
||||
|
queueFactory, statisticsService, serviceInfoProvider, partitionService, producerProvider, queueAdmin |
||||
|
); |
||||
|
ruleEngineConsumerContext.setPollDuration(25); |
||||
|
ruleEngineConsumerContext.setPackProcessingTimeout(2000); |
||||
|
ruleEngineConsumerContext.setStatsEnabled(false); // true by default
|
||||
|
ruleEngineConsumerContext.setPrometheusStatsEnabled(false); |
||||
|
ruleEngineConsumerContext.setTopicDeletionDelayInSec(15); |
||||
|
ruleEngineConsumerContext.setMgmtThreadPoolSize(12); |
||||
|
|
||||
|
// Tell the (mock) context factory to return (mock) message pack context
|
||||
|
given(packProcessingContextFactory.create( |
||||
|
eq(mainQueue.getName()), |
||||
|
any(TbRuleEngineSubmitStrategy.class), |
||||
|
eq(false) |
||||
|
)).willAnswer(invocation -> { |
||||
|
TbRuleEngineSubmitStrategy realStrategy = invocation.getArgument(1); |
||||
|
when(packCtx.getPendingMap()).thenAnswer(i -> realStrategy.getPendingMap()); |
||||
|
when(packCtx.getFailedMap()).thenReturn(new ConcurrentHashMap<>()); |
||||
|
return packCtx; |
||||
|
}); |
||||
|
|
||||
|
// Tell the (mock) context's await() to return 'false' (always timeout) immediately
|
||||
|
given(packCtx.await(anyLong(), any(TimeUnit.class))).willReturn(false); |
||||
|
|
||||
|
consumerManager = TbRuleEngineQueueConsumerManager.create() |
||||
|
.ctx(ruleEngineConsumerContext) |
||||
|
.queueKey(new QueueKey(ServiceType.TB_RULE_ENGINE, mainQueue)) |
||||
|
.consumerExecutor(consumersExecutor) |
||||
|
.scheduler(scheduler) |
||||
|
.taskExecutor(mgmtExecutor) |
||||
|
.packProcessingContextFactory(packProcessingContextFactory) |
||||
|
.build(); |
||||
|
} |
||||
|
|
||||
|
@AfterEach |
||||
|
void destroy() { |
||||
|
MoreExecutors.shutdownAndAwaitTermination(scheduler, Duration.ofSeconds(30)); |
||||
|
MoreExecutors.shutdownAndAwaitTermination(mgmtExecutor, Duration.ofSeconds(30)); |
||||
|
MoreExecutors.shutdownAndAwaitTermination(consumersExecutor, Duration.ofSeconds(30)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void consumerLoopTest_verifyOperationsOrder() throws InterruptedException { |
||||
|
// Create partition
|
||||
|
var partition = TopicPartitionInfo.builder() |
||||
|
.tenantId(TenantId.SYS_TENANT_ID) |
||||
|
.topic(mainQueue.getTopic()) |
||||
|
.partition(0) |
||||
|
.myPartition(true) |
||||
|
.useInternalPartition(false) |
||||
|
.build(); |
||||
|
|
||||
|
// Put 10k messages to the queue
|
||||
|
for (int i = 0; i < 10_000; i++) { |
||||
|
var tbMsg = TbMsg.newMsg() |
||||
|
.type(TbMsgType.POST_TELEMETRY_REQUEST) |
||||
|
.originator(deviceId) |
||||
|
.data("{\"temperature\":123}") |
||||
|
.metaData(TbMsgMetaData.EMPTY) |
||||
|
.build(); |
||||
|
|
||||
|
var toRuleEngineMsg = TransportProtos.ToRuleEngineMsg.newBuilder() |
||||
|
.setTenantIdLSB(tenantId.getId().getLeastSignificantBits()) |
||||
|
.setTenantIdMSB(tenantId.getId().getMostSignificantBits()) |
||||
|
.setTbMsgProto(TbMsg.toProto(tbMsg)) |
||||
|
.addAllRelationTypes(Set.of("Success")) |
||||
|
.build(); |
||||
|
|
||||
|
storage.put(partition.getFullTopicName(), new TbProtoQueueMsg<>(UUID.randomUUID(), toRuleEngineMsg, new DefaultTbQueueMsgHeaders())); |
||||
|
} |
||||
|
|
||||
|
// Count how many polls were made
|
||||
|
var totalPolls = new AtomicInteger(0); |
||||
|
var emptyPolls = new AtomicInteger(0); |
||||
|
doAnswer(invocation -> { |
||||
|
totalPolls.incrementAndGet(); |
||||
|
@SuppressWarnings("unchecked") |
||||
|
var messages = (List<TbQueueMsg>) invocation.callRealMethod(); |
||||
|
if (messages.isEmpty()) { |
||||
|
emptyPolls.incrementAndGet(); |
||||
|
} |
||||
|
return messages; |
||||
|
}).when(consumer).poll(mainQueue.getPollInterval()); |
||||
|
|
||||
|
// Count how many commits were made
|
||||
|
var totalCommits = new AtomicInteger(0); |
||||
|
doAnswer(invocation -> { |
||||
|
totalCommits.incrementAndGet(); |
||||
|
return invocation.callRealMethod(); |
||||
|
}).when(consumer).commit(); |
||||
|
|
||||
|
// Initialize consumer
|
||||
|
consumerManager.init(mainQueue); |
||||
|
|
||||
|
// Assign partition to the consumer
|
||||
|
consumerManager.update(Set.of(partition)); |
||||
|
|
||||
|
// Give some time for the consumer to get all messages
|
||||
|
await().atMost(Duration.ofSeconds(10L)).until(() -> storage.getLagTotal() == 0); |
||||
|
|
||||
|
// Stop consumer
|
||||
|
consumerManager.stop(); |
||||
|
consumerManager.awaitStop(); |
||||
|
|
||||
|
// Determine number of non-empty consumer iterations made, since polling does not stop immediately after consuming all messages and may do a few empty polls
|
||||
|
int nonEmptyPolls = totalPolls.get() - emptyPolls.get(); |
||||
|
|
||||
|
// Verify that there is 10 polls and 10 matching commits
|
||||
|
// Each poll consumes 1k messages and queue has 10k total, so that means 10k total msgs / 1k msgs per poll = 10 polls
|
||||
|
assertThat(nonEmptyPolls).isEqualTo(10).isEqualTo(totalCommits.get()); |
||||
|
|
||||
|
// Verify that poll-await-commit cycle happened in order with correct await timeout
|
||||
|
InOrder inOrder = inOrder(consumer, packCtx); |
||||
|
for (int i = 0; i < nonEmptyPolls; i++) { |
||||
|
inOrder.verify(consumer).poll(mainQueue.getPollInterval()); |
||||
|
inOrder.verify(packCtx).await(mainQueue.getPackProcessingTimeout(), TimeUnit.MILLISECONDS); |
||||
|
inOrder.verify(consumer).commit(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,72 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2026 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.mqtt.sparkplug.connection; |
||||
|
|
||||
|
import org.eclipse.paho.mqttv5.common.MqttException; |
||||
|
import org.junit.After; |
||||
|
import org.junit.Before; |
||||
|
import org.junit.Test; |
||||
|
import org.thingsboard.server.dao.service.DaoSqlTest; |
||||
|
|
||||
|
/** |
||||
|
* Created by nickAS21 on 16.02.26 |
||||
|
*/ |
||||
|
@DaoSqlTest |
||||
|
public class MqttV5ClientSparkplugBConnectionDevicesCreatingBeforeTest extends AbstractMqttV5ClientSparkplugConnectionTest { |
||||
|
|
||||
|
/** |
||||
|
* String deviceName_1 = deviceId + "_1"; Only name device. Without a complete topic: how it was in the old version. |
||||
|
* String deviceName_2 = groupId + DEVICE_NAME_SPLIT_REGEXP + edgeNode + DEVICE_NAME_SPLIT_REGEXP + deviceId + "_2"; With complete topic: how it was in the new version. |
||||
|
* @throws Exception |
||||
|
*/ |
||||
|
@Before |
||||
|
public void beforeTest() throws Exception { |
||||
|
beforeSparkplugTest(); |
||||
|
seedLegacyAndFullPathDevices(); |
||||
|
} |
||||
|
|
||||
|
@After |
||||
|
public void afterTest() throws MqttException { |
||||
|
if (client.isConnected()) { |
||||
|
client.disconnect(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testClientWithCorrectAccessTokenWithNDEATHTwoDevicesCreatingBeforeFirstNameDeviceIdSecondNameFull() throws Exception { |
||||
|
connectClientWithCorrectAccessTokenWithNDEATHDevicesCreatingBefore_Test(2); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testRenameWhenDeviceFullPathAlreadyExists_Collision() throws Exception { |
||||
|
renameCollisionWhenTargetNameAlreadyExists_Test(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testUnauthorizedRenameAttempt() throws Exception { |
||||
|
unauthorizedRenameAttempt_Test(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testConcurrentFirstMessageRegistration() throws Exception { |
||||
|
concurrentFirstMessageRegistration_Test(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
public void testSparkplugSessionStaysAliveWithZeroMsgId() throws Exception { |
||||
|
sparkplugSessionStaysAliveWithZeroMsgId_Test(); |
||||
|
} |
||||
|
} |
||||
Some files were not shown because too many files changed in this diff
Loading…
Reference in new issue