All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.google.web.bindery.requestfactory.shared.impl.AbstractRequestContext Maven / Gradle / Ivy

There is a newer version: 2.10.0
Show newest version
/*
 * Copyright 2010 Google Inc.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.google.web.bindery.requestfactory.shared.impl;

import static com.google.web.bindery.requestfactory.shared.impl.BaseProxyCategory.stableId;
import static com.google.web.bindery.requestfactory.shared.impl.Constants.REQUEST_CONTEXT_STATE;
import static com.google.web.bindery.requestfactory.shared.impl.Constants.STABLE_ID;

import com.google.web.bindery.autobean.shared.AutoBean;
import com.google.web.bindery.autobean.shared.AutoBeanCodex;
import com.google.web.bindery.autobean.shared.AutoBeanFactory;
import com.google.web.bindery.autobean.shared.AutoBeanUtils;
import com.google.web.bindery.autobean.shared.AutoBeanVisitor;
import com.google.web.bindery.autobean.shared.Splittable;
import com.google.web.bindery.autobean.shared.ValueCodex;
import com.google.web.bindery.autobean.shared.impl.AbstractAutoBean;
import com.google.web.bindery.autobean.shared.impl.EnumMap;
import com.google.web.bindery.autobean.shared.impl.StringQuoter;
import com.google.web.bindery.event.shared.UmbrellaException;
import com.google.web.bindery.requestfactory.shared.BaseProxy;
import com.google.web.bindery.requestfactory.shared.EntityProxy;
import com.google.web.bindery.requestfactory.shared.EntityProxyChange;
import com.google.web.bindery.requestfactory.shared.EntityProxyId;
import com.google.web.bindery.requestfactory.shared.FanoutReceiver;
import com.google.web.bindery.requestfactory.shared.Receiver;
import com.google.web.bindery.requestfactory.shared.Request;
import com.google.web.bindery.requestfactory.shared.RequestContext;
import com.google.web.bindery.requestfactory.shared.RequestTransport.TransportReceiver;
import com.google.web.bindery.requestfactory.shared.ServerFailure;
import com.google.web.bindery.requestfactory.shared.WriteOperation;
import com.google.web.bindery.requestfactory.shared.impl.posers.DatePoser;
import com.google.web.bindery.requestfactory.shared.messages.IdMessage;
import com.google.web.bindery.requestfactory.shared.messages.IdMessage.Strength;
import com.google.web.bindery.requestfactory.shared.messages.InvocationMessage;
import com.google.web.bindery.requestfactory.shared.messages.JsonRpcRequest;
import com.google.web.bindery.requestfactory.shared.messages.MessageFactory;
import com.google.web.bindery.requestfactory.shared.messages.OperationMessage;
import com.google.web.bindery.requestfactory.shared.messages.RequestMessage;
import com.google.web.bindery.requestfactory.shared.messages.ResponseMessage;
import com.google.web.bindery.requestfactory.shared.messages.ServerFailureMessage;
import com.google.web.bindery.requestfactory.shared.messages.ViolationMessage;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.validation.ConstraintViolation;
import javax.validation.Path;
import javax.validation.metadata.ConstraintDescriptor;

/**
 * Base implementations for RequestContext services.
 */
public abstract class AbstractRequestContext implements RequestContext, EntityCodex.EntitySource {
  /**
   * Allows the payload dialect to be injected into the AbstractRequestContext
   * without the caller needing to be concerned with how the implementation
   * object is instantiated.
   */
  public enum Dialect {
    STANDARD {
      @Override
      DialectImpl create(AbstractRequestContext context) {
        return context.new StandardPayloadDialect();
      }
    },
    JSON_RPC {
      @Override
      DialectImpl create(AbstractRequestContext context) {
        return context.new JsonRpcPayloadDialect();
      }
    };
    abstract DialectImpl create(AbstractRequestContext context);
  }

  /**
   * Encapsulates all state contained by the AbstractRequestContext.
   */
  protected static class State {
    /**
     * Supports the case where chained contexts are used and a response comes
     * back from the server with a proxy type not reachable from the canonical
     * context.
     */
    public Set appendedContexts;
    public final AbstractRequestContext canonical;
    public final DialectImpl dialect;
    public FanoutReceiver fanout;
    /**
     * When {@code true} the {@link AbstractRequestContext#fire()} method will
     * be a no-op.
     */
    public boolean fireDisabled;
    public final List> invocations = new ArrayList>();

