io.stargate.db.schema.Table Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of persistence-api Show documentation
There is a newer version: 2.1.0-BETA-19
/*
 * Copyright DataStax, Inc. and/or The Stargate Authors
 *
 * 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 io.stargate.db.schema;

import com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import org.immutables.value.Value;

@Value.Immutable(prehash = true)
public abstract class Table extends AbstractTable {
  private static final long serialVersionUID = 5913904335622827700L;

  public abstract List indexes();

  @Value.Lazy
  Map indexMap() {
    return indexes().stream().collect(Collectors.toMap(Index::name, Function.identity()));
  }

  public Index index(String name) {
    return indexMap().get(name);
  }

  public static Table create(
      String keyspace, String name, Iterable columns, Iterable indexes) {
    return ImmutableTable.builder()
        .keyspace(keyspace)
        .name(name)
        .columns(columns)
        .indexes(indexes)
        .build();
  }

  public static Table reference(String keyspace, String name) {
    return ImmutableTable.builder()
        .keyspace(keyspace)
        .name(name)
        .columns(ImmutableList.of())
        .indexes(ImmutableList.of())
        .build();
  }

  @Override
  public int priority() {
    return 0;
  }

  @Override
  public String toString() {
    StringBuilder tableBuilder = new StringBuilder();
    String name = name();
    tableBuilder.append("Table '").append(name).append("'");
    tableBuilder.append(":\n");
    tableBuilder.append(
        columns().stream()
            .map(
                c ->
                    "    "
                        + c.name()
                        + " "
                        + c.type()
                        + (c.kind() == Column.Kind.Regular ? "" : (" " + c.kind())))
            .collect(Collectors.joining("\n")));
    if (!indexes().isEmpty()) {
      tableBuilder.append("\n");
      tableBuilder.append("\n  Table '").append(name).append("' indexes:\n");
      tableBuilder.append(
          indexes().stream().map(idx -> "    " + idx.toString()).collect(Collectors.joining("\n")));
      tableBuilder.append("\n");
    }
    tableBuilder.append("\n");
    return tableBuilder.toString();
  }

  @Override
  public String indexTypeName() {
    return "Table: " + name();
  }
}