mirror of
https://github.com/ClusterCockpit/cc-backend
synced 2025-06-08 00:23:48 +02:00
75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
// Copyright (C) NHR@FAU, University Erlangen-Nuremberg.
|
|
// All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/ClusterCockpit/cc-backend/internal/repository"
|
|
"github.com/ClusterCockpit/cc-backend/pkg/schema"
|
|
)
|
|
|
|
type Node struct {
|
|
Name string `json:"hostname"`
|
|
States []string `json:"states"`
|
|
}
|
|
|
|
// updateNodeStatesRequest model
|
|
type UpdateNodeStatesRequest struct {
|
|
Nodes []Node `json:"nodes"`
|
|
Cluster string `json:"cluster" example:"fritz"`
|
|
}
|
|
|
|
// this routine assumes that only one of them exists per node
|
|
func determineState(states []string) schema.NodeState {
|
|
for _, state := range states {
|
|
switch strings.ToLower(state) {
|
|
case "allocated":
|
|
return schema.NodeStateAllocated
|
|
case "reserved":
|
|
return schema.NodeStateReserved
|
|
case "idle":
|
|
return schema.NodeStateIdle
|
|
case "down":
|
|
return schema.NodeStateDown
|
|
case "mixed":
|
|
return schema.NodeStateMixed
|
|
}
|
|
}
|
|
|
|
return schema.NodeStateUnknown
|
|
}
|
|
|
|
// updateNodeStates godoc
|
|
// @summary Deliver updated Slurm node states
|
|
// @tags node
|
|
// @description Returns a JSON-encoded list of users.
|
|
// @description Required query-parameter defines if all users or only users with additional special roles are returned.
|
|
// @produce json
|
|
// @param request body UpdateNodeStatesRequest true "Request body containing nodes and their states"
|
|
// @success 200 {array} api.SuccessResponse "Success"
|
|
// @failure 400 {string} string "Bad Request"
|
|
// @failure 401 {string} string "Unauthorized"
|
|
// @failure 403 {string} string "Forbidden"
|
|
// @failure 500 {string} string "Internal Server Error"
|
|
// @security ApiKeyAuth
|
|
// @router /api/nodestats/ [post]
|
|
func (api *RestApi) updateNodeStates(rw http.ResponseWriter, r *http.Request) {
|
|
// Parse request body
|
|
req := UpdateNodeStatesRequest{}
|
|
if err := decode(r.Body, &req); err != nil {
|
|
handleError(fmt.Errorf("parsing request body failed: %w", err), http.StatusBadRequest, rw)
|
|
return
|
|
}
|
|
repo := repository.GetNodeRepository()
|
|
|
|
for _, node := range req.Nodes {
|
|
state := determineState(node.States)
|
|
repo.UpdateNodeState(node.Name, req.Cluster, &state)
|
|
}
|
|
}
|