    public boolean locked;
    /**
     * A map of all EntityProxies that the RequestContext has interacted with.
     * Objects are placed into this map by being returned from {@link #create},
     * passed into {@link #edit}, or used as an invocation argument.
     */
    public final Map, AutoBean> editedProxies =
        new LinkedHashMap, AutoBean>();
    /**
     * A map that contains the canonical instance of an entity to return in the
     * return graph, since this is built from scratch.
     */
    public final Map, AutoBean> returnedProxies =
        new HashMap, AutoBean>();

    public final AbstractRequestFactory requestFactory;

    /**
     * A map that allows us to handle the case where the server has sent back an
     * unpersisted entity. Because we assume that the server is stateless, the
     * client will need to swap out the request-local ids with a regular
     * client-allocated id.
     */
    public final Map> syntheticIds =
        new HashMap>();

    public State(AbstractRequestFactory requestFactory, DialectImpl dialect,
        AbstractRequestContext canonical) {
      this.requestFactory = requestFactory;
      this.canonical = canonical;
      this.dialect = dialect;
    }

    public void addContext(AbstractRequestContext ctx) {
      if (appendedContexts == null) {
        appendedContexts = Collections.singleton(ctx);
      } else {
        if (appendedContexts.size() == 1) {
          appendedContexts = new LinkedHashSet(appendedContexts);
        }
        appendedContexts.add(ctx);
      }
    }

    public AbstractRequestContext getCanonicalContext() {
      return canonical;
    }

    public boolean isClean() {
      return editedProxies.isEmpty() && invocations.isEmpty() && !locked
          && returnedProxies.isEmpty() && syntheticIds.isEmpty();
    }

    public boolean isCompatible(State state) {
      // Object comparison intentional
      return requestFactory == state.requestFactory
          && dialect.getClass().equals(state.dialect.getClass());
    }
  }

  interface DialectImpl {

    void addInvocation(AbstractRequest request);

    String makePayload();

    void processPayload(Receiver receiver, String payload);
  }

  class JsonRpcPayloadDialect implements DialectImpl {
    /**
     * Called by generated subclasses to enqueue a method invocation.
     */
    public void addInvocation(AbstractRequest request) {
      /*
       * TODO(bobv): Support for multiple invocations per request needs to be
       * ironed out. Once this is done, addInvocation() can be removed from the
       * DialectImpl interface and restored to to AbstractRequestContext.
       */
      if (!state.invocations.isEmpty()) {
        throw new RuntimeException("Only one invocation per request, pending backend support");
      }
      state.invocations.add(request);
      for (Object arg : request.getRequestData().getOrderedParameters()) {
        retainArg(arg);
      }
    }

    public String makePayload() {
      RequestData data = state.invocations.get(0).getRequestData();

      AutoBean bean = MessageFactoryHolder.FACTORY.jsonRpcRequest();
      JsonRpcRequest request = bean.as();

      request.setVersion("2.0");
      request.setApiVersion(data.getApiVersion());
      request.setId(payloadId++);

      Map params = new HashMap();
      for (Map.Entry entry : data.getNamedParameters().entrySet()) {
        Object obj = entry.getValue();
        Splittable value = encode(obj);
        params.put(entry.getKey(), value);
      }
      if (data.getRequestResource() != null) {
        params.put("resource", encode(data.getRequestResource()));
      }
      request.setParams(params);
      request.setMethod(data.getOperation());

      return AutoBeanCodex.encode(bean).getPayload();
    }

    public void processPayload(Receiver receiver, String payload) {
      Splittable raw = StringQuoter.split(payload);

      @SuppressWarnings("unchecked")
      Receiver callback = (Receiver) state.invocations.get(0).getReceiver();

      if (!raw.isNull("error")) {
        Splittable error = raw.get("error");
        ServerFailure failure =
            new ServerFailure(error.get("message").asString(), error.get("code").asString(),
                payload, true);
        fail(receiver, failure);
        return;
      }

      Splittable result = raw.get("result");
      @SuppressWarnings("unchecked")
      Class target =
          (Class) state.invocations.get(0).getRequestData().getReturnType();

      SimpleProxyId id = getRequestFactory().allocateId(target);
      AutoBean bean = createProxy(target, id, true);
      // XXX expose this as a proper API
      ((AbstractAutoBean) bean).setData(result);
      // AutoBeanCodex.decodeInto(result, bean);

      if (callback != null) {
        callback.onSuccess(bean.as());
      }
      if (receiver != null) {
        receiver.onSuccess(null);
      }
    }

