Examples with GeneralEdgeEventFetcher org.thingsboard.server.service.edge.rpc.fetch.GeneralEdgeEventFetcher used on opensource projects

Example 1 with GeneralEdgeEventFetcher

use of org.thingsboard.server.service.edge.rpc.fetch.GeneralEdgeEventFetcher in project thingsboard by thingsboard.

the class EdgeGrpcSession method processEdgeEvents.

ListenableFuture<Void> processEdgeEvents() throws Exception {
    SettableFuture<Void> result = SettableFuture.create();
    log.trace("[{}] starting processing edge events", this.sessionId);
    if (isConnected() && isSyncCompleted()) {
        Long queueStartTs = getQueueStartTs().get();
        GeneralEdgeEventFetcher fetcher = new GeneralEdgeEventFetcher(queueStartTs, ctx.getEdgeEventService());
        ListenableFuture<UUID> ifOffsetFuture = startProcessingEdgeEvents(fetcher);
        Futures.addCallback(ifOffsetFuture, new FutureCallback<>() {

            @Override
            public void onSuccess(@Nullable UUID ifOffset) {
                if (ifOffset != null) {
                    Long newStartTs = Uuids.unixTimestamp(ifOffset);
                    ListenableFuture<List<Void>> updateFuture = updateQueueStartTs(newStartTs);
                    Futures.addCallback(updateFuture, new FutureCallback<>() {

                        @Override
                        public void onSuccess(@Nullable List<Void> list) {
                            log.debug("[{}] queue offset was updated [{}][{}]", sessionId, ifOffset, newStartTs);
                            result.set(null);
                        }

                        @Override
                        public void onFailure(Throwable t) {
                            log.error("[{}] Failed to update queue offset [{}]", sessionId, ifOffset, t);
                            result.setException(t);
                        }
                    }, ctx.getGrpcCallbackExecutorService());
                } else {
                    log.trace("[{}] ifOffset is null. Skipping iteration without db update", sessionId);
                    result.set(null);
                }
            }

            @Override
            public void onFailure(Throwable t) {
                log.error("[{}] Failed to process events", sessionId, t);
                result.setException(t);
            }
        }, ctx.getGrpcCallbackExecutorService());
    } else {
        log.trace("[{}] edge is not connected or sync is not completed. Skipping iteration", sessionId);
        result.set(null);
    }
    return result;
}