Search in sources :

Example 1 with ToDeviceRpcRequestBody

use of org.thingsboard.server.common.data.rpc.ToDeviceRpcRequestBody in project thingsboard by thingsboard.

the class DefaultTbRuleEngineRpcService method sendRpcRequestToDevice.

@Override
public void sendRpcRequestToDevice(RuleEngineDeviceRpcRequest src, Consumer<RuleEngineDeviceRpcResponse> consumer) {
    ToDeviceRpcRequest request = new ToDeviceRpcRequest(src.getRequestUUID(), src.getTenantId(), src.getDeviceId(), src.isOneway(), src.getExpirationTime(), new ToDeviceRpcRequestBody(src.getMethod(), src.getBody()), src.isPersisted(), src.getRetries(), src.getAdditionalInfo());
    forwardRpcRequestToDeviceActor(request, response -> {
        if (src.isRestApiCall()) {
            sendRpcResponseToTbCore(src.getOriginServiceId(), response);
        }
        consumer.accept(RuleEngineDeviceRpcResponse.builder().deviceId(src.getDeviceId()).requestId(src.getRequestId()).error(response.getError()).response(response.getResponse()).build());
    });
}
Also used : ToDeviceRpcRequest(org.thingsboard.server.common.msg.rpc.ToDeviceRpcRequest) ToDeviceRpcRequestBody(org.thingsboard.server.common.data.rpc.ToDeviceRpcRequestBody)

Example 2 with ToDeviceRpcRequestBody

use of org.thingsboard.server.common.data.rpc.ToDeviceRpcRequestBody in project thingsboard by thingsboard.

the class DefaultGatewayNotificationsService method formDeviceToGatewayRPCRequest.

private ToDeviceRpcRequest formDeviceToGatewayRPCRequest(TenantId tenantId, DeviceId gatewayDeviceId, JsonNode deviceDataNode, String method) {
    ToDeviceRpcRequestBody body = new ToDeviceRpcRequestBody(method, JacksonUtil.toString(deviceDataNode));
    long expTime = System.currentTimeMillis() + rpcTimeout;
    UUID rpcRequestUUID = UUID.randomUUID();
    return new ToDeviceRpcRequest(rpcRequestUUID, tenantId, gatewayDeviceId, true, expTime, body, true, 3, null);
}
Also used : ToDeviceRpcRequest(org.thingsboard.server.common.msg.rpc.ToDeviceRpcRequest) ToDeviceRpcRequestBody(org.thingsboard.server.common.data.rpc.ToDeviceRpcRequestBody) UUID(java.util.UUID)

Example 3 with ToDeviceRpcRequestBody

use of org.thingsboard.server.common.data.rpc.ToDeviceRpcRequestBody in project thingsboard by thingsboard.

the class AbstractRpcController method handleDeviceRPCRequest.

