From 25123515b40e84c8239088cad4482d7b9fb1767e Mon Sep 17 00:00:00 2001 From: Christian Simon Date: Sat, 2 Dec 2017 18:19:50 +0000 Subject: Make add/updateHost a bit more resilient * provide a updateOrAddHost method that first will try to update an existing one, before adding a host entry --- libvirt/utils_libvirt.go | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'libvirt/utils_libvirt.go') diff --git a/libvirt/utils_libvirt.go b/libvirt/utils_libvirt.go index 22816769..0b1da1ef 100644 --- a/libvirt/utils_libvirt.go +++ b/libvirt/utils_libvirt.go @@ -46,6 +46,15 @@ func updateHost(n *libvirt.Network, ip, mac, name string) error { return n.Update(libvirt.NETWORK_UPDATE_COMMAND_MODIFY, libvirt.NETWORK_SECTION_IP_DHCP_HOST, -1, xmlDesc, libvirt.NETWORK_UPDATE_AFFECT_CURRENT) } +// Tries to update first, if that fails, it will add it +func updateOrAddHost(n *libvirt.Network, ip, mac, name string) error { + err := updateHost(n, ip, mac, name) + if virErr, ok := err.(libvirt.Error); ok && virErr.Code == libvirt.ERR_OPERATION_INVALID && virErr.Domain == libvirt.FROM_NETWORK { + return addHost(n, ip, mac, name) + } + return err +} + func getHostArchitecture(virConn *libvirt.Connect) (string, error) { type HostCapabilities struct { XMLName xml.Name `xml:"capabilities"` -- cgit v1.2.3