com.scalar.db.storage.dynamo.DynamoMutation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scalardb Show documentation
A universal transaction manager that achieves database-agnostic transactions and distributed transactions that span multiple databases
There is a newer version: 3.14.0
Show newest version
package com.scalar.db.storage.dynamo;

import com.scalar.db.api.ConditionalExpression;
import com.scalar.db.api.Mutation;
import com.scalar.db.api.Put;
import com.scalar.db.api.TableMetadata;
import com.scalar.db.io.Value;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.annotation.Nonnull;
import javax.annotation.concurrent.Immutable;
import software.amazon.awssdk.services.dynamodb.model.AttributeValue;

/** A utility class for a mutation */
@Immutable
public class DynamoMutation extends DynamoOperation {

  DynamoMutation(Mutation mutation, TableMetadata metadata) {
    super(mutation, metadata);
  }

  @Nonnull
  public String getIfNotExistsCondition() {
    List expressions = new ArrayList<>();
    expressions.add("attribute_not_exists(" + PARTITION_KEY + ")");
    if (getOperation().getClusteringKey().isPresent()) {
      expressions.add("attribute_not_exists(" + CLUSTERING_KEY + ")");
    }

    return String.join(" AND ", expressions);
  }

  @Nonnull
  public String getIfExistsCondition() {
    List expressions = new ArrayList<>();
    expressions.add("attribute_exists(" + PARTITION_KEY + ")");
    if (getOperation().getClusteringKey().isPresent()) {
      expressions.add("attribute_exists(" + CLUSTERING_KEY + ")");
    }

    return String.join(" AND ", expressions);
  }

  @Nonnull
  public String getCondition() {
    ConditionExpressionBuilder builder =
        new ConditionExpressionBuilder(CONDITION_COLUMN_NAME_ALIAS, CONDITION_VALUE_ALIAS);
    Mutation mutation = (Mutation) getOperation();
    mutation.getCondition().ifPresent(c -> c.accept(builder));

    return builder.build();
  }

  @Nonnull
  public String getUpdateExpression() {
    return getUpdateExpression(false);
  }

  @Nonnull
  public String getUpdateExpressionWithKey() {
    return getUpdateExpression(true);
  }

  private String getUpdateExpression(boolean withKey) {
    Put put = (Put) getOperation();
    List expressions = new ArrayList<>();
    int i = 0;

    if (withKey) {
      for (Value unusedKey : put.getPartitionKey().get()) {
        expressions.add(COLUMN_NAME_ALIAS + i + " = " + VALUE_ALIAS + i);
        i++;
      }
      if (put.getClusteringKey().isPresent()) {
        for (Value unusedKey : put.getClusteringKey().get().get()) {
          expressions.add(COLUMN_NAME_ALIAS + i + " = " + VALUE_ALIAS + i);
          i++;
        }
      }
    }

    for (String unusedName : put.getColumns().keySet()) {
      expressions.add(COLUMN_NAME_ALIAS + i + " = " + VALUE_ALIAS + i);
      i++;
    }

    return "SET " + String.join(", ", expressions);
  }

  @Nonnull
  public Map getColumnMap() {
    return getColumnMap(false);
  }

  @Nonnull
  public Map getColumnMapWithKey() {
    return getColumnMap(true);
  }

  private Map getColumnMap(boolean withKey) {
    Put put = (Put) getOperation();
    Map columnMap = new HashMap<>();
    int i = 0;

    if (withKey) {
      for (Value key : put.getPartitionKey().get()) {
        columnMap.put(COLUMN_NAME_ALIAS + i, key.getName());
        i++;
      }
      if (put.getClusteringKey().isPresent()) {
        for (Value key : put.getClusteringKey().get().get()) {
          columnMap.put(COLUMN_NAME_ALIAS + i, key.getName());
          i++;
        }
      }
    }

    for (String name : put.getColumns().keySet()) {
      columnMap.put(COLUMN_NAME_ALIAS + i, name);
      i++;
    }

    return columnMap;
  }

  @Nonnull
  public Map getConditionColumnMap() {
    Map ret = new HashMap<>();
    Mutation mutation = (Mutation) getOperation();
    if (mutation.getCondition().isPresent()) {
      int index = 0;
      for (ConditionalExpression expression : mutation.getCondition().get().getExpressions()) {
        ret.put(CONDITION_COLUMN_NAME_ALIAS + index, expression.getName());
        index++;
      }
    }
    return ret;
  }

  @Nonnull
  public Map getConditionBindMap() {
    ValueBinder binder = new ValueBinder(CONDITION_VALUE_ALIAS);
    Mutation mutation = (Mutation) getOperation();
    mutation
        .getCondition()
        .ifPresent(c -> c.getExpressions().forEach(e -> e.getColumn().accept(binder)));

    return binder.build();
  }

  @Nonnull
  public Map getValueBindMap() {
    return getValueBindMap(false);
  }

  @Nonnull
  public Map getValueBindMapWithKey() {
    return getValueBindMap(true);
  }

  private Map getValueBindMap(boolean withKey) {
    ValueBinder binder = new ValueBinder(VALUE_ALIAS);
    Put put = (Put) getOperation();

    if (withKey) {
      put.getPartitionKey().getColumns().forEach(c -> c.accept(binder));
      put.getClusteringKey().ifPresent(k -> k.getColumns().forEach(c -> c.accept(binder)));
    }

    put.getColumns().values().forEach(c -> c.accept(binder));

    return binder.build();
  }
}