Package google.registry.beam.common

Class RegistryJpaIO

java.lang.Object

google.registry.beam.common.RegistryJpaIO

public final class RegistryJpaIO
extends Object

Contains IO transforms for a BEAM pipeline that interacts with a single database through a JpaTransactionManager.

The JpaTransactionManager is instantiated once on each pipeline worker VM (through RegistryPipelineWorkerInitializer), made available through the static method TransactionManagerFactory.tm(), and is shared by all threads on the VM. Configuration is through RegistryPipelineOptions.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	RegistryJpaIO.Read<R,T>	A transform that transactionally executes a JPA CriteriaQuery and adds the results to the BEAM pipeline.
static class	RegistryJpaIO.Write<T>	A transform that writes a PCollection of entities to the SQL database using the JpaTransactionManager.

Method Summary

Modifier and Type	Method	Description
static <R> RegistryJpaIO.Read<R,R>	read(RegistryQuery.CriteriaQuerySupplier<R> query)
static <R, T> RegistryJpaIO.Read<R,T>	read(RegistryQuery.CriteriaQuerySupplier<R> query, org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)
static <R, T> RegistryJpaIO.Read<R,T>	read(String sql, boolean nativeQuery, org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)
static <R, T> RegistryJpaIO.Read<R,T>	read(String jpql, Class<R> clazz, org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)
static <R, T> RegistryJpaIO.Read<R,T>	read(String sql, Map<String,Object> parameter, boolean nativeQuery, org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)	Returns a RegistryJpaIO.Read connector based on the given native or jpql query string.
static <R, T> RegistryJpaIO.Read<R,T>	read(String jpql, Map<String,Object> parameter, Class<R> clazz, org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)	Returns a RegistryJpaIO.Read connector based on the given jpql typed query string.
static <T> RegistryJpaIO.Write<T>	write()

Methods inherited from class java.lang.Object

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

Method Details

read

public static <R> RegistryJpaIO.Read<R,R> read(RegistryQuery.CriteriaQuerySupplier<R> query)

read

public static <R,
T> RegistryJpaIO.Read<R,T> read(RegistryQuery.CriteriaQuerySupplier<R> query,
 org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)

read

public static <R,
T> RegistryJpaIO.Read<R,T> read(String sql,
 boolean nativeQuery,
 org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)

read

public static <R,
T> RegistryJpaIO.Read<R,T> read(String sql,
 @Nullable
 Map<String,Object> parameter,
 boolean nativeQuery,
 org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)

Returns a RegistryJpaIO.Read connector based on the given native or jpql query string.

User should take care to prevent sql-injection attacks.

read

public static <R,
T> RegistryJpaIO.Read<R,T> read(String jpql,
 Class<R> clazz,
 org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)

read

public static <R,
T> RegistryJpaIO.Read<R,T> read(String jpql,
 @Nullable
 Map<String,Object> parameter,
 Class<R> clazz,
 org.apache.beam.sdk.transforms.SerializableFunction<R,T> resultMapper)

Returns a RegistryJpaIO.Read connector based on the given jpql typed query string.

User should take care to prevent sql-injection attacks.

write

public static <T> RegistryJpaIO.Write<T> write()