Handle cases of missing node name when (re)adding collector

Signed-off-by: Matej Gera <[email protected]>

cmd/otel-allocator/allocation/per_node.go

@@ -86,6 +86,12 @@ func (allocator *perNodeAllocator) handleCollectors(diff diff.Changes[*Collector
 	for _, i := range diff.Additions() {
 		allocator.collectors[i.NodeName] = NewCollector(i.Name, i.NodeName)
 	}
+
+	// For a case where a collector is removed and added back, we need
+	// to re-allocate any already existing targets.
+	for _, item := range allocator.targetItems {
+		allocator.addTargetToTargetItems(item)
+	}
 }
 
 // SetTargets accepts a list of targets that will be used to make

cmd/otel-allocator/collector/collector.go

@@ -128,6 +128,11 @@ func runWatch(ctx context.Context, k *Client, c <-chan watch.Event, collectorMap
 				return ""
 			}
 
+			if pod.Spec.NodeName == "" {
+				k.log.Info("Node name is missing from the spec. Restarting watch routine")
+				return ""
+			}
+
 			switch event.Type { //nolint:exhaustive
 			case watch.Added:
 				collectorMap[pod.Name] = allocation.NewCollector(pod.Name, pod.Spec.NodeName)