test/simulator/main/org/apache/cassandra/simulator/ActionList.java

/*
 * 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.simulator;

import java.util.AbstractCollection;
import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;
import java.util.function.Consumer;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import com.google.common.collect.Iterators;

import org.apache.cassandra.simulator.OrderOn.StrictSequential;
import org.apache.cassandra.utils.Throwables;

import static java.util.Arrays.copyOf;

public class ActionList extends AbstractCollection<Action>
{
    private static final ActionList EMPTY = new ActionList(new Action[0]);
    public static ActionList empty() { return EMPTY; }
    public static ActionList of(Action action) { return new ActionList(new Action[] { action }); }
    public static ActionList of(Stream<Action> action) { return new ActionList(action.toArray(Action[]::new)); }
    public static ActionList of(Collection<Action> actions) { return actions.isEmpty() ? EMPTY : new ActionList(actions.toArray(new Action[0])); }
    public static ActionList of(Action ... actions) { return new ActionList(actions); }

    private final Action[] actions;

    ActionList(Action[] actions)
    {
        this.actions = actions;
    }

    public int size()
    {
        return actions.length;
    }

    public boolean isEmpty()
    {
        return 0 == actions.length;
    }

    public Action get(int i)
    {
        return actions[i];
    }

    public Iterator<Action> iterator()
    {
        return Iterators.forArray(actions);
    }

    public Stream<Action> stream()
    {
        return Stream.of(actions);
    }

    public ActionList transform(Function<Action, Action> apply)
    {
        return ActionList.of(stream().map(apply));
    }

    public ActionList filter(Predicate<Action> apply)
    {
        return ActionList.of(stream().filter(apply));
    }

    public boolean anyMatch(Predicate<Action> test)
    {
        for (int i = 0 ; i < actions.length ; ++i)
            if (test.test(actions[i])) return true;
        return false;
    }

    public ActionList andThen(Action andThen)
    {
        return andThen(ActionList.of(andThen));
    }

    public ActionList andThen(ActionList andThen)
    {
        Action[] result = copyOf(actions, size() + andThen.size());
        System.arraycopy(andThen.actions, 0, result, size(), andThen.size());
        return new ActionList(result);
    }

    public ActionList setStrictlySequential()
    {
        return setStrictlySequentialOn(this);
    }

    public ActionList setStrictlySequentialOn(Object on)
    {
        if (isEmpty()) return this;
        StrictSequential orderOn = new StrictSequential(on);
        forEach(a -> a.orderOn(orderOn));
        return this;
    }

    public Throwable safeForEach(Consumer<Action> forEach)
    {
        Throwable result = null;
        for (Action action : actions)
        {
            try
            {
                forEach.accept(action);
            }
            catch (Throwable t)
            {
                result = Throwables.merge(result, t);
            }
        }
        return result;
    }

    public String toString()
    {
        return Arrays.toString(actions);
    }

    public String toReconcileString()
    {
        return Arrays.stream(actions).map(Action::toReconcileString).collect(Collectors.joining(",", "[", "]"));
    }
}