io.camunda.zeebe.broker.SpringBrokerBridge Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zeebe-broker Show documentation
Zeebe Broker
There is a newer version: 8.7.0-alpha1
/*
 * Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH under
 * one or more contributor license agreements. See the NOTICE file distributed
 * with this work for additional information regarding copyright ownership.
 * Licensed under the Camunda License 1.0. You may not use this file
 * except in compliance with the Camunda License 1.0.
 */
package io.camunda.zeebe.broker;

import io.camunda.zeebe.broker.jobstream.JobStreamService;
import io.camunda.zeebe.broker.system.management.BrokerAdminService;
import io.camunda.zeebe.broker.system.monitoring.BrokerHealthCheckService;
import io.camunda.zeebe.gateway.impl.stream.JobStreamClient;
import java.util.Optional;
import java.util.function.Consumer;
import java.util.function.Supplier;
import org.springframework.stereotype.Component;

/**
 * Helper class that allows Spring beans to access information from the Broker code that is not
 * managed by Spring
 */
@Component
public class SpringBrokerBridge {
  private Supplier healthCheckServiceSupplier;
  private Supplier adminServiceSupplier;
  private Supplier jobStreamServiceSupplier;
  private Supplier jobStreamClientSupplier;

  private Consumer shutdownHelper;

  public void registerBrokerHealthCheckServiceSupplier(
      final Supplier healthCheckServiceSupplier) {
    this.healthCheckServiceSupplier = healthCheckServiceSupplier;
  }

  public Optional getBrokerHealthCheckService() {
    return Optional.ofNullable(healthCheckServiceSupplier).map(Supplier::get);
  }

  public void registerBrokerAdminServiceSupplier(
      final Supplier adminServiceSupplier) {
    this.adminServiceSupplier = adminServiceSupplier;
  }

  public Optional getAdminService() {
    return Optional.ofNullable(adminServiceSupplier).map(Supplier::get);
  }

  public void registerJobStreamClientSupplier(
      final Supplier jobStreamClientSupplier) {
    this.jobStreamClientSupplier = jobStreamClientSupplier;
  }

  public Optional getJobStreamClient() {
    return Optional.ofNullable(jobStreamClientSupplier).map(Supplier::get);
  }

  public void registerJobStreamServiceSupplier(
      final Supplier jobStreamServiceSupplier) {
    this.jobStreamServiceSupplier = jobStreamServiceSupplier;
  }

  public Optional getJobStreamService() {
    return Optional.ofNullable(jobStreamServiceSupplier).map(Supplier::get);
  }

  /**
   * Registers a shutdown helper that can initiate a graceful shutdown of the broker. This will be
   * used when any exceptional cases may need to be handled by shutting down the broker.
   *
   * @param shutdownHelper the shutdown helper
   */
  public void registerShutdownHelper(final Consumer shutdownHelper) {
    this.shutdownHelper = shutdownHelper;
  }

  public void initiateShutdown(final int errorCode) {
    if (shutdownHelper != null) {
      shutdownHelper.accept(errorCode);
    } else {
      System.exit(errorCode);
    }
  }
}