private RemoteDatastoreHelper(DatastoreOptions options) { this.options = options; this.datastore = options.getService(); this.namespace = options.getNamespace(); }
static KeyFactory newKeyFactory(DatastoreOptions options) { return new KeyFactory(options.getProjectId(), options.getNamespace()); }
@Test public void testNamespace() throws Exception { assertTrue(options.build().getNamespace().isEmpty()); assertEquals("ns1", options.setNamespace("ns1").build().getNamespace()); }
@Test public void testNewKeyFactory() { DatastoreOptions options = createMock(DatastoreOptions.class); expect(options.getProjectId()).andReturn("ds1").once(); expect(options.getNamespace()).andReturn("ns1").once(); replay(options); KeyFactory keyFactory = DatastoreHelper.newKeyFactory(options); Key key = keyFactory.setKind("k").newKey("bla"); assertEquals("ds1", key.getProjectId()); assertEquals("ns1", key.getNamespace()); assertEquals("k", key.getKind()); assertEquals("bla", key.getName()); verify(options); }
QueryResultsImpl( DatastoreImpl datastore, com.google.datastore.v1.ReadOptions readOptionsPb, Query<T> query) { this.datastore = datastore; this.readOptionsPb = readOptionsPb; this.query = query; queryResultType = query.getType(); com.google.datastore.v1.PartitionId.Builder pbBuilder = com.google.datastore.v1.PartitionId.newBuilder(); pbBuilder.setProjectId(datastore.getOptions().getProjectId()); if (query.getNamespace() != null) { pbBuilder.setNamespaceId(query.getNamespace()); } else if (datastore.getOptions().getNamespace() != null) { pbBuilder.setNamespaceId(datastore.getOptions().getNamespace()); } partitionIdPb = pbBuilder.build(); sendRequest(); if (runQueryResponsePb.getBatch().getSkippedResults() > 0) { cursor = runQueryResponsePb.getBatch().getSkippedCursor(); } else { cursor = mostRecentQueryPb.getStartCursor(); } }
@Test public void testOptions() { LocalDatastoreHelper helper = LocalDatastoreHelper.create(); DatastoreOptions options = helper.getOptions(); assertTrue(options.getProjectId().startsWith(PROJECT_ID_PREFIX)); assertTrue(options.getHost().startsWith("localhost:")); assertSame(NoCredentials.getInstance(), options.getCredentials()); options = helper.getOptions(NAMESPACE); assertTrue(options.getProjectId().startsWith(PROJECT_ID_PREFIX)); assertTrue(options.getHost().startsWith("localhost:")); assertSame(NoCredentials.getInstance(), options.getCredentials()); assertEquals(NAMESPACE, options.getNamespace()); }
@Test public void testToBuilder() throws Exception { DatastoreOptions original = options.setNamespace("ns1").build(); DatastoreOptions copy = original.toBuilder().build(); assertEquals(original.getProjectId(), copy.getProjectId()); assertEquals(original.getNamespace(), copy.getNamespace()); assertEquals(original.getHost(), copy.getHost()); assertEquals(original.getRetrySettings(), copy.getRetrySettings()); assertEquals(original.getCredentials(), copy.getCredentials()); }
private RemoteDatastoreHelper(DatastoreOptions options) { this.options = options; this.datastore = options.getService(); this.namespace = options.getNamespace(); }
/** * Returns the effective namespace. If a namespace was specified using {@link Tenant}, it will be * returned. Otherwise, the namespace of this EntityManager is returned. * * @return the effective namespace. */ String getEffectiveNamespace() { String namespace = Tenant.getNamespace(); if (namespace == null) { namespace = datastore.getOptions().getNamespace(); } return namespace; }
static KeyFactory newKeyFactory(DatastoreOptions options) { return new KeyFactory(options.getProjectId(), options.getNamespace()); }
QueryResultsImpl( DatastoreImpl datastore, com.google.datastore.v1.ReadOptions readOptionsPb, Query<T> query) { this.datastore = datastore; this.readOptionsPb = readOptionsPb; this.query = query; queryResultType = query.getType(); com.google.datastore.v1.PartitionId.Builder pbBuilder = com.google.datastore.v1.PartitionId.newBuilder(); pbBuilder.setProjectId(datastore.getOptions().getProjectId()); if (query.getNamespace() != null) { pbBuilder.setNamespaceId(query.getNamespace()); } else if (datastore.getOptions().getNamespace() != null) { pbBuilder.setNamespaceId(datastore.getOptions().getNamespace()); } partitionIdPb = pbBuilder.build(); sendRequest(); if (runQueryResponsePb.getBatch().getSkippedResults() > 0) { cursor = runQueryResponsePb.getBatch().getSkippedCursor(); } else { cursor = mostRecentQueryPb.getStartCursor(); } }