From 711e5afa6025a4744f825e45279110a278799824 Mon Sep 17 00:00:00 2001
From: Riccardo Forina <rforina@redhat.com>
Date: Thu, 21 Dec 2023 16:24:47 +0100
Subject: [PATCH] Add distribution chart

---
 .../[kafkaId]/nodes/DistributionChart.tsx     | 142 ++++++++++++++++++
 .../[locale]/kafka/[kafkaId]/nodes/page.tsx   |  16 +-
 2 files changed, 155 insertions(+), 3 deletions(-)
 create mode 100644 ui/app/[locale]/kafka/[kafkaId]/nodes/DistributionChart.tsx

diff --git a/ui/app/[locale]/kafka/[kafkaId]/nodes/DistributionChart.tsx b/ui/app/[locale]/kafka/[kafkaId]/nodes/DistributionChart.tsx
new file mode 100644
index 000000000..a9c79d30c
--- /dev/null
+++ b/ui/app/[locale]/kafka/[kafkaId]/nodes/DistributionChart.tsx
@@ -0,0 +1,142 @@
+"use client";
+
+import { useChartWidth } from "@/app/[locale]/kafka/[kafkaId]/overview/useChartWidth";
+import {
+  Chart,
+  ChartAxis,
+  ChartBar,
+  ChartLegend,
+  ChartStack,
+  ChartVoronoiContainer,
+} from "@/libs/patternfly/react-charts";
+import {
+  Card,
+  CardBody,
+  CardHeader,
+  CardTitle,
+  ToggleGroup,
+  ToggleGroupItem,
+  Tooltip,
+} from "@/libs/patternfly/react-core";
+import { HelpIcon } from "@/libs/patternfly/react-icons";
+import { useState } from "react";
+
+export function DistributionChart({
+  data,
+}: {
+  data: Record<string, { leaders: number; followers: number }>;
+}) {
+  const [containerRef, width] = useChartWidth();
+  const [includeLeaders, setIncludeLeaders] = useState(true);
+  const [includeFollowers, setIncludeFollowers] = useState(true);
+  const label =
+    includeFollowers || includeLeaders
+      ? includeLeaders && includeFollowers
+        ? "total partitions"
+        : includeLeaders
+          ? "leaders"
+          : "followers"
+      : "total partitions";
+  return (
+    <Card className={"pf-v5-u-mb-lg"}>
+      <CardHeader>
+        <CardTitle>
+          Partitions distribution (% of total){" "}
+          <Tooltip
+            content={
+              "The percentage distribution of partitions across brokers in the cluster. Consider rebalancing if the distribution is uneven to ensure efficient resource utilization."
+            }
+          >
+            <HelpIcon />
+          </Tooltip>
+        </CardTitle>
+      </CardHeader>
+      <CardBody>
+        <ToggleGroup isCompact aria-label="Compact variant toggle group">
+          <ToggleGroupItem
+            text={`Leaders (${Object.values(data).reduce(
+              (acc, v) => v.leaders + acc,
+              0,
+            )})`}
+            buttonId="toggle-group-compact-1"
+            isSelected={includeLeaders}
+            onChange={() => setIncludeLeaders((v) => !v)}
+          />
+          <ToggleGroupItem
+            text={`Followers (${Object.values(data).reduce(
+              (acc, v) => v.followers + acc,
+              0,
+            )})`}
+            buttonId="toggle-group-compact-2"
+            isSelected={includeFollowers}
+            onChange={() => setIncludeFollowers((v) => !v)}
+          />
+        </ToggleGroup>
+        <div ref={containerRef}>
+          <Chart
+            ariaDesc="Average number of pets"
+            ariaTitle="Stack chart example"
+            containerComponent={
+              <ChartVoronoiContainer
+                labels={({ datum }) => `${datum.name} ${label}: ${datum.y}`}
+                constrainToVisibleArea
+              />
+            }
+            legendOrientation="horizontal"
+            legendPosition="bottom"
+            legendComponent={
+              <ChartLegend
+                orientation={"horizontal"}
+                data={Object.keys(data).flatMap((node) => [
+                  {
+                    name: `Broker ${node} ${label}`,
+                  },
+                ])}
+                itemsPerRow={width > 600 ? 3 : 1}
+              />
+            }
+            height={100}
+            padding={{
+              bottom: 70,
+              left: 0,
+              right: 0, // Adjusted to accommodate legend
+              top: 30,
+            }}
+            width={width}
+          >
+            <ChartAxis
+              style={{
+                axis: { stroke: "transparent" },
+                ticks: { stroke: "transparent" },
+                tickLabels: { fill: "transparent" },
+              }}
+            />
+            <ChartStack>
+              {Object.entries(data).map(([node, data], idx) => (
+                <ChartBar
+                  key={idx}
+                  horizontal={true}
+                  barWidth={15}
+                  data={[
+                    {
+                      name: `Broker ${node}`,
+                      x: "x",
+                      y:
+                        includeFollowers || includeLeaders
+                          ? includeLeaders && includeFollowers
+                            ? data.leaders + data.followers
+                            : includeLeaders
+                              ? data.leaders
+                              : data.followers
+                          : data.leaders + data.followers,
+                    },
+                  ]}
+                />
+              ))}
+            </ChartStack>
+          </Chart>
+        </div>
+      </CardBody>
+    </Card>
+  );
+}
diff --git a/ui/app/[locale]/kafka/[kafkaId]/nodes/page.tsx b/ui/app/[locale]/kafka/[kafkaId]/nodes/page.tsx
index e18e7a20f..c44baa9e8 100644
--- a/ui/app/[locale]/kafka/[kafkaId]/nodes/page.tsx
+++ b/ui/app/[locale]/kafka/[kafkaId]/nodes/page.tsx
@@ -1,5 +1,6 @@
 import { getKafkaClusterKpis } from "@/api/kafka/actions";
 import { KafkaParams } from "@/app/[locale]/kafka/[kafkaId]/kafka.params";
+import { DistributionChart } from "@/app/[locale]/kafka/[kafkaId]/nodes/DistributionChart";
 import {
   Node,
   NodesTable,
@@ -37,9 +38,18 @@ export default async function NodesPage({ params }: { params: KafkaParams }) {
     };
   });
 
+  const data = Object.fromEntries(
+    nodes.map((n) => {
+      return [n.id, { followers: n.followers, leaders: n.leaders }];
+    }),
+  );
+
   return (
-    <PageSection isFilled>
-      <NodesTable nodes={nodes} />
-    </PageSection>
+    <>
+      <PageSection isFilled>
+        <DistributionChart data={data} />
+        <NodesTable nodes={nodes} />
+      </PageSection>
+    </>
   );
 }