com.datastax.driver.core.ArrayBackedRow Maven / Gradle / Ivy

/*
 * Copyright DataStax, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.datastax.driver.core;

import com.datastax.driver.core.exceptions.DriverInternalError;
import java.nio.ByteBuffer;
import java.util.List;
import java.util.regex.Pattern;

/** Implementation of a Row backed by an ArrayList. */
class ArrayBackedRow extends AbstractGettableData implements Row {

  /**
   * A pattern to parse (non-aliased) token column names of the form token(x). Note that starting
   * from Cassandra 2.2 built-in functions are declared in the system keyspace, so the function name
   * is prefixed with "system.".
   */
  private static final Pattern TOKEN_COLUMN_NAME = Pattern.compile("(system\\.)?token(.*)");

  private final ColumnDefinitions metadata;
  private final Token.Factory tokenFactory;
  private final List data;

  private ArrayBackedRow(
      ColumnDefinitions metadata,
      Token.Factory tokenFactory,
      ProtocolVersion protocolVersion,
      List data) {
    super(protocolVersion);
    this.metadata = metadata;
    this.tokenFactory = tokenFactory;
    this.data = data;
  }

  static Row fromData(
      ColumnDefinitions metadata,
      Token.Factory tokenFactory,
      ProtocolVersion protocolVersion,
      List data) {
    if (data == null) return null;

    return new ArrayBackedRow(metadata, tokenFactory, protocolVersion, data);
  }

  @Override
  public ColumnDefinitions getColumnDefinitions() {
    return metadata;
  }

  @Override
  protected DataType getType(int i) {
    return metadata.getType(i);
  }

  @Override
  protected String getName(int i) {
    return metadata.getName(i);
  }

  @Override
  protected ByteBuffer getValue(int i) {
    return data.get(i);
  }

  @Override
  protected CodecRegistry getCodecRegistry() {
    return metadata.codecRegistry;
  }

  @Override
  protected int getIndexOf(String name) {
    return metadata.getFirstIdx(name);
  }

  @Override
  public Token getToken(int i) {
    if (tokenFactory == null)
      throw new DriverInternalError(
          "Token factory not set. This should only happen at initialization time");

    checkType(i, tokenFactory.getTokenType().getName());

    ByteBuffer value = data.get(i);
    if (value == null || value.remaining() == 0) return null;

    return tokenFactory.deserialize(value, protocolVersion);
  }

  @Override
  public Token getToken(String name) {
    return getToken(metadata.getFirstIdx(name));
  }

  @Override
  public Token getPartitionKeyToken() {
    int i = 0;
    for (ColumnDefinitions.Definition column : metadata) {
      if (TOKEN_COLUMN_NAME.matcher(column.getName()).matches()) return getToken(i);
      i++;
    }
    throw new IllegalStateException(
        "Found no column named 'token(...)'. If the column is aliased, use getToken(String).");
  }

  @Override
  public String toString() {
    StringBuilder sb = new StringBuilder();
    sb.append("Row[");
    for (int i = 0; i < metadata.size(); i++) {
      if (i != 0) sb.append(", ");
      ByteBuffer bb = data.get(i);
      if (bb == null) sb.append("NULL");
      else {
        Object o =
            getCodecRegistry().codecFor(metadata.getType(i)).deserialize(bb, protocolVersion);
        if (o == null) {
          sb.append("NULL");
        } else {
          sb.append(o.toString());
        }
      }
    }
    sb.append(']');
    return sb.toString();
  }
}