AbstractIndexingVisitor.java
/*
* SPDX-FileCopyrightText: none
* SPDX-License-Identifier: CC0-1.0
*/
package gov.nist.secauto.oscal.lib.profile.resolver.support;
import gov.nist.secauto.metaschema.core.metapath.item.node.IAssemblyNodeItem;
import gov.nist.secauto.metaschema.core.metapath.item.node.IRootAssemblyNodeItem;
import gov.nist.secauto.metaschema.core.util.ObjectUtils;
import java.util.EnumSet;
import java.util.Set;
import edu.umd.cs.findbugs.annotations.NonNull;
public abstract class AbstractIndexingVisitor<T, R>
extends AbstractCatalogEntityVisitor<T, R> {
public AbstractIndexingVisitor() {
this(ObjectUtils.notNull(EnumSet.allOf(IEntityItem.ItemType.class)));
}
public AbstractIndexingVisitor(@NonNull Set<IEntityItem.ItemType> itemTypesToIndex) {
super(itemTypesToIndex);
}
@NonNull
protected abstract IIndexer getIndexer(T state);
@Override
public R visitGroup(IAssemblyNodeItem item, R childResult, T state) {
getIndexer(state).addGroup(item);
return childResult;
}
@Override
public R visitControl(IAssemblyNodeItem item, R childResult, T state) {
getIndexer(state).addControl(item);
return childResult;
}
@Override
protected R visitParameter(@NonNull IAssemblyNodeItem parameterItem,
@NonNull IAssemblyNodeItem catalogOrGroupOrControl, T state) {
getIndexer(state).addParameter(parameterItem);
return newDefaultResult(state);
}
@Override
protected void visitPart(@NonNull IAssemblyNodeItem partItem,
@NonNull IAssemblyNodeItem catalogOrGroupOrControl, T state) {
getIndexer(state).addPart(partItem);
}
@Override
protected void visitRole(IAssemblyNodeItem roleItem, IAssemblyNodeItem metadataItem, T state) {
getIndexer(state).addRole(roleItem);
}
@Override
protected void visitLocation(IAssemblyNodeItem locationItem, IAssemblyNodeItem metadataItem,
T state) {
getIndexer(state).addLocation(locationItem);
}
@Override
protected void visitParty(IAssemblyNodeItem partyItem, IAssemblyNodeItem metadataItem, T state) {
getIndexer(state).addParty(partyItem);
}
@Override
protected void visitResource(IAssemblyNodeItem resourceItem, IRootAssemblyNodeItem rootItem, T state) {
getIndexer(state).addResource(resourceItem);
}
}