use of org.gluu.oxtrust.model.scim2.bulk.BulkResponse in project oxTrust by GluuFederation.
the class BulkWebService method processBulkOperations.
@javax.ws.rs.POST
@Consumes({ MEDIA_TYPE_SCIM_JSON, MediaType.APPLICATION_JSON })
@Produces({ MEDIA_TYPE_SCIM_JSON + UTF8_CHARSET_FRAGMENT, MediaType.APPLICATION_JSON + UTF8_CHARSET_FRAGMENT })
@HeaderParam("Accept")
@DefaultValue(MEDIA_TYPE_SCIM_JSON)
@ProtectedApi
@ApiOperation(value = "Bulk Operations", notes = "Bulk Operations (https://tools.ietf.org/html/rfc7644#section-3.7)", response = BulkResponse.class)
public Response processBulkOperations(@ApiParam(value = "BulkRequest", required = true) BulkRequest request) {
Response response = prepareRequest(request, getValueFromHeaders(httpHeaders, "Content-Length"));
if (response == null) {
log.debug("Executing web service method. processBulkOperations");
int i, errors = 0;
List<BulkOperation> operations = request.getOperations();
List<BulkOperation> responseOperations = new ArrayList<BulkOperation>();
Map<String, String> processedBulkIds = new HashMap<String, String>();
for (i = 0; i < operations.size() && errors < request.getFailOnErrors(); i++) {
BulkOperation operation = operations.get(i);
BulkOperation operationResponse = new BulkOperation();
Response subResponse;
String method = operation.getMethod();
String bulkId = operation.getBulkId();
try {
String path = operation.getPath();
BaseScimWebService service = getWSForPath(path);
String fragment = getFragment(path, service, processedBulkIds);
Verb verb = Verb.valueOf(method);
String data = operation.getDataStr();
if (!verb.equals(DELETE))
data = replaceBulkIds(data, processedBulkIds);
Pair<Response, String> pair = execute(verb, service, data, fragment);
String idCreated = pair.getSecond();
subResponse = pair.getFirst();
int status = subResponse.getStatus();
if (familyOf(status).equals(SUCCESSFUL)) {
if (!verb.equals(DELETE)) {
if (verb.equals(POST)) {
// Update bulkIds
processedBulkIds.put(bulkId, idCreated);
fragment = idCreated;
}
String loc = service.getEndpointUrl() + "/" + fragment;
operationResponse.setLocation(loc);
}
} else {
operationResponse.setResponse(subResponse.getEntity());
errors += familyOf(status).equals(CLIENT_ERROR) || familyOf(status).equals(SERVER_ERROR) ? 1 : 0;
}
subResponse.close();
operationResponse.setStatus(Integer.toString(status));
} catch (Exception e) {
log.error(e.getMessage(), e);
subResponse = getErrorResponse(BAD_REQUEST, ErrorScimType.INVALID_SYNTAX, e.getMessage());
operationResponse.setStatus(Integer.toString(BAD_REQUEST.getStatusCode()));
operationResponse.setResponse(subResponse.getEntity());
errors++;
}
operationResponse.setBulkId(bulkId);
operationResponse.setMethod(method);
responseOperations.add(operationResponse);
log.debug("Operation {} processed with status {}. Method {}, Accumulated errors {}", i + 1, operationResponse.getStatus(), method, errors);
}
try {
BulkResponse bulkResponse = new BulkResponse();
bulkResponse.setOperations(responseOperations);
String json = mapper.writeValueAsString(bulkResponse);
response = Response.ok(json).build();
} catch (Exception e) {
log.error(e.getMessage(), e);
response = getErrorResponse(INTERNAL_SERVER_ERROR, e.getMessage());
}
}
return response;
}
use of org.gluu.oxtrust.model.scim2.bulk.BulkResponse in project oxTrust by GluuFederation.
the class BulkWebService method processBulkOperations.
@POST
@Consumes({ Constants.MEDIA_TYPE_SCIM_JSON, MediaType.APPLICATION_JSON })
@Produces({ Constants.MEDIA_TYPE_SCIM_JSON + "; charset=utf-8", MediaType.APPLICATION_JSON + "; charset=utf-8" })
@HeaderParam("Accept")
@DefaultValue(Constants.MEDIA_TYPE_SCIM_JSON)
@ApiOperation(value = "Bulk Operations", notes = "Bulk Operations (https://tools.ietf.org/html/rfc7644#section-3.7)", response = BulkResponse.class)
public Response processBulkOperations(// @Context HttpServletResponse response,
@HeaderParam("Authorization") String authorization, @HeaderParam("Content-Length") int contentLength, @QueryParam(OxTrustConstants.QUERY_PARAMETER_TEST_MODE_OAUTH2_TOKEN) final String token, @ApiParam(value = "BulkRequest", required = true) BulkRequest bulkRequest) throws Exception {
Response authorizationResponse;
if (jsonConfigurationService.getOxTrustappConfiguration().isScimTestMode()) {
log.info(" ##### SCIM Test Mode is ACTIVE");
authorizationResponse = processTestModeAuthorization(token);
} else {
authorizationResponse = processAuthorization(authorization);
}
if (authorizationResponse != null) {
return authorizationResponse;
}
try {
/*
* J2EContext context = new J2EContext(request, response); int
* removePathLength = "/Bulk".length(); String domain =
* context.getFullRequestURL(); if (domain.endsWith("/")) {
* removePathLength++; } domain = domain.substring(0,
* domain.length() - removePathLength);
*/
log.info("##### Operation count = " + bulkRequest.getOperations().size());
log.info("##### Content-Length = " + contentLength);
if (bulkRequest.getOperations().size() > MAX_BULK_OPERATIONS || contentLength > MAX_BULK_PAYLOAD_SIZE) {
StringBuilder message = new StringBuilder("The size of the bulk operation exceeds the ");
if (bulkRequest.getOperations().size() > MAX_BULK_OPERATIONS && contentLength <= MAX_BULK_PAYLOAD_SIZE) {
message.append("maxOperations (").append(MAX_BULK_OPERATIONS).append(")");
} else if (bulkRequest.getOperations().size() <= MAX_BULK_OPERATIONS && contentLength > MAX_BULK_PAYLOAD_SIZE) {
message.append("maxPayloadSize (").append(MAX_BULK_PAYLOAD_SIZE).append(")");
} else if (bulkRequest.getOperations().size() > MAX_BULK_OPERATIONS && contentLength > MAX_BULK_PAYLOAD_SIZE) {
message.append("maxOperations (").append(MAX_BULK_OPERATIONS).append(") ");
message.append("and ");
message.append("maxPayloadSize (").append(MAX_BULK_PAYLOAD_SIZE).append(")");
}
log.info("Payload Too Large: " + message.toString());
return getErrorResponse(413, message.toString());
}
int failOnErrorsLimit = (bulkRequest.getFailOnErrors() != null) ? bulkRequest.getFailOnErrors() : 0;
int failOnErrorsCount = 0;
List<BulkOperation> bulkOperations = bulkRequest.getOperations();
BulkResponse bulkResponse = new BulkResponse();
Map<String, String> processedBulkIds = new LinkedHashMap<String, String>();
operationsLoop: for (BulkOperation operation : bulkOperations) {
log.info(" Checking operations... ");
if (operation.getPath().startsWith("/Users")) {
// operation = processUserOperation(operation, domain);
operation = processUserOperation(operation, processedBulkIds);
} else if (operation.getPath().startsWith("/Groups")) {
// operation = processGroupOperation(operation, domain);
operation = processGroupOperation(operation, processedBulkIds);
}
bulkResponse.getOperations().add(operation);
// Error handling
String okCode = String.valueOf(Response.Status.OK.getStatusCode());
String createdCode = String.valueOf(Response.Status.CREATED.getStatusCode());
if (!operation.getStatus().equalsIgnoreCase(okCode) && !operation.getStatus().equalsIgnoreCase(createdCode)) {
failOnErrorsCount++;
if ((failOnErrorsLimit > 0) && (failOnErrorsCount >= failOnErrorsLimit)) {
break operationsLoop;
}
}
}
URI location = new URI(appConfiguration.getBaseEndpoint() + "/scim/v2/Bulk");
// Serialize to JSON
ObjectMapper mapper = new ObjectMapper();
mapper.disable(SerializationConfig.Feature.FAIL_ON_EMPTY_BEANS);
SimpleModule customBulkOperationsModule = new SimpleModule("CustomBulkOperationsModule", new Version(1, 0, 0, ""));
// Custom serializers for both User and Group
ListResponseUserSerializer userSerializer = new ListResponseUserSerializer();
ListResponseGroupSerializer groupSerializer = new ListResponseGroupSerializer();
customBulkOperationsModule.addSerializer(User.class, userSerializer);
customBulkOperationsModule.addSerializer(Group.class, groupSerializer);
mapper.registerModule(customBulkOperationsModule);
String json = mapper.writeValueAsString(bulkResponse);
return Response.ok(json).location(location).build();
} catch (Exception ex) {
log.error("Error in processBulkOperations", ex);
ex.printStackTrace();
return getErrorResponse(Response.Status.INTERNAL_SERVER_ERROR, INTERNAL_SERVER_ERROR_MESSAGE);
}
}
Aggregations