java/src/main/com/google/appengine/api/datastore/EntityTranslator.java

   1 // Copyright 2007 Google Inc. All rights reserved.
   2
   3 package com.google.appengine.api.datastore;
   4
   5 import com.google.common.collect.Maps;
   6 import com.google.storage.onestore.v3.OnestoreEntity.EntityProto;
   7 import com.google.storage.onestore.v3.OnestoreEntity.Path;
   8 import com.google.storage.onestore.v3.OnestoreEntity.Reference;
   9
  10 import java.util.Collection;
  11 import java.util.Map;
  12
  13 /**
  14  * {@code EntityTranslator} contains the logic to translate an {@code
  15  * Entity} into the protocol buffers that are used to pass it to the
  16  * implementation of the API.
  17  *
  18  */
  19 public class EntityTranslator {
  20
  21   public static Entity createFromPb(EntityProto proto, Collection<Projection> projections) {
  22     Key key = KeyTranslator.createFromPb(proto.getKey());
  23
  24     Entity entity = new Entity(key);
  25     Map<String, Object> values = Maps.newHashMap();
  26     DataTypeTranslator.extractPropertiesFromPb(proto, values);
  27     for (Projection projection : projections) {
  28       entity.setProperty(projection.getName(), projection.getValue(values));
  29     }
  30     return entity;
  31   }
  32
  33   public static Entity createFromPb(EntityProto proto) {
  34     Key key = KeyTranslator.createFromPb(proto.getKey());
  35
  36     Entity entity = new Entity(key);
  37     DataTypeTranslator.extractPropertiesFromPb(proto, entity.getPropertyMap());
  38     return entity;
  39   }
  40
  41   public static Entity createFromPbBytes(byte[] pbBytes) {
  42     EntityProto proto = new EntityProto();
  43     proto.mergeFrom(pbBytes);
  44     return createFromPb(proto);
  45   }
  46
  47   public static EntityProto convertToPb(Entity entity) {
  48     Reference reference = KeyTranslator.convertToPb(entity.getKey());
  49
  50     EntityProto proto = new EntityProto();
  51     proto.setKey(reference);
  52
  53     Path entityGroup = proto.getMutableEntityGroup();
  54     Key key = entity.getKey();
  55     if (key.isComplete()) {
  56       entityGroup.addElement(reference.getPath().elements().get(0));
  57     }
  58
  59     DataTypeTranslator.addPropertiesToPb(entity.getPropertyMap(), proto);
  60     return proto;
  61   }
  62
  63   private EntityTranslator() {
  64   }
  65 }