com.scalar.db.storage.cassandra.ScannerImpl 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.cassandra;

import static com.google.common.base.Preconditions.checkNotNull;

import com.datastax.driver.core.ResultSet;
import com.datastax.driver.core.Row;
import com.scalar.db.api.Result;
import com.scalar.db.api.Scanner;
import com.scalar.db.common.ScannerIterator;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Optional;
import javax.annotation.Nonnull;
import javax.annotation.concurrent.NotThreadSafe;

@NotThreadSafe
public final class ScannerImpl implements Scanner {
  private final ResultSet resultSet;
  private final ResultInterpreter resultInterpreter;

  private ScannerIterator scannerIterator;

  public ScannerImpl(ResultSet resultSet, ResultInterpreter resultInterpreter) {
    this.resultSet = checkNotNull(resultSet);
    this.resultInterpreter = checkNotNull(resultInterpreter);
  }

  @Override
  @Nonnull
  public Optional one() {
    Row row = resultSet.one();
    if (row == null) {
      return Optional.empty();
    }
    return Optional.of(resultInterpreter.interpret(row));
  }

  @Override
  @Nonnull
  public List all() {
    List results = new ArrayList<>();
    resultSet.forEach(r -> results.add(resultInterpreter.interpret(r)));
    return results;
  }

  @Override
  @Nonnull
  public Iterator iterator() {
    if (scannerIterator == null) {
      scannerIterator = new ScannerIterator(this);
    }
    return scannerIterator;
  }

  @Override
  public void close() {}
}