com.navercorp.fixturemonkey.tree.CompositeNodeResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fixture-monkey Show documentation
The easiest way to generate controllable arbitrary test objects
The newest version!
/*
 * Fixture Monkey
 *
 * Copyright (c) 2021-present NAVER Corp.
 *
 * 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 com.navercorp.fixturemonkey.tree;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

import org.apiguardian.api.API;
import org.apiguardian.api.API.Status;

@API(since = "0.4.0", status = Status.MAINTAINED)
public final class CompositeNodeResolver implements NodeResolver {
	private final List nodeResolvers;

	public CompositeNodeResolver(NodeResolver... nodeResolvers) {
		this(Arrays.asList(nodeResolvers));
	}

	public CompositeNodeResolver(List nodeResolvers) {
		this.nodeResolvers = distinct(nodeResolvers);
	}

	@Override
	public List resolve(ObjectNode objectNode) {
		LinkedList nextNodes = new LinkedList<>();
		nextNodes.add(objectNode);
		for (NodeResolver nodeResolver : nodeResolvers) {
			List resolvedNodes = new LinkedList<>();
			while (!nextNodes.isEmpty()) {
				ObjectNode currentNode = nextNodes.pop();
				resolvedNodes.addAll(nodeResolver.resolve(currentNode));
			}
			nextNodes.addAll(resolvedNodes);
		}
		return nextNodes;
	}

	public List flatten() {
		List flatten = new ArrayList<>();

		for (NodeResolver nodeResolver : nodeResolvers) {
			if (nodeResolver instanceof CompositeNodeResolver) {
				flatten.addAll(((CompositeNodeResolver)nodeResolver).nodeResolvers);
			} else {
				flatten.add(nodeResolver);
			}
		}

		return flatten;
	}

	private List distinct(List resolvers) {
		if (resolvers.isEmpty()) {
			return Collections.emptyList();
		}

		int length = resolvers.size();
		List result = new ArrayList<>();
		result.add(resolvers.get(0));

		for (int i = 1; i < length; i++) {
			NodeResolver resolver = resolvers.get(i);
			if (resolver instanceof CompositeNodeResolver) {
				CompositeNodeResolver compositeNodeResolver = (CompositeNodeResolver)resolver;
				List componentNodeResolvers = compositeNodeResolver.flatten().stream()
					.filter(it -> !(it instanceof IdentityNodeResolver))
					.collect(Collectors.toList());

				result.add(new CompositeNodeResolver(componentNodeResolvers));
			} else {
				result.add(resolver);
			}
		}
		return result;
	}

	@Override
	public boolean equals(Object obj) {
		if (this == obj) {
			return true;
		}
		if (obj == null || getClass() != obj.getClass()) {
			return false;
		}
		CompositeNodeResolver that = (CompositeNodeResolver)obj;
		return nodeResolvers.equals(that.nodeResolvers);
	}

	@Override
	public int hashCode() {
		return Objects.hash(nodeResolvers);
	}

	@Override
	public List toNextNodePredicate() {
		return flatten().stream()
			.flatMap(it -> it.toNextNodePredicate().stream())
			.filter(Objects::nonNull)
			.collect(Collectors.toList());
	}
}