mirror of
https://github.com/ClusterCockpit/cc-backend
synced 2025-10-31 07:55:06 +01:00
90 lines
2.8 KiB
Go
90 lines
2.8 KiB
Go
// Copyright (C) NHR@FAU, University Erlangen-Nuremberg.
|
|
// All rights reserved. This file is part of cc-backend.
|
|
// 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"
|
|
"time"
|
|
|
|
"github.com/ClusterCockpit/cc-backend/internal/repository"
|
|
"github.com/ClusterCockpit/cc-lib/schema"
|
|
)
|
|
|
|
type Node struct {
|
|
Hostname string `json:"hostname"`
|
|
States []string `json:"states"`
|
|
CpusAllocated int `json:"cpusAllocated"`
|
|
MemoryAllocated int `json:"memoryAllocated"`
|
|
GpusAllocated int `json:"gpusAllocated"`
|
|
JobsRunning int `json:"jobsRunning"`
|
|
}
|
|
|
|
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.SchedulerState {
|
|
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 Nodestates
|
|
// @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 {object} api.DefaultApiResponse "Success message"
|
|
// @failure 400 {object} api.ErrorResponse "Bad Request"
|
|
// @failure 401 {object} api.ErrorResponse "Unauthorized"
|
|
// @failure 403 {object} api.ErrorResponse "Forbidden"
|
|
// @failure 500 {object} api.ErrorResponse "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)
|
|
nodeState := schema.NodeState{
|
|
TimeStamp: time.Now().Unix(), NodeState: state,
|
|
CpusAllocated: node.CpusAllocated,
|
|
MemoryAllocated: node.MemoryAllocated,
|
|
GpusAllocated: node.GpusAllocated,
|
|
HealthState: schema.MonitoringStateFull,
|
|
JobsRunning: node.JobsRunning,
|
|
}
|
|
|
|
repo.UpdateNodeState(node.Hostname, req.Cluster, &nodeState)
|
|
}
|
|
}
|