protected DeferredResult<ResponseEntity> handleDeviceRPCRequest(boolean oneWay, DeviceId deviceId, String requestBody, HttpStatus timeoutStatus, HttpStatus noActiveConnectionStatus) throws ThingsboardException {
    try {
        JsonNode rpcRequestBody = JacksonUtil.toJsonNode(requestBody);
        ToDeviceRpcRequestBody body = new ToDeviceRpcRequestBody(rpcRequestBody.get("method").asText(), JacksonUtil.toString(rpcRequestBody.get("params")));
        SecurityUser currentUser = getCurrentUser();
        TenantId tenantId = currentUser.getTenantId();
        final DeferredResult<ResponseEntity> response = new DeferredResult<>();
        long timeout = rpcRequestBody.has(DataConstants.TIMEOUT) ? rpcRequestBody.get(DataConstants.TIMEOUT).asLong() : defaultTimeout;
        long expTime = rpcRequestBody.has(DataConstants.EXPIRATION_TIME) ? rpcRequestBody.get(DataConstants.EXPIRATION_TIME).asLong() : System.currentTimeMillis() + Math.max(minTimeout, timeout);
        UUID rpcRequestUUID = rpcRequestBody.has("requestUUID") ? UUID.fromString(rpcRequestBody.get("requestUUID").asText()) : UUID.randomUUID();
        boolean persisted = rpcRequestBody.has(DataConstants.PERSISTENT) && rpcRequestBody.get(DataConstants.PERSISTENT).asBoolean();
        String additionalInfo = JacksonUtil.toString(rpcRequestBody.get(DataConstants.ADDITIONAL_INFO));
        Integer retries = rpcRequestBody.has(DataConstants.RETRIES) ? rpcRequestBody.get(DataConstants.RETRIES).asInt() : null;
        accessValidator.validate(currentUser, Operation.RPC_CALL, deviceId, new HttpValidationCallback(response, new FutureCallback<>() {

            @Override
            public void onSuccess(@Nullable DeferredResult<ResponseEntity> result) {
                ToDeviceRpcRequest rpcRequest = new ToDeviceRpcRequest(rpcRequestUUID, tenantId, deviceId, oneWay, expTime, body, persisted, retries, additionalInfo);
                deviceRpcService.processRestApiRpcRequest(rpcRequest, fromDeviceRpcResponse -> reply(new LocalRequestMetaData(rpcRequest, currentUser, result), fromDeviceRpcResponse, timeoutStatus, noActiveConnectionStatus), currentUser);
            }

            @Override
            public void onFailure(Throwable e) {
                ResponseEntity entity;
                if (e instanceof ToErrorResponseEntity) {
                    entity = ((ToErrorResponseEntity) e).toErrorResponseEntity();
                } else {
                    entity = new ResponseEntity(HttpStatus.UNAUTHORIZED);
                }
                logRpcCall(currentUser, deviceId, body, oneWay, Optional.empty(), e);
                response.setResult(entity);
            }
        }));
        return response;
    } catch (IllegalArgumentException ioe) {
        throw new ThingsboardException("Invalid request body", ioe, ThingsboardErrorCode.BAD_REQUEST_PARAMS);
    }
}
Also used : JsonNode(com.fasterxml.jackson.databind.JsonNode) TenantId(org.thingsboard.server.common.data.id.TenantId) ToErrorResponseEntity(org.thingsboard.server.service.telemetry.exception.ToErrorResponseEntity) ResponseEntity(org.springframework.http.ResponseEntity) SecurityUser(org.thingsboard.server.service.security.model.SecurityUser) LocalRequestMetaData(org.thingsboard.server.service.rpc.LocalRequestMetaData) ToDeviceRpcRequest(org.thingsboard.server.common.msg.rpc.ToDeviceRpcRequest) ToDeviceRpcRequestBody(org.thingsboard.server.common.data.rpc.ToDeviceRpcRequestBody) ThingsboardException(org.thingsboard.server.common.data.exception.ThingsboardException) UUID(java.util.UUID) FutureCallback(com.google.common.util.concurrent.FutureCallback) Nullable(javax.annotation.Nullable) ToErrorResponseEntity(org.thingsboard.server.service.telemetry.exception.ToErrorResponseEntity) DeferredResult(org.springframework.web.context.request.async.DeferredResult)

Aggregations

ToDeviceRpcRequestBody (org.thingsboard.server.common.data.rpc.ToDeviceRpcRequestBody)3 ToDeviceRpcRequest (org.thingsboard.server.common.msg.rpc.ToDeviceRpcRequest)3 UUID (java.util.UUID)2 JsonNode (com.fasterxml.jackson.databind.JsonNode)1 FutureCallback (com.google.common.util.concurrent.FutureCallback)1 Nullable (javax.annotation.Nullable)1 ResponseEntity (org.springframework.http.ResponseEntity)1 DeferredResult (org.springframework.web.context.request.async.DeferredResult)1 ThingsboardException (org.thingsboard.server.common.data.exception.ThingsboardException)1 TenantId (org.thingsboard.server.common.data.id.TenantId)1 LocalRequestMetaData (org.thingsboard.server.service.rpc.LocalRequestMetaData)1 SecurityUser (org.thingsboard.server.service.security.model.SecurityUser)1 ToErrorResponseEntity (org.thingsboard.server.service.telemetry.exception.ToErrorResponseEntity)1