org.apache.cassandra.schema.Keyspaces Maven / Gradle / Ivy
Show all versions of cassandra-all Show documentation
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.cassandra.schema;
import java.util.Collection;
import java.util.Iterator;
import java.util.Optional;
import java.util.function.Predicate;
import java.util.stream.Stream;
import javax.annotation.Nullable;
import com.google.common.collect.*;
import org.apache.cassandra.schema.KeyspaceMetadata.KeyspaceDiff;
public final class Keyspaces implements Iterable
{
private static final Keyspaces NONE = builder().build();
private final ImmutableMap keyspaces;
private final ImmutableMap tables;
private Keyspaces(Builder builder)
{
keyspaces = builder.keyspaces.build();
tables = builder.tables.build();
}
public static Builder builder()
{
return new Builder();
}
public static Keyspaces none()
{
return NONE;
}
public static Keyspaces of(KeyspaceMetadata... keyspaces)
{
return builder().add(keyspaces).build();
}
public Iterator iterator()
{
return keyspaces.values().iterator();
}
public Stream stream()
{
return keyspaces.values().stream();
}
public ImmutableSet names()
{
return keyspaces.keySet();
}
/**
* Get the keyspace with the specified name
*
* @param name a non-qualified keyspace name
* @return an empty {@link Optional} if the table name is not found; a non-empty optional of {@link KeyspaceMetadata} otherwise
*/
public Optional get(String name)
{
return Optional.ofNullable(keyspaces.get(name));
}
@Nullable
public KeyspaceMetadata getNullable(String name)
{
return keyspaces.get(name);
}
public boolean containsKeyspace(String name)
{
return keyspaces.containsKey(name);
}
@Nullable
public TableMetadata getTableOrViewNullable(TableId id)
{
return tables.get(id);
}
public boolean isEmpty()
{
return keyspaces.isEmpty();
}
public Keyspaces filter(Predicate predicate)
{
Builder builder = builder();
stream().filter(predicate).forEach(builder::add);
return builder.build();
}
/**
* Creates a Keyspaces instance with the keyspace with the provided name removed
*/
public Keyspaces without(String name)
{
KeyspaceMetadata keyspace = getNullable(name);
if (keyspace == null)
throw new IllegalStateException(String.format("Keyspace %s doesn't exists", name));
return filter(k -> k != keyspace);
}
public Keyspaces without(Collection names)
{
return filter(k -> !names.contains(k.name));
}
public Keyspaces withAddedOrUpdated(KeyspaceMetadata keyspace)
{
return builder().add(Iterables.filter(this, k -> !k.name.equals(keyspace.name)))
.add(keyspace)
.build();
}
/**
* Returns a new {@link Keyspaces} equivalent to this one, but with the provided keyspace metadata either added (if
* this {@link Keyspaces} does not have that keyspace), or replaced by the provided definition.
*
* Note that if this contains the provided keyspace, its pre-existing definition is discarded and completely
* replaced with the newly provided one. See {@link #withAddedOrUpdated(KeyspaceMetadata)} if you wish the provided
* definition to be "merged" with the existing one instead.
*
* @param keyspace the keyspace metadata to add, or replace the existing definition with.
* @return the newly created object.
*/
public Keyspaces withAddedOrReplaced(KeyspaceMetadata keyspace)
{
return builder().add(Iterables.filter(this, k -> !k.name.equals(keyspace.name)))
.add(keyspace)
.build();
}
/**
* Calls {@link #withAddedOrReplaced(KeyspaceMetadata)} on all the keyspaces of the provided {@link Keyspaces}.
*
* @param keyspaces the keyspaces to add, or replace if existing.
* @return the newly created object.
*/
public Keyspaces withAddedOrReplaced(Keyspaces keyspaces)
{
return builder().add(Iterables.filter(this, k -> !keyspaces.containsKeyspace(k.name)))
.add(keyspaces)
.build();
}
public void validate()
{
keyspaces.values().forEach(KeyspaceMetadata::validate);
}
@Override
public boolean equals(Object o)
{
return this == o || (o instanceof Keyspaces && keyspaces.equals(((Keyspaces) o).keyspaces));
}
@Override
public int hashCode()
{
return keyspaces.hashCode();
}
@Override
public String toString()
{
return keyspaces.values().toString();
}
public int size()
{
return keyspaces.size();
}
public static final class Builder
{
private final ImmutableMap.Builder keyspaces = new ImmutableMap.Builder<>();
private final ImmutableMap.Builder tables = new ImmutableMap.Builder<>();
private Builder()
{
}
public Keyspaces build()
{
return new Keyspaces(this);
}
public Builder add(KeyspaceMetadata keyspace)
{
keyspaces.put(keyspace.name, keyspace);
keyspace.tables.forEach(t -> tables.put(t.id, t));
keyspace.views.forEach(v -> tables.put(v.metadata.id, v.metadata));
return this;
}
public Builder add(KeyspaceMetadata... keyspaces)
{
for (KeyspaceMetadata keyspace : keyspaces)
add(keyspace);
return this;
}
public Builder add(Iterable keyspaces)
{
keyspaces.forEach(this::add);
return this;
}
}
public static KeyspacesDiff diff(Keyspaces before, Keyspaces after)
{
return KeyspacesDiff.diff(before, after);
}
public static final class KeyspacesDiff
{
public static final KeyspacesDiff NONE = new KeyspacesDiff(Keyspaces.none(), Keyspaces.none(), ImmutableList.of());
public final Keyspaces created;
public final Keyspaces dropped;
public final ImmutableList altered;
private KeyspacesDiff(Keyspaces created, Keyspaces dropped, ImmutableList altered)
{
this.created = created;
this.dropped = dropped;
this.altered = altered;
}
private static KeyspacesDiff diff(Keyspaces before, Keyspaces after)
{
if (before == after)
return NONE;
Keyspaces created = after.filter(k -> !before.containsKeyspace(k.name));
Keyspaces dropped = before.filter(k -> !after.containsKeyspace(k.name));
ImmutableList.Builder altered = ImmutableList.builder();
before.forEach(keyspaceBefore ->
{
KeyspaceMetadata keyspaceAfter = after.getNullable(keyspaceBefore.name);
if (null != keyspaceAfter)
KeyspaceMetadata.diff(keyspaceBefore, keyspaceAfter).ifPresent(altered::add);
});
return new KeyspacesDiff(created, dropped, altered.build());
}
public boolean isEmpty()
{
return created.isEmpty() && dropped.isEmpty() && altered.isEmpty();
}
@Override
public String toString()
{
return "KeyspacesDiff{" +
"created=" + created +
", dropped=" + dropped +
", altered=" + altered +
'}';
}
}
}