ClusterCockpit/cc-backend
1
0
Fork 0
mirror of https://github.com/ClusterCockpit/cc-backend synced 2025-10-03 13:04:32 +02:00
Code Issues Packages Releases Activity

Update nodestate API and db adapter

Browse Source
This commit is contained in:
Jan Eitzinger
2025-09-28 08:26:44 +02:00
parent 015583f1cd
commit beb92967e5
4 changed files with 38 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
internal/repository/node.go
View File

@@ -144,8 +144,10 @@ func (r *NodeRepository) GetNode(id int64, withMeta bool) (*schema.Node, error)
}
const NamedNodeInsert string = `
INSERT INTO node (hostname, cluster, subcluster, node_state, health_state)
VALUES (:hostname, :cluster, :subcluster, :node_state, :health_state);`
INSERT INTO node (time_stamp, hostname, cluster, subcluster, node_state, health_state,
cpus_allocated, cpus_total, memory_allocated, memory_total, gpus_allocated, gpus_total)
VALUES (:time_stamp, :hostname, :cluster, :subcluster, :node_state, :health_state,
:cpus_allocated, :cpus_total, :memory_allocated, :memory_total, :gpus_allocated, :gpus_total);`
func (r *NodeRepository) AddNode(node *schema.Node) (int64, error) {
var err error
@@ -164,6 +166,24 @@ func (r *NodeRepository) AddNode(node *schema.Node) (int64, error) {
return node.ID, nil
}
func (r *NodeRepository) InsertNodeState(nodeState *schema.Node) error {
subcluster, err := archive.GetSubClusterByNode(nodeState.Cluster, nodeState.Hostname)
if err != nil {
cclog.Errorf("Error while getting subcluster for node '%s' in cluster '%s': %v", nodeState.Hostname, nodeState.Cluster, err)
return err
}
nodeState.SubCluster = subcluster
_, err = r.DB.NamedExec(NamedNodeInsert, nodeState)
if err != nil {
cclog.Errorf("Error while adding node '%v' to database", nodeState.Hostname)
return err
}
return nil
}
func (r *NodeRepository) UpdateNodeState(hostname string, cluster string, nodeState *schema.NodeState) error {
var id int64
if err := sq.Select("id").From("node").