com.github.nosan.embedded.cassandra.cql.CqlDataSet Maven / Gradle / Ivy
/*
* Copyright 2020-2021 the original author or 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
*
* https://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.github.nosan.embedded.cassandra.cql;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.function.Consumer;
import java.util.stream.Collectors;
import com.github.nosan.embedded.cassandra.commons.ClassPathResource;
import com.github.nosan.embedded.cassandra.commons.Resource;
/**
* {@link CqlDataSet} represents the set of {@link CqlScript}.
*
* @author Dmytro Nosan
* @see CqlScript
* @see DefaultCqlDataSet
* @since 4.0.1
*/
@FunctionalInterface
public interface CqlDataSet extends CqlScript {
/**
* Creates {@link CqlDataSet} with the specified resource names and default charset.
*
* @param names the resource names
* @return a new {@link CqlDataSet}
*/
static CqlDataSet ofClassPaths(String... names) {
return ofClassPaths(Charset.defaultCharset(), names);
}
/**
* Creates {@link CqlDataSet} with the specified resource names and charset.
*
* @param names the resource names
* @param charset the encoding to use
* @return a new {@link CqlDataSet}
*/
static CqlDataSet ofClassPaths(Charset charset, String... names) {
Objects.requireNonNull(charset, "Charset must not be null");
Objects.requireNonNull(names, "Classpath resources must not be null");
return new DefaultCqlDataSet(Arrays.stream(names)
.map(ClassPathResource::new)
.map(resource -> new ResourceCqlScript(resource, charset))
.collect(Collectors.toList()));
}
/**
* Creates {@link CqlDataSet} with the specified resources and default charset.
*
* @param resources the resources to use
* @return a new {@link CqlDataSet}
*/
static CqlDataSet ofResources(Resource... resources) {
return ofResources(Charset.defaultCharset(), resources);
}
/**
* Creates {@link CqlDataSet} with the specified resources and charset.
*
* @param resources the resources to use
* @param charset the encoding to use
* @return a new {@link CqlDataSet}
*/
static CqlDataSet ofResources(Charset charset, Resource... resources) {
Objects.requireNonNull(charset, "Charset must not be null");
Objects.requireNonNull(resources, "Resources must not be null");
return new DefaultCqlDataSet(Arrays.stream(resources)
.map(resource -> new ResourceCqlScript(resource, charset))
.collect(Collectors.toList()));
}
/**
* Creates {@link CqlDataSet} with the specified CQL scripts.
*
* @param scripts the scripts to use
* @return a new {@link CqlDataSet}
*/
static CqlDataSet ofScripts(CqlScript... scripts) {
Objects.requireNonNull(scripts, "Cql Scripts must not be null");
List result = new ArrayList<>();
for (CqlScript script : scripts) {
if (script instanceof CqlDataSet) {
result.addAll(((CqlDataSet) script).getScripts());
}
else {
result.add(script);
}
}
return new DefaultCqlDataSet(result);
}
/**
* Performs the given {@code callback} for each script of the {@link CqlDataSet}.
*
* @param callback The action to be performed for each script
*/
default void forEachScript(Consumer callback) {
Objects.requireNonNull(callback, "Callback must not be null");
getScripts().forEach(callback);
}
@Override
default List getStatements() {
List statements = new ArrayList<>();
forEachScript(script -> statements.addAll(script.getStatements()));
return Collections.unmodifiableList(statements);
}
/**
* Gets {@code CQL} scripts.
*
* @return {@code CQL} scripts (never null)
*/
List getScripts();
}