forked from JointCloud/pcm-coordinator
110 lines
3.9 KiB
Go
110 lines
3.9 KiB
Go
package tianhe
|
|
|
|
/*
|
|
//#cgo LDFLAGS: -lslurm
|
|
#include<stdlib.h>
|
|
#include<slurm/slurm.h>
|
|
|
|
|
|
|
|
struct partition_info_msg *get_partition_info(){
|
|
struct partition_info_msg* partition_buffer;
|
|
if( slurm_load_partitions ((time_t) NULL,
|
|
&partition_buffer, SHOW_ALL))
|
|
return NULL;
|
|
return partition_buffer;
|
|
}
|
|
struct partition_info* partition_from_list(struct partition_info_msg *list, int i){
|
|
return &list->partition_array[i];
|
|
}
|
|
void free_partition_buffer(void* buffer){
|
|
|
|
slurm_free_partition_info_msg ((struct partition_info_msg*)buffer);
|
|
}
|
|
*/
|
|
import "C"
|
|
import (
|
|
pbslurm "code.gitlink.org.cn/JCCE/PCM.git/adaptor/pcm_slurm/gen/idl"
|
|
)
|
|
|
|
func Partition_info_convert_c_to_go(c_struct *C.struct_partition_info) pbslurm.PartitionInfo {
|
|
var go_struct pbslurm.PartitionInfo
|
|
|
|
go_struct.AllowAllocNodes = C.GoString(c_struct.allow_alloc_nodes)
|
|
go_struct.AllowGroups = C.GoString(c_struct.allow_groups)
|
|
go_struct.Alternate = C.GoString(c_struct.alternate)
|
|
go_struct.CrType = uint32(c_struct.cr_type)
|
|
go_struct.DefMemPerCpu = uint64(c_struct.def_mem_per_cpu)
|
|
go_struct.DefaultTime = uint32(c_struct.default_time)
|
|
go_struct.Flags = uint32(c_struct.flags)
|
|
go_struct.GraceTime = uint32(c_struct.grace_time)
|
|
go_struct.MaxCpusPerNode = uint32(c_struct.max_cpus_per_node)
|
|
go_struct.MaxMemPerCpu = uint64(c_struct.max_mem_per_cpu)
|
|
go_struct.MaxNodes = uint32(c_struct.max_nodes)
|
|
go_struct.MaxShare = uint32(c_struct.max_share)
|
|
go_struct.MaxTime = uint32(c_struct.max_time)
|
|
go_struct.MinNodes = uint32(c_struct.min_nodes)
|
|
go_struct.Name = C.GoString(c_struct.name)
|
|
//go_struct.Node_inx = int32(C.int32_ptr(c_struct.node_inx))
|
|
/*t := C.find_node_inx(c_struct.node_inx)
|
|
fmt.Printf("%d", t)
|
|
go_struct.Node_inx = make([]int32, t, t)
|
|
for i := int32(0); i < int32(t); i++ {
|
|
go_struct.Node_inx[i] = int32(C.int32_ptr(c_struct.node_inx, C.int(i)))
|
|
|
|
}*/
|
|
go_struct.Nodes = C.GoString(c_struct.nodes)
|
|
go_struct.PreemptMode = uint32(c_struct.preempt_mode)
|
|
go_struct.StateUp = uint32(c_struct.state_up)
|
|
go_struct.TotalCpus = uint32(c_struct.total_cpus)
|
|
go_struct.TotalNodes = uint32(c_struct.total_nodes)
|
|
return go_struct
|
|
}
|
|
|
|
type PartitionInfoMsg struct {
|
|
LastUpdate int64
|
|
RecordCount uint32
|
|
PartitionInfoList []pbslurm.PartitionInfo
|
|
}
|
|
|
|
func Get_partitions() PartitionInfoMsg {
|
|
var go_partition_buffer PartitionInfoMsg
|
|
c_partition_buffer := C.get_partition_info()
|
|
if c_partition_buffer == nil {
|
|
go_partition_buffer.LastUpdate = int64(0)
|
|
go_partition_buffer.RecordCount = uint32(0)
|
|
return go_partition_buffer
|
|
}
|
|
go_partition_buffer.LastUpdate = int64(c_partition_buffer.last_update)
|
|
go_partition_buffer.RecordCount = uint32(c_partition_buffer.record_count)
|
|
go_partition_buffer.PartitionInfoList = make([]pbslurm.PartitionInfo, c_partition_buffer.record_count, c_partition_buffer.record_count)
|
|
for i := uint32(0); i < go_partition_buffer.RecordCount; i++ {
|
|
partition := C.partition_from_list(c_partition_buffer, C.int(i))
|
|
go_partition := Partition_info_convert_c_to_go(partition)
|
|
go_partition_buffer.PartitionInfoList[i] = go_partition
|
|
}
|
|
C.slurm_free_partition_info_msg(c_partition_buffer)
|
|
|
|
return go_partition_buffer
|
|
}
|
|
|
|
func GetPartitionsInfo() PartitionInfoMsg {
|
|
var goPartitionBuffer PartitionInfoMsg
|
|
cPartitionBuffer := C.get_partition_info()
|
|
goPartitionBuffer.RecordCount = uint32(cPartitionBuffer.record_count)
|
|
goPartitionBuffer.PartitionInfoList = make([]pbslurm.PartitionInfo, cPartitionBuffer.record_count, cPartitionBuffer.record_count)
|
|
|
|
for i := uint32(0); i < goPartitionBuffer.RecordCount; i++ {
|
|
partition := C.partition_from_list(cPartitionBuffer, C.int(i))
|
|
goPartition := PartitionDescriptorConvertCToGo(partition)
|
|
goPartitionBuffer.PartitionInfoList[i] = goPartition
|
|
}
|
|
return goPartitionBuffer
|
|
}
|
|
|
|
func PartitionDescriptorConvertCToGo(cStruct *C.struct_partition_info) pbslurm.PartitionInfo {
|
|
var goStruct pbslurm.PartitionInfo
|
|
goStruct.Name = C.GoString(cStruct.name)
|
|
return goStruct
|
|
}
|