ru.testit.services.ThreadContext Maven / Gradle / Ivy

Go to download
package ru.testit.services;

import java.util.LinkedList;
import java.util.Objects;
import java.util.Optional;

/**
 * Storage that stores information about not finished tests and steps.
 */
public class ThreadContext {
    private final Context context = new Context();

    /**
     * Returns last (most recent) uuid.
     *
     * @return last uuid.
     */
    public Optional getCurrent() {
        final LinkedList uuids = context.get();
        return uuids.isEmpty()
                ? Optional.empty()
                : Optional.of(uuids.getFirst());
    }

    /**
     * Returns first (oldest) uuid.
     *
     * @return first uuid.
     */
    public Optional getRoot() {
        final LinkedList uuids = context.get();
        return uuids.isEmpty()
                ? Optional.empty()
                : Optional.of(uuids.getLast());
    }

    /**
     * Adds new uuid.
     *
     * @param uuid the value
     */
    public void start(final String uuid) {
        Objects.requireNonNull(uuid, "step uuid");
        context.get().push(uuid);
    }

    /**
     * Removes latest added uuid. Ignores empty context.
     *
     * @return removed uuid.
     */
    public Optional stop() {
        final LinkedList uuids = context.get();
        if (!uuids.isEmpty()) {
            return Optional.of(uuids.pop());
        }
        return Optional.empty();
    }

    /**
     * Removes all the data stored for current thread.
     */
    public void clear() {
        context.remove();
    }

    /**
     * Thread local context that stores information about not finished tests and steps.
     */

    private static class Context extends InheritableThreadLocal> {

        @Override
        public LinkedList initialValue() {
            return new LinkedList<>();
        }

        @Override
        protected LinkedList childValue(final LinkedList parentStepContext) {
            return new LinkedList<>(parentStepContext);
        }

    }
}