// Copyright (c) Microsoft Corporation. All rights reserved. // Licensed under the MIT license. package com.mojang.serialization.codecs; import com.mojang.datafixers.util.Either; import com.mojang.datafixers.util.Pair; import com.mojang.serialization.Codec; import com.mojang.serialization.DataResult; import com.mojang.serialization.DynamicOps; import java.util.Objects; public final class EitherCodec<F, S> implements Codec<Either<F, S>> { private final Codec<F> first; private final Codec<S> second; public EitherCodec(final Codec<F> first, final Codec<S> second) { this.first = first; this.second = second; } @Override public <T> DataResult<Pair<Either<F, S>, T>> decode(final DynamicOps<T> ops, final T input) { final DataResult<Pair<Either<F, S>, T>> firstRead = first.decode(ops, input).map(vo -> vo.mapFirst(Either::left)); if (firstRead.result().isPresent()) { return firstRead; } return second.decode(ops, input).map(vo -> vo.mapFirst(Either::right)); } @Override public <T> DataResult<T> encode(final Either<F, S> input, final DynamicOps<T> ops, final T prefix) { return input.map( value1 -> first.encode(value1, ops, prefix), value2 -> second.encode(value2, ops, prefix) ); } @Override public boolean equals(final Object o) { if (this == o) { return true; } if (o == null || getClass() != o.getClass()) { return false; } final EitherCodec<?, ?> eitherCodec = ((EitherCodec<?, ?>) o); return Objects.equals(first, eitherCodec.first) && Objects.equals(second, eitherCodec.second); } @Override public int hashCode() { return Objects.hash(first, second); } @Override public String toString() { return "EitherCodec[" + first + ", " + second + ']'; } }