nl.topicus.jdbc.statement.DeleteKeyBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spanner-jdbc Show documentation

JDBC Driver for Google Cloud Spanner

There is a newer version: 1.1.6

package nl.topicus.jdbc.statement;

import java.sql.Date;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.util.HashMap;
import java.util.Map;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Key;
import nl.topicus.jdbc.shaded.com.google.rpc.Code;
import nl.topicus.jdbc.MetaDataStore.TableKeyMetaData;
import nl.topicus.jdbc.exception.CloudSpannerSQLException;
import nl.topicus.jdbc.util.CloudSpannerConversionUtil;

public class DeleteKeyBuilder {
  private final TableKeyMetaData table;

  private final boolean generateParameterMetaData;

  private Map keyValues = new HashMap<>();

  private String currentColumn = null;

  public DeleteKeyBuilder(TableKeyMetaData table, boolean generateParameterMetaData) {
    this.table = table;
    this.generateParameterMetaData = generateParameterMetaData;
  }

  public void set(String column) {
    currentColumn = column.toUpperCase();
  }

  public void to(Object value) {
    if (currentColumn == null)
      throw new IllegalArgumentException("No column set");
    keyValues.put(currentColumn, value);
    currentColumn = null;
  }

  public Key.Builder getKeyBuilder() throws SQLException {
    Key.Builder builder = Key.newBuilder();
    for (String key : table.getKeyColumns()) {
      Object value = keyValues.get(key);
      if (!generateParameterMetaData && value == null && !keyValues.containsKey(key)) {
        throw new CloudSpannerSQLException(
            "No value supplied for key column " + key
                + ". All key columns must be specified in the WHERE-clause of a DELETE-statement.",
            Code.INVALID_ARGUMENT);
      }
      builder.appendObject(convert(value));
    }
    return builder;
  }

  private Object convert(Object value) {
    if (value != null) {
      if (Date.class.isAssignableFrom(value.getClass())) {
        Date dateValue = (Date) value;
        return CloudSpannerConversionUtil.toCloudSpannerDate(dateValue);
      } else if (Timestamp.class.isAssignableFrom(value.getClass())) {
        Timestamp timeValue = (Timestamp) value;
        return CloudSpannerConversionUtil.toCloudSpannerTimestamp(timeValue);
      }
    }
    return value;
  }

}