package server

import (
	nodeinfo "code.gitlink.org.cn/JCCE/PCM.git/adaptor/pcm_slurm/cgo/src/slurm/nodeinfo"
	pbslurm "code.gitlink.org.cn/JCCE/PCM.git/adaptor/pcm_slurm/gen/idl"
	"context"
)

func ListNodes(ctx context.Context, req *pbslurm.ListNodesReq) (*pbslurm.ListNodesResp, error) {
	nodeList := nodeinfo.Get_all_nodes()

	var resp = pbslurm.ListNodesResp{}
	for _, node := range nodeList.Node_list {
		nodeInfoResult := pbslurm.NodeInfo{}
		nodeInfoResult.Cpus = int32(node.Cpus)
		nodeInfoResult.Boards = int32(node.Boards)
		nodeInfoResult.RealMemory = int32(node.Real_memory)
		nodeInfoResult.Sockets = int32(node.Sockets)
		nodeInfoResult.Threads = int32(node.Threads)
		resp.NodeInfos = append(resp.NodeInfos, &nodeInfoResult)
	}

	return &resp, nil
}