Skip to content

Add reactive global temporary tables creation strategies #1556

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Apr 4, 2023
Merged

Add reactive global temporary tables creation strategies #1556

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions ...rg/hibernate/reactive/provider/service/ReactiveSqmMultiTableMutationStrategyProvider.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,9 @@
import org.hibernate.metamodel.spi.RuntimeModelCreationContext;
import org.hibernate.query.sqm.mutation.internal.cte.CteInsertStrategy;
import org.hibernate.query.sqm.mutation.internal.cte.CteMutationStrategy;
import org.hibernate.query.sqm.mutation.internal.temptable.GlobalTemporaryTableInsertStrategy;
import org.hibernate.query.sqm.mutation.internal.temptable.GlobalTemporaryTableMutationStrategy;
import org.hibernate.query.sqm.mutation.internal.temptable.GlobalTemporaryTableStrategy;
import org.hibernate.query.sqm.mutation.internal.temptable.LocalTemporaryTableInsertStrategy;
import org.hibernate.query.sqm.mutation.internal.temptable.LocalTemporaryTableMutationStrategy;
import org.hibernate.query.sqm.mutation.internal.temptable.PersistentTableInsertStrategy;
Expand All @@ -20,6 +23,8 @@
import org.hibernate.query.sqm.mutation.spi.SqmMultiTableMutationStrategyProvider;
import org.hibernate.reactive.query.sqm.mutation.internal.cte.ReactiveCteInsertStrategy;
import org.hibernate.reactive.query.sqm.mutation.internal.cte.ReactiveCteMutationStrategy;
import org.hibernate.reactive.query.sqm.mutation.internal.temptable.ReactiveGlobalTemporaryTableInsertStrategy;
import org.hibernate.reactive.query.sqm.mutation.internal.temptable.ReactiveGlobalTemporaryTableMutationStrategy;
import org.hibernate.reactive.query.sqm.mutation.internal.temptable.ReactiveLocalTemporaryTableInsertStrategy;
import org.hibernate.reactive.query.sqm.mutation.internal.temptable.ReactiveLocalTemporaryTableMutationStrategy;
import org.hibernate.reactive.query.sqm.mutation.internal.temptable.ReactivePersistentTableInsertStrategy;
Expand All @@ -42,6 +47,9 @@ public SqmMultiTableMutationStrategy createMutationStrategy(
if ( mutationStrategy instanceof PersistentTableMutationStrategy ) {
return new ReactivePersistentTableMutationStrategy( (PersistentTableMutationStrategy) mutationStrategy );
}
if ( mutationStrategy instanceof GlobalTemporaryTableMutationStrategy ) {
return new ReactiveGlobalTemporaryTableMutationStrategy( (GlobalTemporaryTableStrategy) mutationStrategy );
}
return mutationStrategy;
}

Expand Down Expand Up @@ -69,6 +77,9 @@ public SqmMultiTableInsertStrategy createInsertStrategy(
if ( insertStrategy instanceof PersistentTableInsertStrategy ) {
return new ReactivePersistentTableInsertStrategy( (PersistentTableInsertStrategy) insertStrategy );
}
if ( insertStrategy instanceof GlobalTemporaryTableInsertStrategy ) {
return new ReactiveGlobalTemporaryTableInsertStrategy( (GlobalTemporaryTableStrategy) insertStrategy );
}
return insertStrategy;
}

Expand Down
97 changes: 97 additions & 0 deletions ...ive/query/sqm/mutation/internal/temptable/ReactiveGlobalTemporaryTableInsertStrategy.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,97 @@
/* Hibernate, Relational Persistence for Idiomatic Java
*
* SPDX-License-Identifier: Apache-2.0
* Copyright: Red Hat Inc. and Hibernate Authors
*/
package org.hibernate.reactive.query.sqm.mutation.internal.temptable;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;

import org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess;
import org.hibernate.engine.spi.SessionFactoryImplementor;
import org.hibernate.metamodel.mapping.internal.MappingModelCreationProcess;
import org.hibernate.query.spi.DomainQueryExecutionContext;
import org.hibernate.query.sqm.internal.DomainParameterXref;
import org.hibernate.query.sqm.mutation.internal.temptable.GlobalTemporaryTableStrategy;
import org.hibernate.query.sqm.tree.insert.SqmInsertStatement;
import org.hibernate.reactive.query.sqm.mutation.spi.ReactiveSqmMultiTableInsertStrategy;

