forked from DGNum/infrastructure
fix(netbox-agent): batch requests filtering on interfaces
Re-enable the service on vault01 now that it works
This commit is contained in:
parent
3494f609bb
commit
2c88c2bad7
4 changed files with 45 additions and 2 deletions
|
@ -11,7 +11,6 @@ lib.extra.mkConfig {
|
||||||
"k-radius"
|
"k-radius"
|
||||||
"networking"
|
"networking"
|
||||||
"ups"
|
"ups"
|
||||||
"netbox-agent-disable"
|
|
||||||
];
|
];
|
||||||
|
|
||||||
extraConfig = {
|
extraConfig = {
|
||||||
|
|
|
@ -1 +0,0 @@
|
||||||
{ dgn-netbox-agent.enable = false; }
|
|
35
modules/dgn-netbox-agent/01-batch-filter.patch
Normal file
35
modules/dgn-netbox-agent/01-batch-filter.patch
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
diff --git a/netbox_agent/network.py b/netbox_agent/network.py
|
||||||
|
index 673dfc1..8ef60aa 100644
|
||||||
|
--- a/netbox_agent/network.py
|
||||||
|
+++ b/netbox_agent/network.py
|
||||||
|
@@ -1,7 +1,7 @@
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
-from itertools import chain
|
||||||
|
+from itertools import chain, islice
|
||||||
|
|
||||||
|
import netifaces
|
||||||
|
from netaddr import IPAddress
|
||||||
|
@@ -413,11 +413,17 @@ class Network(object):
|
||||||
|
|
||||||
|
# delete IP on netbox that are not known on this server
|
||||||
|
if len(nb_nics):
|
||||||
|
- netbox_ips = nb.ipam.ip_addresses.filter(
|
||||||
|
- **{self.intf_type: [x.id for x in nb_nics]}
|
||||||
|
- )
|
||||||
|
+
|
||||||
|
+ def batched(it, n):
|
||||||
|
+ while batch := tuple(islice(it, n)):
|
||||||
|
+ yield batch
|
||||||
|
+
|
||||||
|
+ netbox_ips = []
|
||||||
|
+ for ids in batched((x.id for x in nb_nics), 25):
|
||||||
|
+ netbox_ips += list(
|
||||||
|
+ nb.ipam.ip_addresses.filter(**{self.intf_type: ids})
|
||||||
|
+ )
|
||||||
|
|
||||||
|
- netbox_ips = list(netbox_ips)
|
||||||
|
all_local_ips = list(chain.from_iterable([
|
||||||
|
x['ip'] for x in self.nics if x['ip'] is not None
|
||||||
|
]))
|
|
@ -14,9 +14,19 @@ in
|
||||||
default = true;
|
default = true;
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
config = lib.mkIf config.dgn-netbox-agent.enable {
|
config = lib.mkIf config.dgn-netbox-agent.enable {
|
||||||
|
nixpkgs.overlays = [
|
||||||
|
(_: super: {
|
||||||
|
netbox-agent = super.netbox-agent.overrideAttrs (old: {
|
||||||
|
patches = (old.patches or [ ]) ++ [ ./01-batch-filter.patch ];
|
||||||
|
});
|
||||||
|
})
|
||||||
|
];
|
||||||
|
|
||||||
services.netbox-agent = {
|
services.netbox-agent = {
|
||||||
enable = true;
|
enable = true;
|
||||||
|
|
||||||
settings = {
|
settings = {
|
||||||
netbox.url = "https://netbox.dgnum.eu/";
|
netbox.url = "https://netbox.dgnum.eu/";
|
||||||
network.ignore_interfaces = "(lo|dummy.*|docker.*|podman.*)";
|
network.ignore_interfaces = "(lo|dummy.*|docker.*|podman.*)";
|
||||||
|
|
Loading…
Reference in a new issue