client: Add GetNetworkLoadBalancerState

Signed-off-by: Stéphane Graber <stgraber@stgraber.org> Sponsored-by: Luizalabs (https://luizalabs.com)
lxc · Sep 25, 2024 · ee2cb9a · ee2cb9a
1 parent f905c77
commit ee2cb9a
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/client/incus_network_load_balancers.go b/client/incus_network_load_balancers.go
@@ -111,3 +111,22 @@ func (r *ProtocolIncus) DeleteNetworkLoadBalancer(networkName string, listenAddr
 
 	return nil
 }
+
+// GetNetworkLoadBalancerState returns a Network load balancer state for the provided network and listen address.
+func (r *ProtocolIncus) GetNetworkLoadBalancerState(networkName string, listenAddress string) (*api.NetworkLoadBalancerState, error) {
+	err := r.CheckExtension("network_load_balancer_state")
+	if err != nil {
+		return nil, err
+	}
+
+	lbState := api.NetworkLoadBalancerState{}
+
+	// Fetch the raw value.
+	u := api.NewURL().Path("networks", networkName, "load-balancers", listenAddress, "state")
+	_, err = r.queryStruct("GET", u.String(), nil, "", &lbState)
+	if err != nil {
+		return nil, err
+	}
+
+	return &lbState, nil
+}
diff --git a/client/interfaces.go b/client/interfaces.go
@@ -215,6 +215,7 @@ type InstanceServer interface {
 	CreateNetworkLoadBalancer(networkName string, forward api.NetworkLoadBalancersPost) error
 	UpdateNetworkLoadBalancer(networkName string, listenAddress string, forward api.NetworkLoadBalancerPut, ETag string) (err error)
 	DeleteNetworkLoadBalancer(networkName string, listenAddress string) (err error)
+	GetNetworkLoadBalancerState(networkName string, listenAddress string) (lbState *api.NetworkLoadBalancerState, err error)
 
 	// Network peer functions ("network_peer" API extension)
 	GetNetworkPeerNames(networkName string) ([]string, error)