@Override public Object getValue() { return entity.getProperty(property); }
protected static <E> List<E> getListProperty(String propertyName, Entity entity) { @SuppressWarnings("unchecked") List<E> list = (List<E>) entity.getProperty(propertyName); return list == null ? new LinkedList<E>() : list; }
public Slot(Entity entity, boolean lazy) { super(entity); filled = (Boolean) entity.getProperty(FILLED_PROPERTY); fillTime = (Date) entity.getProperty(FILL_TIME_PROPERTY); sourceJobKey = (Key) entity.getProperty(SOURCE_JOB_KEY_PROPERTY); waitingOnMeKeys = getListProperty(WAITING_ON_ME_PROPERTY, entity); if (lazy) { serializedVersion = entity.getProperty(VALUE_PROPERTY); } else { value = deserializeValue(entity.getProperty(VALUE_PROPERTY)); } }
public ShardedValue(Entity entity) { super(entity); this.shardId = (Long) entity.getProperty(SHARD_ID_PROPERTY); this.value = ((Blob) entity.getProperty(VALUE_PROPERTY)).getBytes(); }
@Override public Set<String> call() { Set<String> pipelines = new LinkedHashSet<>(); for (Entity entity : preparedQuery.asIterable()) { pipelines.add((String) entity.getProperty(JobRecord.ROOT_JOB_DISPLAY_NAME)); } return pipelines; } });
public FanoutTaskRecord(Entity entity) { super(entity); Blob payloadBlob = (Blob) entity.getProperty(PAYLOAD_PROPERTY); payload = payloadBlob.getBytes(); }
public Barrier(Entity entity) { super(entity); jobKey = (Key) entity.getProperty(JOB_KEY_PROPERTY); type = Type.valueOf((String) entity.getProperty(TYPE_PROPERTY)); released = (Boolean) entity.getProperty(RELEASED_PROPERTY); waitingOnKeys = getListProperty(WAITING_ON_KEYS_PROPERTY, entity); waitingOnGroupSizes = getListProperty(WAITING_ON_GROUP_SIZES_PROPERTY, entity); }
private GcsFileMetadata createGcsFileMetadataFromBlobstore(Entity entity, GcsFilename filename) { return new GcsFileMetadata( filename, GcsFileOptions.getDefaultInstance(), "", (Long) entity.getProperty("size"), (Date) entity.getProperty("creation")); }
static public void copyProperties(Entity src, Entity dest) { for (Map.Entry<String, Object> entry : getPropertyMap(src).entrySet()) { // barf if (entry.getValue() != null && entry.getValue().getClass().getName().equals("com.google.appengine.api.datastore.Entity$UnindexedValue")) { dest.setUnindexedProperty(entry.getKey(), src.getProperty(entry.getKey())); } else { dest.setProperty(entry.getKey(), entry.getValue()); } } }
/** * Translates Google App Engine Datastore entity to Acid House {@code Lock} * entity. * * @param entity oogle App Engine Datastore entity. * @return Acid House {@code Lock} entity. */ public static Lock toLock(Entity entity) { return new Lock(entity.getKey().getName(), (String) entity.getProperty(TRANSACTION_PROPERTY), (Date) entity.getProperty(TIMESTAMP_PROPERTY)); }
/** Deserializes the specified object from a Blob using an {@link ObjectInputStream}. */ private V deserialize(Entity entity) throws IOException { Blob blob = (Blob) entity.getProperty(FIELD_VALUE); return IOUtils.deserialize(blob.getBytes()); }
public ExceptionRecord(Entity entity) { super(entity); Blob serializedExceptionBlob = (Blob) entity.getProperty(EXCEPTION_PROPERTY); byte[] serializedException = serializedExceptionBlob.getBytes(); try { exception = (Throwable) SerializationUtils.deserialize(serializedException); } catch (IOException e) { throw new RuntimeException("Failed to deserialize exception for " + getKey(), e); } }
@Override public Entity update(Entity blockEntity, int recordOffset, @Nullable FieldValue fieldValue) { Blob strings = (Blob) blockEntity.getProperty("ids"); int length = StringPools.size(strings); // since we only ever update when adding a new record, we should be able to simply append to the string pool if(recordOffset != length) { throw new IllegalStateException("length = " + length + ", recordOffset = " + recordOffset); } blockEntity.setProperty("ids", StringPools.appendString(strings, ((TextValue) fieldValue).asString())); return blockEntity; }
private String getServerHostAndPort() throws EntityNotFoundException { String serverName = (String) getTestDataEntity().getProperty("serverName"); long serverPort = (Long) getTestDataEntity().getProperty("serverPort"); return serverName + (serverPort == 80 ? "" : (":" + serverPort)); }
public static JsonNode entityToJson(Entity entity) throws LeanException { ObjectNode json = getObjectMapper().createObjectNode(); json.put("_id", entity.getKey().getId()); json.putPOJO("_kind", entity.getKind()); json.putPOJO("_account", entity.getProperty("_account")); Map<String, Object> props = entity.getProperties(); for (Map.Entry<String, Object> prop : props.entrySet()) { addTypedNode(json, prop.getKey(), prop.getValue()); } return json; }
@Test @InSequence(2) public void testStep2BigAsList() { Query q = new Query(kindName, rootKey); q.addSort("count", Query.SortDirection.DESCENDING); List<Entity> eData = service.prepare(q).asList(fo); assertEquals(bigCount, eData.size()); assertEquals(new Integer(bigCount - 1).longValue(), eData.get(0).getProperty("count")); }
@Test public void testSortPredicates() { Query query = new Query(kindName, rootKey); query.addSort("intData", Query.SortDirection.DESCENDING); List<Entity> es = service.prepare(query).asList(fo); assertEquals((long) 40, es.get(0).getProperty("intData")); List<Query.SortPredicate> qsp = query.getSortPredicates(); assertEquals("check SortPredicate name", "intData", qsp.get(0).getPropertyName()); assertEquals("check SortPredicate direction", Query.SortDirection.DESCENDING, qsp.get(0).getDirection()); }
@Test public void testAllowMultipleGroupTrue() throws Exception { clearData(kindName); clearData(otherKind); writeMultipleGroup(true); Query q = new Query(kindName); Entity e = service.prepare(q).asSingleEntity(); assertEquals("parent", e.getProperty("check")); q = new Query(otherKind); e = service.prepare(q).asSingleEntity(); assertEquals("other", e.getProperty("check")); }
private void checkCount(String statsKind) { FetchOptions fo = FetchOptions.Builder.withDefaults(); Query query = new Query(statsKind); assertTrue(service.prepare(query).countEntities(fo) > 0); for (Entity readRec : service.prepare(query).asIterable()) { assertTrue((Long) readRec.getProperty("count") > 0); } } }
@Test public void testNullPropertyValue() throws Exception { Key parentKey = createQueryBasicsTestParent("testNullPropertyValue"); createEntity("Entry", parentKey) .withProperty("user", null) .store(); Entity entity = service.prepare(new Query("Entry") .setAncestor(parentKey)).asSingleEntity(); assertNull(entity.getProperty("user")); }