Function subscribeHandler

packages/joint-react/src/utils/subscriber-handler.ts:15–41

(
  beforeSubscribe?: () => Set<dia.Cell.ID> | undefined
)

 * @group utils
 */
export function subscribeHandler(
  beforeSubscribe?: () => Set<dia.Cell.ID> | undefined
): SubscribeHandler {
  let isScheduled = false;
  const subscribers = new Set<(changedIds?: Set<dia.Cell.ID>) => void>();

  return {
    subscribe: (onStoreChange: (changedIds?: Set<dia.Cell.ID>) => void) => {
      subscribers.add(onStoreChange);
      return () => {
        subscribers.delete(onStoreChange);
      };
    },
    notifySubscribers: () => {
      if (!isScheduled) {
        isScheduled = true;
        requestAnimationFrame(() => {
          const changedIds = beforeSubscribe?.();
          for (const subscriber of subscribers) {
            subscriber(changedIds);
          }
          isScheduled = false;
        });
      }
    },
  };
}