fix(comp): prevent divide-by-zero error when calculating array size f…

…or sharded targets (#6326)

CHANGELOG.md

@@ -73,6 +73,8 @@ Main (unreleased)
 - Fix an issue with static integrations-next marshaling where non singletons
   would cause `/-/config` to fail to marshal. (@erikbaranowski)
 
+- Fix divide-by-zero issue when sharding targets. (@hainenber) 
+
 ### Other changes
 
 - Removed support for Windows 2012 in line with Microsoft end of life. (@mattdurham)

component/discovery/discovery.go

@@ -43,7 +43,13 @@ func (t *DistributedTargets) Get() []Target {
 		return t.targets
 	}
 
-	res := make([]Target, 0, (len(t.targets)+1)/len(t.cluster.Peers()))
+	peerCount := len(t.cluster.Peers())
+	resCap := (len(t.targets) + 1)
+	if peerCount != 0 {
+		resCap = (len(t.targets) + 1) / peerCount
+	}
+
+	res := make([]Target, 0, resCap)
 
 	for _, tgt := range t.targets {
 		peers, err := t.cluster.Lookup(shard.StringKey(tgt.NonMetaLabels().String()), 1, shard.OpReadWrite)
@@ -53,7 +59,7 @@ func (t *DistributedTargets) Get() []Target {
 			// back to owning the target ourselves.
 			res = append(res, tgt)
 		}
-		if peers[0].Self {
+		if len(peers) == 0 || peers[0].Self {
 			res = append(res, tgt)
 		}
 	}

component/loki/source/podlogs/reconciler.go

@@ -131,7 +131,13 @@ func distributeTargets(c cluster.Cluster, targets []*kubetail.Target) []*kubetai
 		return targets
 	}
 
-	res := make([]*kubetail.Target, 0, (len(targets)+1)/len(c.Peers()))
+	peerCount := len(c.Peers())
+	resCap := len(targets) + 1
+	if peerCount != 0 {
+		resCap = (len(targets) + 1) / peerCount
+	}
+
+	res := make([]*kubetail.Target, 0, resCap)
 
 	for _, target := range targets {
 		peers, err := c.Lookup(shard.StringKey(target.Labels().String()), 1, shard.OpReadWrite)
@@ -141,7 +147,7 @@ func distributeTargets(c cluster.Cluster, targets []*kubetail.Target) []*kubetai
 			// back to owning the target ourselves.
 			res = append(res, target)
 		}
-		if peers[0].Self {
+		if len(peers) == 0 || peers[0].Self {
 			res = append(res, target)
 		}
 	}