org.openmetadata.service.events.subscription.EventsSubscriptionRegistry Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of openmetadata-service Show documentation

There is a newer version: 1.5.11

package org.openmetadata.service.events.subscription;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import org.openmetadata.schema.type.FilterResourceDescriptor;
import org.openmetadata.service.exception.CatalogExceptionMessage;

public class EventsSubscriptionRegistry {

  private static final List ENTITY_NOTIFICATION_DESCRIPTORS =
      new ArrayList<>();

  private static final List OBSERVABILITY_DESCRIPTORS = new ArrayList<>();

  private EventsSubscriptionRegistry() {}

  public static void initialize(
      List entityNotificationDescriptor,
      List observabilityDescriptors) {
    // Entity notification descriptors
    ENTITY_NOTIFICATION_DESCRIPTORS.clear();
    ENTITY_NOTIFICATION_DESCRIPTORS.addAll(entityNotificationDescriptor);
    ENTITY_NOTIFICATION_DESCRIPTORS.sort(Comparator.comparing(FilterResourceDescriptor::getName));

    // Observability descriptors
    OBSERVABILITY_DESCRIPTORS.clear();
    OBSERVABILITY_DESCRIPTORS.addAll(observabilityDescriptors);
    OBSERVABILITY_DESCRIPTORS.sort(Comparator.comparing(FilterResourceDescriptor::getName));
  }

  public static List listEntityNotificationDescriptors() {
    return Collections.unmodifiableList(ENTITY_NOTIFICATION_DESCRIPTORS);
  }

  public static List listObservabilityDescriptors() {
    return Collections.unmodifiableList(OBSERVABILITY_DESCRIPTORS);
  }

  public static FilterResourceDescriptor getEntityNotificationDescriptor(String resourceType) {
    FilterResourceDescriptor rd =
        ENTITY_NOTIFICATION_DESCRIPTORS.stream()
            .filter(r -> r.getName().equalsIgnoreCase(resourceType))
            .findAny()
            .orElse(null);
    if (rd == null) {
      throw new IllegalArgumentException(
          CatalogExceptionMessage.resourceTypeNotFound(resourceType));
    }
    return rd;
  }

  public static FilterResourceDescriptor getObservabilityDescriptor(String resourceType) {
    FilterResourceDescriptor rd =
        OBSERVABILITY_DESCRIPTORS.stream()
            .filter(r -> r.getName().equalsIgnoreCase(resourceType))
            .findAny()
            .orElse(null);
    if (rd == null) {
      throw new IllegalArgumentException(
          CatalogExceptionMessage.resourceTypeNotFound(resourceType));
    }
    return rd;
  }
}