/** 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; }