/**
* @see org.hibernate.query.sqm.mutation.internal.temptable.GlobalTemporaryTableInsertStrategy
*/
public class ReactiveGlobalTemporaryTableInsertStrategy extends GlobalTemporaryTableStrategy
implements ReactiveGlobalTemporaryTableStrategy, ReactiveSqmMultiTableInsertStrategy {

private final CompletableFuture<Void> tableCreatedStage = new CompletableFuture();

private final CompletableFuture<Void> tableDroppedStage = new CompletableFuture();

private boolean prepared;

private boolean dropIdTables;

public ReactiveGlobalTemporaryTableInsertStrategy(GlobalTemporaryTableStrategy strategy) {
super( strategy.getTemporaryTable(), strategy.getSessionFactory() );
}

@Override
public void prepare(
MappingModelCreationProcess mappingModelCreationProcess,
JdbcConnectionAccess connectionAccess) {
prepare( mappingModelCreationProcess, connectionAccess, tableCreatedStage );
}

@Override
public void release(SessionFactoryImplementor sessionFactory, JdbcConnectionAccess connectionAccess) {
release( sessionFactory, connectionAccess, tableDroppedStage );
}

@Override
public CompletionStage<Integer> reactiveExecuteInsert(
SqmInsertStatement<?> sqmInsertStatement,
DomainParameterXref domainParameterXref,
DomainQueryExecutionContext context) {
return tableCreatedStage.thenCompose( v -> new ReactiveTableBasedInsertHandler(
sqmInsertStatement,
domainParameterXref,
getTemporaryTable(),
getSessionFactory().getJdbcServices().getDialect().getTemporaryTableAfterUseAction(),
// generally a global temp table should already track a Connection-specific uid,
// but just in case a particular env needs it...
ReactiveGlobalTemporaryTableStrategy::sessionIdentifier,
getSessionFactory()
).reactiveExecute( context ) );
}


@Override
public boolean isPrepared() {
return prepared;
}

@Override
public void setPrepared(boolean prepared) {
this.prepared = prepared;
}

@Override
public boolean isDropIdTables() {
return dropIdTables;
}

@Override
public void setDropIdTables(boolean dropIdTables) {
this.dropIdTables = dropIdTables;
}

@Override
public CompletionStage<Void> getDropTableActionStage() {
return tableDroppedStage;
}

@Override
public CompletionStage<Void> getCreateTableActionStage() {
return tableCreatedStage;
}
}
113 changes: 113 additions & 0 deletions ...e/query/sqm/mutation/internal/temptable/ReactiveGlobalTemporaryTableMutationStrategy.java
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,113 @@
/* Hibernate, Relational Persistence for Idiomatic Java
*
* SPDX-License-Identifier: Apache-2.0
* Copyright: Red Hat Inc. and Hibernate Authors
*/
package org.hibernate.reactive.query.sqm.mutation.internal.temptable;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;

import org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess;
import org.hibernate.engine.spi.SessionFactoryImplementor;
import org.hibernate.metamodel.mapping.internal.MappingModelCreationProcess;
import org.hibernate.query.spi.DomainQueryExecutionContext;
import org.hibernate.query.sqm.internal.DomainParameterXref;
import org.hibernate.query.sqm.mutation.internal.temptable.GlobalTemporaryTableStrategy;
import org.hibernate.query.sqm.tree.delete.SqmDeleteStatement;
import org.hibernate.query.sqm.tree.update.SqmUpdateStatement;
import org.hibernate.reactive.query.sqm.mutation.spi.ReactiveSqmMultiTableMutationStrategy;

/**
* @see org.hibernate.query.sqm.mutation.internal.temptable.GlobalTemporaryTableMutationStrategy
*/
public class ReactiveGlobalTemporaryTableMutationStrategy extends GlobalTemporaryTableStrategy
implements ReactiveGlobalTemporaryTableStrategy, ReactiveSqmMultiTableMutationStrategy {

private final CompletableFuture<Void> tableCreatedStage = new CompletableFuture();

private final CompletableFuture<Void> tableDroppedStage = new CompletableFuture();

private boolean prepared;

private boolean dropIdTables;

public ReactiveGlobalTemporaryTableMutationStrategy(GlobalTemporaryTableStrategy strategy) {
super( strategy.getTemporaryTable(), strategy.getSessionFactory() );
}


@Override
public void prepare(
MappingModelCreationProcess mappingModelCreationProcess,
JdbcConnectionAccess connectionAccess) {
prepare( mappingModelCreationProcess, connectionAccess, tableCreatedStage );
}

@Override
public void release(SessionFactoryImplementor sessionFactory, JdbcConnectionAccess connectionAccess) {
release( sessionFactory, connectionAccess, tableDroppedStage );
}

@Override
public CompletionStage<Integer> reactiveExecuteUpdate(
SqmUpdateStatement<?> sqmUpdateStatement,
DomainParameterXref domainParameterXref,
DomainQueryExecutionContext context) {
return tableCreatedStage
.thenCompose( v -> new ReactiveTableBasedUpdateHandler(
sqmUpdateStatement,
domainParameterXref,
getTemporaryTable(),
getSessionFactory().getJdbcServices().getDialect().getTemporaryTableAfterUseAction(),
ReactivePersistentTableStrategy::sessionIdentifier,
getSessionFactory()
).reactiveExecute( context ) );
}

@Override
public CompletionStage<Integer> reactiveExecuteDelete(
SqmDeleteStatement<?> sqmDeleteStatement,
DomainParameterXref domainParameterXref,
DomainQueryExecutionContext context) {
return tableCreatedStage
.thenCompose( v -> new ReactiveTableBasedDeleteHandler(
sqmDeleteStatement,
domainParameterXref,
getTemporaryTable(),
getSessionFactory().getJdbcServices().getDialect().getTemporaryTableAfterUseAction(),
ReactiveGlobalTemporaryTableStrategy::sessionIdentifier,
getSessionFactory()
).reactiveExecute( context ) );
}

@Override
public CompletionStage<Void> getDropTableActionStage() {
return tableDroppedStage;
}

@Override
public CompletionStage<Void> getCreateTableActionStage() {
return tableCreatedStage;
}

@Override
public boolean isPrepared() {
return prepared;
}

@Override
public void setPrepared(boolean prepared) {
this.prepared = prepared;
}

@Override
public boolean isDropIdTables() {
return dropIdTables;
}

@Override
public void setDropIdTables(boolean dropIdTables) {
this.dropIdTables = dropIdTables;
}
}
Loading