    Splittable encode(Object obj) {
      Splittable value;
      if (obj == null) {
        return Splittable.NULL;
      } else if (obj.getClass().isEnum() && getAutoBeanFactory() instanceof EnumMap) {
        value = ValueCodex.encode(((EnumMap) getAutoBeanFactory()).getToken((Enum) obj));
      } else if (ValueCodex.canDecode(obj.getClass())) {
        value = ValueCodex.encode(obj);
      } else {
        // XXX user-provided implementation of interface?
        value = AutoBeanCodex.encode(AutoBeanUtils.getAutoBean(obj));
      }
      return value;
    }
  }

  class StandardPayloadDialect implements DialectImpl {

    /**
     * Called by generated subclasses to enqueue a method invocation.
     */
    public void addInvocation(AbstractRequest request) {
      state.invocations.add(request);
      for (Object arg : request.getRequestData().getOrderedParameters()) {
        retainArg(arg);
      }
    }

    /**
     * Assemble all of the state that has been accumulated in this context. This
     * includes:
     * 
    *
  • Diffs accumulated on objects passed to {@link #edit}. *
  • Invocations accumulated as Request subtypes passed to * {@link #addInvocation}. *
*/ public String makePayload() { // Get the factory from the runtime-specific holder. MessageFactory f = MessageFactoryHolder.FACTORY; List operations = makePayloadOperations(); List invocationMessages = makePayloadInvocations(); // Create the outer envelope message AutoBean bean = f.request(); RequestMessage requestMessage = bean.as(); requestMessage.setRequestFactory(getRequestFactory().getFactoryTypeToken()); if (!invocationMessages.isEmpty()) { requestMessage.setInvocations(invocationMessages); } if (!operations.isEmpty()) { requestMessage.setOperations(operations); } return AutoBeanCodex.encode(bean).getPayload(); } public void processPayload(final Receiver receiver, String payload) { ResponseMessage response = AutoBeanCodex.decode(MessageFactoryHolder.FACTORY, ResponseMessage.class, payload).as(); if (response.getGeneralFailure() != null) { ServerFailureMessage failure = response.getGeneralFailure(); ServerFailure fail = new ServerFailure(failure.getMessage(), failure.getExceptionType(), failure .getStackTrace(), failure.isFatal()); fail(receiver, fail); return; } // Process violations and then stop if (response.getViolations() != null) { Set> errors = new HashSet>(); for (ViolationMessage message : response.getViolations()) { errors.add(new MyConstraintViolation(message)); } violation(receiver, errors); return; } // Process operations processReturnOperations(response); // Send return values Set causes = null; for (int i = 0, j = state.invocations.size(); i < j; i++) { try { if (response.getStatusCodes().get(i)) { state.invocations.get(i).onSuccess(response.getInvocationResults().get(i)); } else { ServerFailureMessage failure = AutoBeanCodex.decode(MessageFactoryHolder.FACTORY, ServerFailureMessage.class, response.getInvocationResults().get(i)).as(); state.invocations.get(i).onFail( new ServerFailure(failure.getMessage(), failure.getExceptionType(), failure .getStackTrace(), failure.isFatal())); } } catch (Throwable t) { if (causes == null) { causes = new HashSet(); } causes.add(t); } } if (receiver != null) { try { receiver.onSuccess(null); } catch (Throwable t) { if (causes == null) { causes = new HashSet(); } causes.add(t); } } // After success, shut down the context state.editedProxies.clear(); state.invocations.clear(); state.returnedProxies.clear(); if (causes != null) { throw new UmbrellaException(causes); } } } private class MyConstraintViolation implements ConstraintViolation { private final BaseProxy leafBean; private final String messageTemplate; private final String message; private final String path; private final BaseProxy rootBean; private final Class rootBeanClass; public MyConstraintViolation(ViolationMessage msg) { AutoBean leafProxy = findEditedProxy(msg.getLeafBeanId()); leafBean = leafProxy == null ? null : leafProxy.as(); message = msg.getMessage(); messageTemplate = msg.getMessageTemplate(); path = msg.getPath(); AutoBean rootProxy = findEditedProxy(msg.getRootBeanId()); rootBeanClass = rootProxy.getType(); rootBean = rootProxy.as(); } public ConstraintDescriptor getConstraintDescriptor() { return null; } public Object getInvalidValue() { return null; } public Object getLeafBean() { return leafBean; } public String getMessage() { return message; } public String getMessageTemplate() { return messageTemplate; } public Path getPropertyPath() { return new Path() { public Iterator iterator() { return Collections. emptyList().iterator(); } @Override public String toString() { return path; } }; } public BaseProxy getRootBean() { return rootBean; } @SuppressWarnings("unchecked") public Class getRootBeanClass() { return (Class) rootBeanClass; } private AutoBean findEditedProxy(IdMessage idMessage) { // Support violations for value objects. SimpleProxyId rootId = getId(idMessage); // The stub is empty, since we don't process any OperationMessages AutoBean stub = getProxyForReturnPayloadGraph(rootId); // So pick up the instance that we just sent to the server return state.editedProxies.get(BaseProxyCategory.stableId(stub)); } } private static final WriteOperation[] DELETE_ONLY = {WriteOperation.DELETE}; private static final WriteOperation[] PERSIST_AND_UPDATE = { WriteOperation.PERSIST, WriteOperation.UPDATE}; private static final WriteOperation[] UPDATE_ONLY = {WriteOperation.UPDATE}; private static int payloadId = 100; private State state; protected AbstractRequestContext(AbstractRequestFactory factory, Dialect dialect) { setState(new State(factory, dialect.create(this), this)); } public T append(T other) { AbstractRequestContext child = (AbstractRequestContext) other; if (!state.isCompatible(child.state)) { throw new IllegalStateException(getClass().getName() + " and " + child.getClass().getName() + " are not compatible"); } if (!child.state.isClean()) { throw new IllegalStateException("The provided RequestContext has been changed"); } child.setState(state); return other; } /** * Create a new object, with an ephemeral id. */ public T create(Class clazz) { checkLocked(); SimpleProxyId id = state.requestFactory.allocateId(clazz); AutoBean created = createProxy(clazz, id, false); return takeOwnership(created); } public T edit(T object) { return editProxy(object); } /** * Take ownership of a proxy instance and make it editable. */ public T editProxy(T object) { AutoBean bean = checkStreamsNotCrossed(object); checkLocked(); @SuppressWarnings("unchecked") AutoBean previouslySeen = (AutoBean) state.editedProxies.get(BaseProxyCategory.stableId(bean)); if (previouslySeen != null && !previouslySeen.isFrozen()) { /* * If we've seen the object before, it might be because it was passed in * as a method argument. This does not guarantee its mutability, so check * that here before returning the cached object. */ return previouslySeen.as(); } // Create editable copies AutoBean parent = bean; bean = cloneBeanAndCollections(bean); bean.setTag(Constants.PARENT_OBJECT, parent); return bean.as(); } @Override public

Request

find(final EntityProxyId

proxyId) { return new AbstractRequest

(this) { { requestContext.addInvocation(this); } @Override protected RequestData makeRequestData() { // This method is normally generated, hence the ugly constructor return new RequestData(Constants.FIND_METHOD_OPERATION, new Object[] {proxyId}, propertyRefs, proxyId.getProxyClass(), null); } }; } /** * Make sure there's a default receiver so errors don't get dropped. This * behavior should be revisited when chaining is supported, depending on * whether or not chained invocations can fail independently. */ public void fire() { boolean needsReceiver = true; for (AbstractRequest request : state.invocations) { if (request.hasReceiver()) { needsReceiver = false; break; } } if (needsReceiver) { doFire(new Receiver() { @Override public void onSuccess(Void response) { // Don't care } }); } else { doFire(null); } } public void fire(final Receiver receiver) { if (receiver == null) { throw new IllegalArgumentException(); } doFire(receiver); } /** * EntityCodex support. */ public AutoBean getBeanForPayload(Splittable serializedProxyId) { IdMessage ref = AutoBeanCodex.decode(MessageFactoryHolder.FACTORY, IdMessage.class, serializedProxyId).as(); @SuppressWarnings("unchecked") SimpleProxyId id = (SimpleProxyId) getId(ref); return getProxyForReturnPayloadGraph(id); } public AbstractRequestFactory getRequestFactory() { return state.requestFactory; } /** * EntityCodex support. */ public Splittable getSerializedProxyId(SimpleProxyId stableId) { AutoBean bean = MessageFactoryHolder.FACTORY.id(); IdMessage ref = bean.as(); ref.setServerId(stableId.getServerId()); ref.setTypeToken(getRequestFactory().getTypeToken(stableId.getProxyClass())); if (stableId.isSynthetic()) { ref.setStrength(Strength.SYNTHETIC); ref.setSyntheticId(stableId.getSyntheticId()); } else if (stableId.isEphemeral()) { ref.setStrength(Strength.EPHEMERAL); ref.setClientId(stableId.getClientId()); } return AutoBeanCodex.encode(bean); } public boolean isChanged() { /* * NB: Don't use the presence of ephemeral objects for this test. * * Diff the objects until one is found to be different. It's not just a * simple flag-check because of the possibility of "unmaking" a change, per * the JavaDoc. */ for (AutoBean bean : state.editedProxies.values()) { AutoBean previous = bean.getTag(Constants.PARENT_OBJECT); if (previous == null) { // Compare to empty object Class proxyClass = stableId(bean).getProxyClass(); previous = getAutoBeanFactory().create(proxyClass); } if (!AutoBeanUtils.diff(previous, bean).isEmpty()) { return true; } } return false; } /** * EntityCodex support. */ public boolean isEntityType(Class clazz) { return state.requestFactory.isEntityType(clazz); } public boolean isLocked() { return state.locked; } /** * EntityCodex support. */ public boolean isValueType(Class clazz) { return state.requestFactory.isValueType(clazz); } public void setFireDisabled(boolean disabled) { state.fireDisabled = disabled; } /** * Called by generated subclasses to enqueue a method invocation. */ protected void addInvocation(AbstractRequest request) { state.dialect.addInvocation(request); }; /** * Creates a new proxy with an assigned ID. * * @param clazz The proxy type * @param id The id to be assigned to the new proxy * @param useAppendedContexts if {@code true} use the AutoBeanFactory types * associated with any contexts that have been passed into * {@link #append(RequestContext)}. If {@code false}, this method * will only create proxy types reachable from the implemented * RequestContext interface. * @throws IllegalArgumentException if the requested proxy type cannot be * created */ protected AutoBean createProxy(Class clazz, SimpleProxyId id, boolean useAppendedContexts) { AutoBean created = null; if (useAppendedContexts) { for (AbstractRequestContext ctx : state.appendedContexts) { created = ctx.getAutoBeanFactory().create(clazz); if (created != null) { break; } } } else { created = getAutoBeanFactory().create(clazz); } if (created != null) { created.setTag(STABLE_ID, id); return created; } throw new IllegalArgumentException("Unknown proxy type " + clazz.getName()); } /** * Invoke the appropriate {@code onFailure} callbacks, possibly throwing an * {@link UmbrellaException} if one or more callbacks fails. */ protected void fail(Receiver receiver, ServerFailure failure) { reuse(); failure.setRequestContext(this); Set causes = null; for (AbstractRequest request : new ArrayList>(state.invocations)) { try { request.onFail(failure); } catch (Throwable t) { if (causes == null) { causes = new HashSet(); } causes.add(t); } } if (receiver != null) { try { receiver.onFailure(failure); } catch (Throwable t) { if (causes == null) { causes = new HashSet(); } causes.add(t); } } if (causes != null) { throw new UmbrellaException(causes); } } /** * Returns an AutoBeanFactory that can produce the types reachable only from * this RequestContext. */ protected abstract AutoBeanFactory getAutoBeanFactory(); /** * Invoke the appropriate {@code onViolation} callbacks, possibly throwing an * {@link UmbrellaException} if one or more callbacks fails. */ protected void violation(final Receiver receiver, Set> errors) { reuse(); Set causes = null; for (AbstractRequest request : new ArrayList>(state.invocations)) { try { request.onViolation(errors); } catch (Throwable t) { if (causes == null) { causes = new HashSet(); } causes.add(t); } } if (receiver != null) { try { receiver.onConstraintViolation(errors); } catch (Throwable t) { if (causes == null) { causes = new HashSet(); } causes.add(t); } } if (causes != null) { throw new UmbrellaException(causes); } } /** * Resolves an IdMessage into an SimpleProxyId. */ SimpleProxyId getId(IdMessage op) { if (Strength.SYNTHETIC.equals(op.getStrength())) { return allocateSyntheticId(op.getTypeToken(), op.getSyntheticId()); } return state.requestFactory.getId(op.getTypeToken(), op.getServerId(), op.getClientId()); } /** * Creates or retrieves a new canonical AutoBean to represent the given id in * the returned payload. */ AutoBean getProxyForReturnPayloadGraph(SimpleProxyId id) { @SuppressWarnings("unchecked") AutoBean bean = (AutoBean) state.returnedProxies.get(id); if (bean == null) { Class proxyClass = id.getProxyClass(); bean = createProxy(proxyClass, id, true); state.returnedProxies.put(id, bean); } return bean; } /** * Create a single OperationMessage that encapsulates the state of a proxy * AutoBean. */ AutoBean makeOperationMessage(SimpleProxyId stableId, AutoBean proxyBean, boolean useDelta) { // The OperationMessages describes operations on exactly one entity AutoBean toReturn = MessageFactoryHolder.FACTORY.operation(); OperationMessage operation = toReturn.as(); operation.setTypeToken(state.requestFactory.getTypeToken(stableId.getProxyClass())); // Find the object to compare against AutoBean parent; if (stableId.isEphemeral()) { // Newly-created object, use a blank object to compare against parent = createProxy(stableId.getProxyClass(), stableId, true); // Newly-created objects go into the persist operation bucket operation.setOperation(WriteOperation.PERSIST); // The ephemeral id is passed to the server operation.setClientId(stableId.getClientId()); operation.setStrength(Strength.EPHEMERAL); } else if (stableId.isSynthetic()) { // Newly-created object, use a blank object to compare against parent = createProxy(stableId.getProxyClass(), stableId, true); // Newly-created objects go into the persist operation bucket operation.setOperation(WriteOperation.PERSIST); // The ephemeral id is passed to the server operation.setSyntheticId(stableId.getSyntheticId()); operation.setStrength(Strength.SYNTHETIC); } else { parent = proxyBean.getTag(Constants.PARENT_OBJECT); // Requests involving existing objects use the persisted id operation.setServerId(stableId.getServerId()); operation.setOperation(WriteOperation.UPDATE); } assert !useDelta || parent != null; // Send our version number to the server to cut down on future payloads String version = proxyBean.getTag(Constants.VERSION_PROPERTY_B64); if (version != null) { operation.setVersion(version); } Map diff = Collections.emptyMap(); if (isEntityType(stableId.getProxyClass())) { // Compute what's changed on the client diff = useDelta ? AutoBeanUtils.diff(parent, proxyBean) : AutoBeanUtils .getAllProperties(proxyBean); } else if (isValueType(stableId.getProxyClass())) { // Send everything diff = AutoBeanUtils.getAllProperties(proxyBean); } if (!diff.isEmpty()) { Map propertyMap = new HashMap(); for (Map.Entry entry : diff.entrySet()) { propertyMap.put(entry.getKey(), EntityCodex.encode(this, entry.getValue())); } operation.setPropertyMap(propertyMap); } return toReturn; } /** * Create a new EntityProxy from a snapshot in the return payload. * * @param id the EntityProxyId of the object * @param returnRecord the JSON map containing property/value pairs * @param operations the WriteOperation eventns to broadcast over the EventBus */ Q processReturnOperation(SimpleProxyId id, OperationMessage op, WriteOperation... operations) { AutoBean toMutate = getProxyForReturnPayloadGraph(id); toMutate.setTag(Constants.VERSION_PROPERTY_B64, op.getVersion()); final Map properties = op.getPropertyMap(); if (properties != null) { // Apply updates toMutate.accept(new AutoBeanVisitor() { @Override public boolean visitReferenceProperty(String propertyName, AutoBean value, PropertyContext ctx) { if (ctx.canSet()) { if (properties.containsKey(propertyName)) { Splittable raw = properties.get(propertyName); Class elementType = ctx instanceof CollectionPropertyContext ? ((CollectionPropertyContext) ctx) .getElementType() : null; Object decoded = EntityCodex.decode(AbstractRequestContext.this, ctx.getType(), elementType, raw); ctx.set(decoded); } } return false; } @Override public boolean visitValueProperty(String propertyName, Object value, PropertyContext ctx) { if (ctx.canSet()) { if (properties.containsKey(propertyName)) { Splittable raw = properties.get(propertyName); Object decoded = ValueCodex.decode(ctx.getType(), raw); /* * Hack for Date subtypes, consider generalizing for * "custom serializers" */ if (decoded != null && Date.class.equals(ctx.getType())) { decoded = new DatePoser((Date) decoded); } ctx.set(decoded); } } return false; } }); } // Finished applying updates, freeze the bean makeImmutable(toMutate); Q proxy = toMutate.as(); /* * Notify subscribers if the object differs from when it first came into the * RequestContext. */ if (operations != null && state.requestFactory.isEntityType(id.getProxyClass())) { for (WriteOperation writeOperation : operations) { if (writeOperation.equals(WriteOperation.UPDATE) && !state.requestFactory.hasVersionChanged(id, op.getVersion())) { // No updates if the server reports no change continue; } state.requestFactory.getEventBus().fireEventFromSource( new EntityProxyChange((EntityProxy) proxy, writeOperation), id.getProxyClass()); } } return proxy; } /** * Get-or-create method for synthetic ids. * * @see #syntheticIds */ private SimpleProxyId allocateSyntheticId(String typeToken, int syntheticId) { @SuppressWarnings("unchecked") SimpleProxyId toReturn = (SimpleProxyId) state.syntheticIds.get(syntheticId); if (toReturn == null) { toReturn = state.requestFactory.allocateId(state.requestFactory. getTypeFromToken(typeToken)); state.syntheticIds.put(syntheticId, toReturn); } return toReturn; } private void checkLocked() { if (state.locked) { throw new IllegalStateException("A request is already in progress"); } } /** * This method checks that a proxy object is either immutable, or already * edited by this context. */ private AutoBean checkStreamsNotCrossed(T object) { AutoBean bean = AutoBeanUtils.getAutoBean(object); if (bean == null) { // Unexpected; some kind of foreign implementation? throw new IllegalArgumentException(object.getClass().getName()); } State otherState = bean.getTag(REQUEST_CONTEXT_STATE); if (!bean.isFrozen() && otherState != this.state) { /* * This means something is way off in the weeds. If a bean is editable, * it's supposed to be associated with a RequestContext. */ assert otherState != null : "Unfrozen bean with null RequestContext"; /* * Already editing the object in another context or it would have been in * the editing map. */ throw new IllegalArgumentException("Attempting to edit an EntityProxy" + " previously edited by another RequestContext"); } return bean; } /** * Shallow-clones an autobean and makes duplicates of the collection types. A * regular {@link AutoBean#clone} won't duplicate reference properties. */ private AutoBean cloneBeanAndCollections(final AutoBean toClone) { AutoBean clone = toClone.getFactory().create(toClone.getType()); clone.setTag(STABLE_ID, toClone.getTag(STABLE_ID)); clone.setTag(Constants.VERSION_PROPERTY_B64, toClone.getTag(Constants.VERSION_PROPERTY_B64)); /* * Take ownership here to prevent cycles in value objects from overflowing * the stack. */ takeOwnership(clone); clone.accept(new AutoBeanVisitor() { final Map values = AutoBeanUtils.getAllProperties(toClone); @Override public boolean visitCollectionProperty(String propertyName, AutoBean> value, CollectionPropertyContext ctx) { // javac generics bug value = AutoBeanUtils., Collection> getAutoBean((Collection) values .get(propertyName)); if (value != null) { Collection collection; if (List.class == ctx.getType()) { collection = new ArrayList(); } else if (Set.class == ctx.getType()) { collection = new HashSet(); } else { // Should not get here if the validator works correctly throw new IllegalArgumentException(ctx.getType().getName()); } if (isValueType(ctx.getElementType()) || isEntityType(ctx.getElementType())) { /* * Proxies must be edited up-front so that the elements in the * collection have stable identity. */ for (Object o : value.as()) { if (o == null) { collection.add(null); } else { collection.add(editProxy((BaseProxy) o)); } } } else { // For simple values, just copy the values collection.addAll(value.as()); } ctx.set(collection); } return false; } @Override public boolean visitReferenceProperty(String propertyName, AutoBean value, PropertyContext ctx) { value = AutoBeanUtils.getAutoBean(values.get(propertyName)); if (value != null) { if (isValueType(ctx.getType()) || isEntityType(ctx.getType())) { /* * Value proxies must be cloned upfront, since the value is replaced * outright. */ @SuppressWarnings("unchecked") AutoBean valueBean = (AutoBean) value; ctx.set(editProxy(valueBean.as())); } else { ctx.set(value.as()); } } return false; } @Override public boolean visitValueProperty(String propertyName, Object value, PropertyContext ctx) { ctx.set(values.get(propertyName)); return false; } }); return clone; } private void doFire(Receiver receiver) { final Receiver finalReceiver; if (state.fireDisabled) { if (receiver != null) { if (state.fanout == null) { state.fanout = new FanoutReceiver(); } state.fanout.add(receiver); } return; } else if (state.fanout != null) { if (receiver != null) { state.fanout.add(receiver); } finalReceiver = state.fanout; } else { finalReceiver = receiver; } checkLocked(); state.locked = true; freezeEntities(true); String payload = state.dialect.makePayload(); state.requestFactory.getRequestTransport().send(payload, new TransportReceiver() { public void onTransportFailure(ServerFailure failure) { fail(finalReceiver, failure); } public void onTransportSuccess(String payload) { state.dialect.processPayload(finalReceiver, payload); } }); } /** * Set the frozen status of all EntityProxies owned by this context. */ private void freezeEntities(boolean frozen) { for (AutoBean bean : state.editedProxies.values()) { bean.setFrozen(frozen); } } /** * Make an EntityProxy immutable. */ private void makeImmutable(final AutoBean toMutate) { // Always diff'ed against itself, producing a no-op toMutate.setTag(Constants.PARENT_OBJECT, toMutate); // Act with entity-identity semantics toMutate.setTag(REQUEST_CONTEXT_STATE, null); toMutate.setFrozen(true); } /** * Create an InvocationMessage for each remote method call being made by the * context. */ private List makePayloadInvocations() { MessageFactory f = MessageFactoryHolder.FACTORY; List invocationMessages = new ArrayList(); for (AbstractRequest invocation : state.invocations) { // RequestData is produced by the generated subclass RequestData data = invocation.getRequestData(); InvocationMessage message = f.invocation().as(); // Operation; essentially a method descriptor message.setOperation(data.getOperation()); // The arguments to the with() calls Set refsToSend = data.getPropertyRefs(); if (!refsToSend.isEmpty()) { message.setPropertyRefs(refsToSend); } // Parameter values or references List parameters = new ArrayList(data.getOrderedParameters().length); for (Object param : data.getOrderedParameters()) { parameters.add(EntityCodex.encode(this, param)); } if (!parameters.isEmpty()) { message.setParameters(parameters); } invocationMessages.add(message); } return invocationMessages; } /** * Compute deltas for each entity seen by the context. */ private List makePayloadOperations() { List operations = new ArrayList(); for (AutoBean currentView : state.editedProxies.values()) { OperationMessage operation = makeOperationMessage(BaseProxyCategory.stableId(currentView), currentView, true).as(); operations.add(operation); } return operations; } /** * Process an array of OperationMessages. */ private void processReturnOperations(ResponseMessage response) { List ops = response.getOperations(); // If there are no observable effects, this will be null if (ops == null) { return; } for (OperationMessage op : ops) { SimpleProxyId id = getId(op); WriteOperation[] toPropagate = null; // May be null if the server is returning an unpersisted object WriteOperation effect = op.getOperation(); if (effect != null) { switch (effect) { case DELETE: toPropagate = DELETE_ONLY; break; case PERSIST: toPropagate = PERSIST_AND_UPDATE; break; case UPDATE: toPropagate = UPDATE_ONLY; break; default: // Should never reach here throw new RuntimeException(effect.toString()); } } processReturnOperation(id, op, toPropagate); } } /** * Ensures that any method arguments are retained in the context's sphere of * influence. */ private void retainArg(Object arg) { if (arg instanceof Iterable) { for (Object o : (Iterable) arg) { retainArg(o); } } else if (arg instanceof BaseProxy) { // Calling edit will validate and set up the tracking we need edit((BaseProxy) arg); } } /** * Returns the requests that were dequeued as part of reusing the context. */ private void reuse() { freezeEntities(false); state.locked = false; } private void setState(State state) { this.state = state; state.addContext(this); } /** * Make the EnityProxy bean edited and owned by this RequestContext. */ private T takeOwnership(AutoBean bean) { state.editedProxies.put(stableId(bean), bean); bean.setTag(REQUEST_CONTEXT_STATE, this.state); return bean.as(); } }