incusd/internal: Add rebalance endpoint

Signed-off-by: Piotr Resztak <[email protected]>

Author: presztak

cmd/incusd/api_internal.go

@@ -62,6 +62,7 @@ var apiInternal = []APIEndpoint{
 	internalImageOptimizeCmd,
 	internalImageRefreshCmd,
 	internalRAFTSnapshotCmd,
+	internalRebalanceLoadCmd,
 	internalReadyCmd,
 	internalShutdownCmd,
 	internalSQLCmd,
@@ -147,6 +148,12 @@ var internalBGPStateCmd = APIEndpoint{
 	Get: APIEndpointAction{Handler: internalBGPState, AccessHandler: allowPermission(auth.ObjectTypeServer, auth.EntitlementCanEdit)},
 }
 
+var internalRebalanceLoadCmd = APIEndpoint{
+	Path: "rebalance",
+
+	Get: APIEndpointAction{Handler: internalRebalanceLoad, AccessHandler: allowPermission(auth.ObjectTypeServer, auth.EntitlementCanEdit)},
+}
+
 type internalImageOptimizePost struct {
 	Image api.Image `json:"image" yaml:"image"`
 	Pool  string    `json:"pool"  yaml:"pool"`
@@ -1074,3 +1081,12 @@ func internalBGPState(d *Daemon, r *http.Request) response.Response {
 
 	return response.SyncResponse(true, s.BGP.Debug())
 }
+
+func internalRebalanceLoad(d *Daemon, r *http.Request) response.Response {
+	err := autoRebalanceLoad(context.TODO(), d)
+	if err != nil {
+		return response.SmartError(err)
+	}
+
+	return response.EmptySyncResponse
+}