org.wildfly.clustering.marshalling.protostream.CompositeSerializationContextInitializer Maven / Gradle / Ivy

Go to download
/*
 * Copyright The WildFly Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.wildfly.clustering.marshalling.protostream;

import java.util.EnumSet;
import java.util.List;

/**
 * {@link SerializationContextInitializer} that registers a set of {@link SerializationContextInitializer} instances.
 * @author Paul Ferraro
 */
public class CompositeSerializationContextInitializer implements SerializationContextInitializer {

	private final Iterable initializers;

	public CompositeSerializationContextInitializer(SerializationContextInitializer initializer1, SerializationContextInitializer initializer2) {
		this(List.of(initializer1, initializer2));
	}

	public CompositeSerializationContextInitializer(SerializationContextInitializer... initializers) {
		this(List.of(initializers));
	}

	public  & SerializationContextInitializer> CompositeSerializationContextInitializer(Class enumClass) {
		this(EnumSet.allOf(enumClass));
	}

	public CompositeSerializationContextInitializer(Iterable initializers) {
		this.initializers = initializers;
	}

	@Override
	public void registerSchema(SerializationContext context) {
		for (SerializationContextInitializer initializer : this.initializers) {
			initializer.registerSchema(context);
		}
	}

	@Override
	public void registerMarshallers(SerializationContext context) {
		for (SerializationContextInitializer initializer : this.initializers) {
			initializer.registerMarshallers(context);
		}
	}
}