org.apache.beam.it.gcp.datastore

Class DatastoreResourceManager

java.lang.Object

org.apache.beam.it.gcp.datastore.DatastoreResourceManager

All Implemented Interfaces:

org.apache.beam.it.common.ResourceManager

public class DatastoreResourceManager
extends java.lang.Object
implements org.apache.beam.it.common.ResourceManager

Client for managing Datastore resources.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DatastoreResourceManager.Builder

Constructor Summary

Constructors

Constructor and Description
DatastoreResourceManager(DatastoreResourceManager.Builder builder)

Method Summary

Modifier and Type	Method and Description
static DatastoreResourceManager.Builder	builder(java.lang.String project, java.lang.String namespace, com.google.auth.Credentials credentials)
void	cleanupAll() Deletes all created entities and cleans up the Datastore client.
java.util.List<com.google.cloud.datastore.Entity>	insert(java.lang.String kind, java.util.Map<java.lang.Long,com.google.cloud.datastore.FullEntity<?>> entities) Insert entities to Datastore.
java.util.List<com.google.cloud.datastore.Entity>	query(java.lang.String gqlQuery) Run a Gql Query and return the results in entity format.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DatastoreResourceManager

public DatastoreResourceManager(DatastoreResourceManager.Builder builder)

Method Detail

insert

public java.util.List<com.google.cloud.datastore.Entity> insert(java.lang.String kind,
                                                                java.util.Map<java.lang.Long,com.google.cloud.datastore.FullEntity<?>> entities)

Insert entities to Datastore.

Parameters:

kind - Kind of document to insert.

entities - Entities to insert to Datastore.

Returns:

Entities.

query

public java.util.List<com.google.cloud.datastore.Entity> query(java.lang.String gqlQuery)

Run a Gql Query and return the results in entity format.

Parameters:

gqlQuery - Gql Query to run.

Returns:

Entities returned from the query.

cleanupAll

public void cleanupAll()
                throws DatastoreResourceManagerException

Deletes all created entities and cleans up the Datastore client.

Specified by:

cleanupAll in interface org.apache.beam.it.common.ResourceManager

Throws:

DatastoreResourceManagerException - if there is an error deleting the entities in Datastore.

builder

public static DatastoreResourceManager.Builder builder(java.lang.String project,
                                                       java.lang.String namespace,
                                                       com.google.auth.Credentials credentials)