feat: add sync data podmonitor

GreptimeTeam · Jul 10, 2023 · d769009 · d769009
1 parent 56b3e04
commit d769009
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/controllers/greptimedbcluster/deployers/datanode.go b/controllers/greptimedbcluster/deployers/datanode.go
@@ -244,6 +244,30 @@ func (b *datanodeBuilder) BuildStatefulSet() deployer.Builder {
 	return b
 }
 
+func (b *datanodeBuilder) BuildPodMonitor() deployer.Builder {
+	if b.Err != nil {
+		return b
+	}
+
+	if b.Cluster.Spec.Datanode == nil {
+		return b
+	}
+
+	if b.Cluster.Spec.PrometheusMonitor == nil || !b.Cluster.Spec.PrometheusMonitor.Enabled {
+		return b
+	}
+
+	pm, err := b.GeneratePodMonitor()
+	if err != nil {
+		b.Err = err
+		return b
+	}
+
+	b.Objects = append(b.Objects, pm)
+
+	return b
+}
+
 func (b *datanodeBuilder) generateMainContainerArgs() []string {
 	return []string{
 		"datanode", "start",