com.azure.cosmos.implementation.changefeed.common.DefaultObserverFactory Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of azure-cosmos Show documentation

This Package contains Microsoft Azure Cosmos SDK (with Reactive Extension Reactor support) for Azure Cosmos DB SQL API

There is a newer version: 4.63.3

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
package com.azure.cosmos.implementation.changefeed.common;

import com.azure.cosmos.implementation.changefeed.ChangeFeedObserver;
import com.azure.cosmos.implementation.changefeed.ChangeFeedObserverFactory;
import com.azure.cosmos.ChangeFeedProcessorContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;
import java.util.function.BiConsumer;
import java.util.function.Consumer;

public class DefaultObserverFactory implements ChangeFeedObserverFactory {
    private final Logger log = LoggerFactory.getLogger(DefaultObserverFactory.class);
    private final Consumer> consumer;
    private final BiConsumer, ChangeFeedProcessorContext> biConsumer;

    public DefaultObserverFactory(Consumer> consumer) {
        this.consumer = consumer;
        this.biConsumer = null;
    }

    public DefaultObserverFactory(BiConsumer, ChangeFeedProcessorContext> biConsumer) {
        this.biConsumer = biConsumer;
        this.consumer = null;
    }

    @Override
    public ChangeFeedObserver createObserver() {

        if (consumer != null) {
            return new DefaultObserver<>(consumer);
        } else if (biConsumer != null) {
            return new DefaultObserver<>(biConsumer);
        }

        throw new IllegalStateException("Both consumer and biConsumer cannot be null");
    }
}