Add check for ignored nodes

mindbox-cloud · Feb 5, 2024 · c9c7e24 · c9c7e24
1 parent cde210d
commit c9c7e24
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -167,7 +167,7 @@ func answerForNodes(domain string) []dns.RR {
 	}
 
 	for _, node := range nodes {
-		if node.IsManager || returnWorkers {
+		if (node.IsManager || returnWorkers) && !node.Ignore {
 			rr := new(dns.A)
 			rr.Hdr = dns.RR_Header{Name: domain, Rrtype: dns.TypeA, Class: dns.ClassINET, Ttl: uint32(TTL)}
 			rr.A = net.ParseIP(node.Ip)

diff --git a/swarm/client.go b/swarm/client.go
@@ -13,6 +13,7 @@ type SwarmNode struct {
 	Ip        string
 	Hostname  string
 	IsManager bool
+	Ignore    bool
 	DnsNames  []string
 }
 
@@ -55,11 +56,14 @@ func (client swarmClient) ListActiveNodes() ([]SwarmNode, error) {
 				ip = node.Status.Addr
 			}
 
+			_, ignore := node.Spec.Annotations.Labels["swarmdns-ignore"]
+
 			nodes = append(nodes,
 				SwarmNode{
 					Ip:        ip,
 					Hostname:  getHostname(node),
 					IsManager: node.ManagerStatus != nil,
+					Ignore:    ignore,
 					DnsNames:  getDnsNames(node),
 				})
 		}