net.pincette.mongo.streams.Context Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pincette-mongo-streams Show documentation
MongoDB Utilities
The newest version!
package net.pincette.mongo.streams;

import com.mongodb.reactivestreams.client.MongoDatabase;
import java.util.Map;
import java.util.concurrent.CompletionStage;
import java.util.function.BiFunction;
import java.util.logging.Logger;
import javax.json.JsonObject;
import net.pincette.mongo.Features;
import net.pincette.rs.streams.Message;

/**
 * The context for a pipeline stage.
 *
 * @author Werner Donn\u00e9
 * @since 1.1
 */
public class Context {
  /** The application. */
  public final String app;

  /** The MongoDB database. */
  public final MongoDatabase database;

  /** Extra features for the underlying MongoDB aggregation expression language and JSLT. */
  public final Features features;

  /** A logger in case stages have something to log. */
  public final Logger logger;

  /** The Kafka producer for stages that need to send messages top topics. */
  public final BiFunction, CompletionStage> producer;

  /** Extra stages that will be merged with the built-in stages, which always have precedence. */
  public final Map stageExtensions;

  /**
   * Writes tracing of the stages to the logger "net.pincette.mongo.streams" at log level INFO
   * .
   */
  public final boolean trace;

  public Context() {
    this(null, null, null, false, null, null, null);
  }

  private Context(
      final String app,
      final MongoDatabase database,
      final BiFunction, CompletionStage> producer,
      final boolean trace,
      final Features features,
      final Map stageExtensions,
      final Logger logger) {
    this.app = app;
    this.database = database;
    this.producer = producer;
    this.trace = trace;
    this.features = features;
    this.stageExtensions = stageExtensions;
    this.logger = logger;
  }

  public Context withApp(final String app) {
    return new Context(app, database, producer, trace, features, stageExtensions, logger);
  }

  public Context withDatabase(final MongoDatabase database) {
    return new Context(app, database, producer, trace, features, stageExtensions, logger);
  }

  public Context withFeatures(final Features features) {
    return new Context(app, database, producer, trace, features, stageExtensions, logger);
  }

  public Context withLogger(final Logger logger) {
    return new Context(app, database, producer, trace, features, stageExtensions, logger);
  }

  public Context withProducer(
      final BiFunction, CompletionStage> producer) {
    return new Context(app, database, producer, trace, features, stageExtensions, logger);
  }

  public Context withStageExtensions(final Map stageExtensions) {
    return new Context(app, database, producer, trace, features, stageExtensions, logger);
  }

  public Context withTrace(final boolean trace) {
    return new Context(app, database, producer, trace, features, stageExtensions, logger);
  }
}