private void processDelayedUpdates( final List<EntityLinkDesc> delayedUpdates, int pos, final TransactionItems items, final ODataChangeset changeset) { for (EntityLinkDesc delayedUpdate : delayedUpdates) { pos++; items.put(delayedUpdate.getSource(), pos); final ODataEntity changes = ODataFactory.newEntity(delayedUpdate.getSource().getEntity().getName()); AttachedEntityStatus status = EntityContainerFactory.getContext().entityContext().getStatus(delayedUpdate.getSource()); final URI sourceURI; if (status == AttachedEntityStatus.CHANGED) { sourceURI = URIUtils.getURI( factory.getServiceRoot(), delayedUpdate.getSource().getEntity().getEditLink().toASCIIString()); } else { int sourcePos = items.get(delayedUpdate.getSource()); sourceURI = URI.create("$" + sourcePos); } for (EntityTypeInvocationHandler target : delayedUpdate.getTargets()) { status = EntityContainerFactory.getContext().entityContext().getStatus(target); final URI targetURI; if (status == AttachedEntityStatus.CHANGED) { targetURI = URIUtils.getURI( factory.getServiceRoot(), target.getEntity().getEditLink().toASCIIString()); } else { int targetPos = items.get(target); targetURI = URI.create("$" + targetPos); } changes.addLink( delayedUpdate.getType() == ODataLinkType.ENTITY_NAVIGATION ? ODataFactory.newEntityNavigationLink(delayedUpdate.getSourceName(), targetURI) : ODataFactory.newFeedNavigationLink(delayedUpdate.getSourceName(), targetURI)); LOG.debug( "'{}' from {} to {}", new Object[] {delayedUpdate.getType().name(), sourceURI, targetURI}); } batchUpdate(delayedUpdate.getSource(), sourceURI, changes, changeset); } }
private void batchDelete( final EntityTypeInvocationHandler handler, final ODataEntity entity, final ODataChangeset changeset) { LOG.debug("Delete '{}'", entity.getEditLink()); final ODataDeleteRequest req = ODataCUDRequestFactory.getDeleteRequest( URIUtils.getURI(factory.getServiceRoot(), entity.getEditLink().toASCIIString())); if (StringUtils.isNotBlank(handler.getETag())) { req.setIfMatch(handler.getETag()); } changeset.addRequest(req); }
private void batchUpdate( final EntityTypeInvocationHandler handler, final ODataEntity changes, final ODataChangeset changeset) { LOG.debug("Update '{}'", changes.getEditLink()); final ODataEntityUpdateRequest req = ODataCUDRequestFactory.getEntityUpdateRequest(UpdateType.PATCH, changes); req.setPrefer(ODataHeaderValues.preferReturnContent); if (StringUtils.isNotBlank(handler.getETag())) { req.setIfMatch(handler.getETag()); } changeset.addRequest(req); }
private void batchUpdateMediaResource( final EntityTypeInvocationHandler handler, final URI uri, final InputStream input, final ODataChangeset changeset) { LOG.debug("Update media entity '{}'", uri); final ODataStreamUpdateRequest req = ODataStreamedRequestFactory.getStreamUpdateRequest(uri, input); if (StringUtils.isNotBlank(handler.getETag())) { req.setIfMatch(handler.getETag()); } changeset.addRequest(req); }
private void batchCreate( final EntityTypeInvocationHandler handler, final ODataEntity entity, final ODataChangeset changeset) { LOG.debug("Create '{}'", handler); final ODataURIBuilder uriBuilder = new ODataURIBuilder(factory.getServiceRoot()) .appendEntitySetSegment(handler.getEntitySetName()); changeset.addRequest(ODataCUDRequestFactory.getEntityCreateRequest(uriBuilder.build(), entity)); }
private void batchUpdateMediaEntity( final EntityTypeInvocationHandler handler, final URI uri, final InputStream input, final ODataChangeset changeset) { LOG.debug("Update media entity '{}'", uri); final ODataMediaEntityUpdateRequest req = ODataStreamedRequestFactory.getMediaEntityUpdateRequest(uri, input); req.setContentType( StringUtils.isBlank(handler.getEntity().getMediaContentType()) ? ODataMediaFormat.WILDCARD.toString() : ODataMediaFormat.fromFormat(handler.getEntity().getMediaContentType()).toString()); if (StringUtils.isNotBlank(handler.getETag())) { req.setIfMatch(handler.getETag()); } changeset.addRequest(req); }
private void batch( final EntityTypeInvocationHandler handler, final ODataEntity entity, final ODataChangeset changeset) { switch (EntityContainerFactory.getContext().entityContext().getStatus(handler)) { case NEW: batchCreate(handler, entity, changeset); break; case CHANGED: batchUpdate(handler, entity, changeset); break; case DELETED: batchDelete(handler, entity, changeset); break; default: if (handler.isChanged()) { batchUpdate(handler, entity, changeset); } } }
/** Transactional changes commit. */ @Override public void flush() { final ODataBatchRequest request = ODataBatchRequestFactory.getBatchRequest(factory.getServiceRoot()); final ODataBatchRequest.BatchStreamManager streamManager = request.execute(); final ODataChangeset changeset = streamManager.addChangeset(); final TransactionItems items = new TransactionItems(); final List<EntityLinkDesc> delayedUpdates = new ArrayList<EntityLinkDesc>(); int pos = 0; for (AttachedEntity attachedEntity : EntityContainerFactory.getContext().entityContext()) { final AttachedEntityStatus status = attachedEntity.getStatus(); if (((status != AttachedEntityStatus.ATTACHED && status != AttachedEntityStatus.LINKED) || attachedEntity.getEntity().isChanged()) && !items.contains(attachedEntity.getEntity())) { pos++; pos = processEntityContext(attachedEntity.getEntity(), pos, items, delayedUpdates, changeset); } } processDelayedUpdates(delayedUpdates, pos, items, changeset); final ODataBatchResponse response = streamManager.getResponse(); if (response.getStatusCode() != 202) { throw new IllegalStateException("Operation failed"); } final Iterator<ODataBatchResponseItem> iter = response.getBody(); if (!items.isEmpty()) { if (!iter.hasNext()) { throw new IllegalStateException("Unexpected operation result"); } final ODataBatchResponseItem item = iter.next(); if (!(item instanceof ODataChangesetResponseItem)) { throw new IllegalStateException( "Unexpected batch response item " + item.getClass().getSimpleName()); } final ODataChangesetResponseItem chgres = (ODataChangesetResponseItem) item; for (Integer changesetItemId : items.sortedValues()) { LOG.debug("Expected changeset item {}", changesetItemId); final ODataResponse res = chgres.next(); if (res.getStatusCode() >= 400) { throw new IllegalStateException("Transaction failed: " + res.getStatusMessage()); } final EntityTypeInvocationHandler handler = items.get(changesetItemId); if (handler != null) { if (res instanceof ODataEntityCreateResponse) { LOG.debug("Upgrade created object '{}'", handler); handler.setEntity(((ODataEntityCreateResponse) res).getBody()); } else if (res instanceof ODataEntityUpdateResponse) { LOG.debug("Upgrade updated object '{}'", handler); handler.setEntity(((ODataEntityUpdateResponse) res).getBody()); } } } } EntityContainerFactory.getContext().detachAll(); }
private int processEntityContext( final EntityTypeInvocationHandler handler, int pos, final TransactionItems items, final List<EntityLinkDesc> delayedUpdates, final ODataChangeset changeset) { LOG.debug("Process '{}'", handler); items.put(handler, null); final ODataEntity entity = SerializationUtils.clone(handler.getEntity()); entity.getNavigationLinks().clear(); final AttachedEntityStatus currentStatus = EntityContainerFactory.getContext().entityContext().getStatus(handler); if (AttachedEntityStatus.DELETED != currentStatus) { entity.getProperties().clear(); EngineUtils.addProperties(factory.getMetadata(), handler.getPropertyChanges(), entity); } for (Map.Entry<NavigationProperty, Object> property : handler.getLinkChanges().entrySet()) { final ODataLinkType type = Collection.class.isAssignableFrom(property.getValue().getClass()) ? ODataLinkType.ENTITY_SET_NAVIGATION : ODataLinkType.ENTITY_NAVIGATION; final Set<EntityTypeInvocationHandler> toBeLinked = new HashSet<EntityTypeInvocationHandler>(); final String serviceRoot = factory.getServiceRoot(); for (Object proxy : type == ODataLinkType.ENTITY_SET_NAVIGATION ? (Collection) property.getValue() : Collections.singleton(property.getValue())) { final EntityTypeInvocationHandler target = (EntityTypeInvocationHandler) Proxy.getInvocationHandler(proxy); final AttachedEntityStatus status = EntityContainerFactory.getContext().entityContext().getStatus(target); final URI editLink = target.getEntity().getEditLink(); if ((status == AttachedEntityStatus.ATTACHED || status == AttachedEntityStatus.LINKED) && !target.isChanged()) { entity.addLink( buildNavigationLink( property.getKey().name(), URIUtils.getURI(serviceRoot, editLink.toASCIIString()), type)); } else { if (!items.contains(target)) { pos = processEntityContext(target, pos, items, delayedUpdates, changeset); pos++; } final Integer targetPos = items.get(target); if (targetPos == null) { // schedule update for the current object LOG.debug("Schedule '{}' from '{}' to '{}'", type.name(), handler, target); toBeLinked.add(target); } else if (status == AttachedEntityStatus.CHANGED) { entity.addLink( buildNavigationLink( property.getKey().name(), URIUtils.getURI(serviceRoot, editLink.toASCIIString()), type)); } else { // create the link for the current object LOG.debug("'{}' from '{}' to (${}) '{}'", type.name(), handler, targetPos, target); entity.addLink( buildNavigationLink(property.getKey().name(), URI.create("$" + targetPos), type)); } } } if (!toBeLinked.isEmpty()) { delayedUpdates.add(new EntityLinkDesc(property.getKey().name(), handler, toBeLinked, type)); } } // insert into the batch LOG.debug("{}: Insert '{}' into the batch", pos, handler); batch(handler, entity, changeset); items.put(handler, pos); int startingPos = pos; if (handler.getEntity().isMediaEntity()) { // update media properties if (!handler.getPropertyChanges().isEmpty()) { final URI targetURI = currentStatus == AttachedEntityStatus.NEW ? URI.create("$" + startingPos) : URIUtils.getURI( factory.getServiceRoot(), handler.getEntity().getEditLink().toASCIIString()); batchUpdate(handler, targetURI, entity, changeset); pos++; items.put(handler, pos); } // update media content if (handler.getStreamChanges() != null) { final URI targetURI = currentStatus == AttachedEntityStatus.NEW ? URI.create("$" + startingPos + "/$value") : URIUtils.getURI( factory.getServiceRoot(), handler.getEntity().getEditLink().toASCIIString() + "/$value"); batchUpdateMediaEntity(handler, targetURI, handler.getStreamChanges(), changeset); // update media info (use null key) pos++; items.put(null, pos); } } for (Map.Entry<String, InputStream> streamedChanges : handler.getStreamedPropertyChanges().entrySet()) { final URI targetURI = currentStatus == AttachedEntityStatus.NEW ? URI.create("$" + startingPos) : URIUtils.getURI( factory.getServiceRoot(), EngineUtils.getEditMediaLink(streamedChanges.getKey(), entity).toASCIIString()); batchUpdateMediaResource(handler, targetURI, streamedChanges.getValue(), changeset); // update media info (use null key) pos++; items.put(handler, pos); } return pos; }