io.camunda.zeebe.broker.bootstrap.BrokerAdminServiceStep 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.bootstrap;

import io.camunda.zeebe.broker.system.management.BrokerAdminServiceImpl;
import io.camunda.zeebe.scheduler.ConcurrencyControl;
import io.camunda.zeebe.scheduler.future.ActorFuture;

final class BrokerAdminServiceStep extends AbstractBrokerStartupStep {

  @Override
  public String getName() {
    return "Broker Admin Interface";
  }

  @Override
  void startupInternal(
      final BrokerStartupContext brokerStartupContext,
      final ConcurrencyControl concurrencyControl,
      final ActorFuture startupFuture) {

    final var adminService = new BrokerAdminServiceImpl(brokerStartupContext.getPartitionManager());

    final var submitActorFuture =
        brokerStartupContext.getActorSchedulingService().submitActor(adminService);

    concurrencyControl.runOnCompletion(
        submitActorFuture,
        (ok, error) -> {
          if (error != null) {
            startupFuture.complete(brokerStartupContext);
            return;
          }

          forwardExceptions(
              () -> {
                brokerStartupContext
                    .getSpringBrokerBridge()
                    .registerBrokerAdminServiceSupplier(() -> adminService);

                brokerStartupContext.setBrokerAdminService(adminService);
                startupFuture.complete(brokerStartupContext);
              },
              startupFuture);
        });
  }

  @Override
  void shutdownInternal(
      final BrokerStartupContext brokerShutdownContext,
      final ConcurrencyControl concurrencyControl,
      final ActorFuture shutdownFuture) {

    final var adminService = brokerShutdownContext.getBrokerAdminService();

    if (adminService == null) {
      shutdownFuture.complete(brokerShutdownContext);
      return;
    }
    final var closeFuture = adminService.closeAsync();

    concurrencyControl.runOnCompletion(
        closeFuture,
        (ok, error) -> {
          if (error != null) {
            shutdownFuture.completeExceptionally(error);
            return;
          }

          forwardExceptions(
              () -> {
                brokerShutdownContext.setBrokerAdminService(null);

                shutdownFuture.complete(brokerShutdownContext);
              },
              shutdownFuture);
        });
  }
}