import { PROM_REGISTRY } from "@t9tlai/host-runner-utils";
import { Gauge } from "prom-client";
import { getSuggestionsMetadataRepository } from "./repositories/suggestionsMetadata.repository";

export default function setupMonitoring(): void {
  PROM_REGISTRY.registerMetric(
    new Gauge({
      name: "team_spaces_suggestions_data_count",
      help: "Number of data points (suggestions related) in the database",
      async collect() {
        const count = await getSuggestionsMetadataRepository().count();
        this.set(count);
      },
    })
  );
}