com.microsoft.kiota.serialization.ParseNodeProxyFactory Maven / Gradle / Ivy

Go to download
package com.microsoft.kiota.serialization;

import jakarta.annotation.Nonnull;
import jakarta.annotation.Nullable;

import java.io.InputStream;
import java.util.Objects;
import java.util.function.Consumer;

/** Proxy factory that allows the composition of before and after callbacks on existing factories. */
public abstract class ParseNodeProxyFactory implements ParseNodeFactory {
    @Nonnull public String getValidContentType() {
        return _concrete.getValidContentType();
    }

    private final ParseNodeFactory _concrete;
    private final Consumer _onBefore;
    private final Consumer _onAfter;

    /**
     * Creates a new proxy factory that wraps the specified concrete factory while composing the before and after callbacks.
     * @param concrete the concrete factory to wrap
     * @param onBefore the callback to invoke before the deserialization of any model object.
     * @param onAfter the callback to invoke after the deserialization of any model object.
     */
    public ParseNodeProxyFactory(
            @Nonnull final ParseNodeFactory concrete,
            @Nullable final Consumer onBefore,
            @Nullable final Consumer onAfter) {
        _concrete = Objects.requireNonNull(concrete);
        _onBefore = onBefore;
        _onAfter = onAfter;
    }

    @Nonnull public ParseNode getParseNode(
            @Nonnull final String contentType, @Nonnull final InputStream rawResponse) {
        Objects.requireNonNull(contentType);
        Objects.requireNonNull(rawResponse);
        final ParseNode node = _concrete.getParseNode(contentType, rawResponse);
        final Consumer originalOnBefore = node.getOnBeforeAssignFieldValues();
        final Consumer originalOnAfter = node.getOnAfterAssignFieldValues();
        node.setOnBeforeAssignFieldValues(
                x -> {
                    if (this._onBefore != null) {
                        this._onBefore.accept(x);
                    }
                    if (originalOnBefore != null) {
                        originalOnBefore.accept(x);
                    }
                });
        node.setOnAfterAssignFieldValues(
                x -> {
                    if (this._onAfter != null) {
                        this._onAfter.accept(x);
                    }
                    if (originalOnAfter != null) {
                        originalOnAfter.accept(x);
                    }
                });
        return node;
    }
}