use of io.vertx.core.Future in project hono by eclipse.
the class FileBasedTenantServiceTest method testDoStartCreatesFile.
/**
* Verifies that the tenant service creates a file for persisting tenants
* data if it does not exist yet during startup.
*
* @param ctx The vert.x context.
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testDoStartCreatesFile(final TestContext ctx) {
// GIVEN a tenant service configured to persist data to a not yet existing file
props.setSaveToFile(true);
props.setFilename(FILE_NAME);
when(fileSystem.existsBlocking(FILE_NAME)).thenReturn(Boolean.FALSE);
doAnswer(invocation -> {
final Handler handler = invocation.getArgument(1);
handler.handle(Future.succeededFuture());
return null;
}).when(fileSystem).createFile(eq(props.getFilename()), any(Handler.class));
doAnswer(invocation -> {
final Handler handler = invocation.getArgument(1);
handler.handle(Future.failedFuture("malformed file"));
return null;
}).when(fileSystem).readFile(eq(props.getFilename()), any(Handler.class));
// WHEN starting the service
final Async startup = ctx.async();
final Future<Void> startupTracker = Future.future();
startupTracker.setHandler(ctx.asyncAssertSuccess(started -> {
startup.complete();
}));
svc.doStart(startupTracker);
// THEN the file gets created
startup.await();
verify(fileSystem).createFile(eq(FILE_NAME), any(Handler.class));
}
use of io.vertx.core.Future in project hono by eclipse.
the class BaseCredentialsService method processRemoveRequest.
private Future<EventBusMessage> processRemoveRequest(final EventBusMessage request) {
final String tenantId = request.getTenant();
final JsonObject payload = request.getJsonPayload();
if (tenantId == null || payload == null) {
return Future.failedFuture(new ClientErrorException(HttpURLConnection.HTTP_BAD_REQUEST));
} else {
final String type = getTypesafeValueForField(payload, CredentialsConstants.FIELD_TYPE);
final String authId = getTypesafeValueForField(payload, CredentialsConstants.FIELD_AUTH_ID);
final String deviceId = getTypesafeValueForField(payload, CredentialsConstants.FIELD_PAYLOAD_DEVICE_ID);
if (type == null) {
log.debug("remove credentials request does not contain mandatory type parameter");
return Future.failedFuture(new ClientErrorException(HttpURLConnection.HTTP_BAD_REQUEST));
} else if (!type.equals(CredentialsConstants.SPECIFIER_WILDCARD) && authId != null) {
// delete a single credentials instance
log.debug("removing specific credentials [tenant: {}, type: {}, auth-id: {}]", tenantId, type, authId);
final Future<CredentialsResult<JsonObject>> result = Future.future();
remove(tenantId, type, authId, result.completer());
return result.map(res -> {
return request.getResponse(res.getStatus()).setCacheDirective(res.getCacheDirective());
});
} else if (deviceId != null && type.equals(CredentialsConstants.SPECIFIER_WILDCARD)) {
// delete all credentials for device
log.debug("removing all credentials for device [tenant: {}, device-id: {}]", tenantId, deviceId);
final Future<CredentialsResult<JsonObject>> result = Future.future();
removeAll(tenantId, deviceId, result.completer());
return result.map(res -> {
return request.getResponse(res.getStatus()).setDeviceId(deviceId).setCacheDirective(res.getCacheDirective());
});
} else {
log.debug("remove credentials request contains invalid search criteria [type: {}, device-id: {}, auth-id: {}]", type, deviceId, authId);
return Future.failedFuture(new ClientErrorException(HttpURLConnection.HTTP_BAD_REQUEST));
}
}
}
use of io.vertx.core.Future in project hono by eclipse.
the class BaseCredentialsService method processUpdateRequest.
private Future<EventBusMessage> processUpdateRequest(final EventBusMessage request) {
final String tenantId = request.getTenant();
final CredentialsObject payload = Optional.ofNullable(request.getJsonPayload()).map(json -> json.mapTo(CredentialsObject.class)).orElse(null);
if (tenantId == null || payload == null) {
return Future.failedFuture(new ClientErrorException(HttpURLConnection.HTTP_BAD_REQUEST));
} else if (payload.isValid()) {
final Future<CredentialsResult<JsonObject>> result = Future.future();
update(tenantId, JsonObject.mapFrom(payload), result.completer());
return result.map(res -> {
return request.getResponse(res.getStatus()).setDeviceId(payload.getDeviceId()).setCacheDirective(res.getCacheDirective());
});
} else {
return Future.failedFuture(new ClientErrorException(HttpURLConnection.HTTP_BAD_REQUEST));
}
}
use of io.vertx.core.Future in project hono by eclipse.
the class BaseRegistrationService method assertRegistration.
/**
* {@inheritDoc}
* <p>
* Subclasses may override this method in order to implement a more sophisticated approach for asserting registration status, e.g.
* using cached information etc.
*/
@Override
public void assertRegistration(final String tenantId, final String deviceId, final String gatewayId, final Handler<AsyncResult<RegistrationResult>> resultHandler) {
Objects.requireNonNull(tenantId);
Objects.requireNonNull(deviceId);
Objects.requireNonNull(gatewayId);
Objects.requireNonNull(resultHandler);
Future<RegistrationResult> deviceInfoTracker = Future.future();
Future<RegistrationResult> gatewayInfoTracker = Future.future();
getDevice(tenantId, deviceId, deviceInfoTracker.completer());
getDevice(tenantId, gatewayId, gatewayInfoTracker.completer());
CompositeFuture.all(deviceInfoTracker, gatewayInfoTracker).compose(ok -> {
final RegistrationResult deviceResult = deviceInfoTracker.result();
final RegistrationResult gatewayResult = gatewayInfoTracker.result();
if (!isDeviceEnabled(deviceResult)) {
return Future.succeededFuture(RegistrationResult.from(HttpURLConnection.HTTP_NOT_FOUND));
} else if (!isDeviceEnabled(gatewayResult)) {
return Future.succeededFuture(RegistrationResult.from(HttpURLConnection.HTTP_FORBIDDEN));
} else {
final JsonObject deviceData = deviceResult.getPayload().getJsonObject(RegistrationConstants.FIELD_DATA, new JsonObject());
final JsonObject gatewayData = gatewayResult.getPayload().getJsonObject(RegistrationConstants.FIELD_DATA, new JsonObject());
if (isGatewayAuthorized(gatewayId, gatewayData, deviceId, deviceData)) {
return Future.succeededFuture(RegistrationResult.from(HttpURLConnection.HTTP_OK, getAssertionPayload(tenantId, deviceId, deviceData), CacheDirective.maxAgeDirective(assertionFactory.getAssertionLifetime())));
} else {
return Future.succeededFuture(RegistrationResult.from(HttpURLConnection.HTTP_FORBIDDEN));
}
}
}).setHandler(resultHandler);
}
use of io.vertx.core.Future in project hono by eclipse.
the class BaseRegistrationService method processAssertRequest.
private Future<EventBusMessage> processAssertRequest(final EventBusMessage request) {
final String tenantId = request.getTenant();
final String deviceId = request.getDeviceId();
final String gatewayId = request.getGatewayId();
if (tenantId == null || deviceId == null) {
return Future.failedFuture(new ClientErrorException(HttpURLConnection.HTTP_BAD_REQUEST));
} else if (gatewayId == null) {
log.debug("asserting registration of device [{}] with tenant [{}]", deviceId, tenantId);
final Future<RegistrationResult> result = Future.future();
assertRegistration(tenantId, deviceId, result.completer());
return result.map(res -> {
return request.getResponse(res.getStatus()).setDeviceId(deviceId).setJsonPayload(res.getPayload()).setCacheDirective(res.getCacheDirective());
});
} else {
log.debug("asserting registration of device [{}] with tenant [{}] for gateway [{}]", deviceId, tenantId, gatewayId);
final Future<RegistrationResult> result = Future.future();
assertRegistration(tenantId, deviceId, gatewayId, result.completer());
return result.map(res -> {
return request.getResponse(res.getStatus()).setDeviceId(deviceId).setJsonPayload(res.getPayload()).setCacheDirective(res.getCacheDirective());
});
}
}
Aggregations