de.arbeitsagentur.opdt.keycloak.cassandra.clientScope.persistence.ClientScopeDaoImpl__MapperGenerated Maven / Gradle / Ivy

Go to download
package de.arbeitsagentur.opdt.keycloak.cassandra.clientScope.persistence;

import com.datastax.oss.driver.api.core.cql.BoundStatement;
import com.datastax.oss.driver.api.core.cql.BoundStatementBuilder;
import com.datastax.oss.driver.api.core.cql.PreparedStatement;
import com.datastax.oss.driver.api.core.cql.ResultSet;
import com.datastax.oss.driver.api.core.cql.SimpleStatement;
import com.datastax.oss.driver.api.mapper.MapperContext;
import com.datastax.oss.driver.api.mapper.entity.saving.NullSavingStrategy;
import com.datastax.oss.driver.internal.core.util.concurrent.BlockingOperation;
import com.datastax.oss.driver.internal.core.util.concurrent.CompletableFutures;
import com.datastax.oss.driver.internal.mapper.DaoBase;
import com.datastax.oss.driver.internal.querybuilder.update.DefaultUpdate;
import de.arbeitsagentur.opdt.keycloak.cassandra.clientScope.persistence.entities.ClientScopes;
import de.arbeitsagentur.opdt.keycloak.cassandra.clientScope.persistence.entities.ClientScopesHelper__MapperGenerated;
import java.lang.Boolean;
import java.lang.Override;
import java.lang.String;
import java.lang.SuppressWarnings;
import java.lang.Throwable;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Generated by the DataStax driver mapper, do not edit directly.
 */
@SuppressWarnings("all")
public class ClientScopeDaoImpl__MapperGenerated extends DaoBase implements ClientScopeDao {
  private static final Logger LOG = LoggerFactory.getLogger(ClientScopeDaoImpl__MapperGenerated.class);

  private final ClientScopesHelper__MapperGenerated clientScopesHelper;

  private final PreparedStatement getClientScopesByRealmIdStatement;

  private final PreparedStatement deleteAllClientScopesStatement;

  private final PreparedStatement insertStatement;

  private final PreparedStatement updateStatement;

  private final PreparedStatement deleteStatement;

  private ClientScopeDaoImpl__MapperGenerated(MapperContext context,
      ClientScopesHelper__MapperGenerated clientScopesHelper,
      PreparedStatement getClientScopesByRealmIdStatement,
      PreparedStatement deleteAllClientScopesStatement, PreparedStatement insertStatement,
      PreparedStatement updateStatement, PreparedStatement deleteStatement) {
    super(context);
    this.clientScopesHelper = clientScopesHelper;
    this.getClientScopesByRealmIdStatement = getClientScopesByRealmIdStatement;
    this.deleteAllClientScopesStatement = deleteAllClientScopesStatement;
    this.insertStatement = insertStatement;
    this.updateStatement = updateStatement;
    this.deleteStatement = deleteStatement;
  }

  @Override
  public ClientScopes getClientScopesByRealmId(String realmId) {
    BoundStatementBuilder boundStatementBuilder = getClientScopesByRealmIdStatement.boundStatementBuilder();
    boundStatementBuilder = boundStatementBuilder.set("realmId", realmId, String.class);
    BoundStatement boundStatement = boundStatementBuilder.build();
    return executeAndMapToSingleEntity(boundStatement, clientScopesHelper);
  }

  @Override
  public void deleteAllClientScopes(String realmId) {
    BoundStatementBuilder boundStatementBuilder = deleteAllClientScopesStatement.boundStatementBuilder();
    boundStatementBuilder = boundStatementBuilder.set("realm_id", realmId, String.class);
    BoundStatement boundStatement = boundStatementBuilder.build();
    execute(boundStatement);
  }

  @Override
  public void insert(ClientScopes entity) {
    BoundStatementBuilder boundStatementBuilder = insertStatement.boundStatementBuilder();
    clientScopesHelper.set(entity, boundStatementBuilder, NullSavingStrategy.SET_TO_NULL, false);
    BoundStatement boundStatement = boundStatementBuilder.build();
    execute(boundStatement);
  }

  @Override
  public ResultSet update(ClientScopes entity, long expectedVersion) {
    BoundStatementBuilder boundStatementBuilder = updateStatement.boundStatementBuilder();
    clientScopesHelper.set(entity, boundStatementBuilder, NullSavingStrategy.SET_TO_NULL, false);
    boundStatementBuilder = boundStatementBuilder.setLong("expectedVersion", expectedVersion);
    BoundStatement boundStatement = boundStatementBuilder.build();
    return execute(boundStatement);
  }

  @Override
  public void delete(ClientScopes entity) {
    BoundStatementBuilder boundStatementBuilder = deleteStatement.boundStatementBuilder();
    boundStatementBuilder = boundStatementBuilder.set("realm_id", entity.getRealmId(), String.class);
    BoundStatement boundStatement = boundStatementBuilder.build();
    execute(boundStatement);
  }

  public static CompletableFuture initAsync(MapperContext context) {
    LOG.debug("[{}] Initializing new instance for keyspace = {} and table = {}",
        context.getSession().getName(),
        context.getKeyspaceId(),
        context.getTableId());
    try {
      // Initialize all entity helpers
      ClientScopesHelper__MapperGenerated clientScopesHelper = new ClientScopesHelper__MapperGenerated(context);
      if ((Boolean)context.getCustomState().get("datastax.mapper.schemaValidationEnabled")) {
        clientScopesHelper.validateEntityFields();
      }
      List> prepareStages = new ArrayList<>();
      // Prepare the statement for `getClientScopesByRealmId(java.lang.String)`:
      SimpleStatement getClientScopesByRealmIdStatement_simple = clientScopesHelper.selectStart().whereRaw("realm_id = :realmId").build();
      LOG.debug("[{}] Preparing query `{}` for method getClientScopesByRealmId(java.lang.String)",
          context.getSession().getName(),
          getClientScopesByRealmIdStatement_simple.getQuery());
      CompletionStage getClientScopesByRealmIdStatement = prepare(getClientScopesByRealmIdStatement_simple, context);
      prepareStages.add(getClientScopesByRealmIdStatement);
      // Prepare the statement for `deleteAllClientScopes(java.lang.String)`:
      SimpleStatement deleteAllClientScopesStatement_simple = clientScopesHelper.deleteByPrimaryKeyParts(1).ifExists().build();
      LOG.debug("[{}] Preparing query `{}` for method deleteAllClientScopes(java.lang.String)",
          context.getSession().getName(),
          deleteAllClientScopesStatement_simple.getQuery());
      CompletionStage deleteAllClientScopesStatement = prepare(deleteAllClientScopesStatement_simple, context);
      prepareStages.add(deleteAllClientScopesStatement);
      // Prepare the statement for `insert(T)`:
      SimpleStatement insertStatement_simple = clientScopesHelper.insert().ifNotExists().build();
      LOG.debug("[{}] Preparing query `{}` for method insert(T)",
          context.getSession().getName(),
          insertStatement_simple.getQuery());
      CompletionStage insertStatement = prepare(insertStatement_simple, context);
      prepareStages.add(insertStatement);
      // Prepare the statement for `update(T,long)`:
      SimpleStatement updateStatement_simple = SimpleStatement.newInstance(((DefaultUpdate)clientScopesHelper.updateByPrimaryKey()).ifRaw("version = :expectedVersion").asCql());
      LOG.debug("[{}] Preparing query `{}` for method update(T,long)",
          context.getSession().getName(),
          updateStatement_simple.getQuery());
      CompletionStage updateStatement = prepare(updateStatement_simple, context);
      prepareStages.add(updateStatement);
      // Prepare the statement for `delete(T)`:
      SimpleStatement deleteStatement_simple = clientScopesHelper.deleteByPrimaryKeyParts(1).ifExists().build();
      LOG.debug("[{}] Preparing query `{}` for method delete(T)",
          context.getSession().getName(),
          deleteStatement_simple.getQuery());
      CompletionStage deleteStatement = prepare(deleteStatement_simple, context);
      prepareStages.add(deleteStatement);
      // Initialize all method invokers
      // Build the DAO when all statements are prepared
      return CompletableFutures.allSuccessful(prepareStages)
          .thenApply(v -> (ClientScopeDao) new ClientScopeDaoImpl__MapperGenerated(context,
              clientScopesHelper,
              CompletableFutures.getCompleted(getClientScopesByRealmIdStatement),
              CompletableFutures.getCompleted(deleteAllClientScopesStatement),
              CompletableFutures.getCompleted(insertStatement),
              CompletableFutures.getCompleted(updateStatement),
              CompletableFutures.getCompleted(deleteStatement)))
          .toCompletableFuture();
    } catch (Throwable t) {
      return CompletableFutures.failedFuture(t);
    }
  }

  public static ClientScopeDao init(MapperContext context) {
    BlockingOperation.checkNotDriverThread();
    return CompletableFutures.getUninterruptibly(initAsync(context));
  }
}