forked from k8snetworkplumbingwg/sriov-network-device-plugin
/
rdma.go
43 lines (36 loc) · 1.05 KB
/
rdma.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package resources
import (
"github.com/Mellanox/rdmamap"
pluginapi "k8s.io/kubernetes/pkg/kubelet/apis/deviceplugin/v1beta1"
"github.com/intel/sriov-network-device-plugin/pkg/types"
)
type rdmaSpec struct {
isSupportRdma bool
deviceSpec []*pluginapi.DeviceSpec
}
// NewRdmaSpec returns the RdmaSpec
func NewRdmaSpec(pciAddrs string) types.RdmaSpec {
deviceSpec := make([]*pluginapi.DeviceSpec, 0)
isSupportRdma := false
rdmaResources := rdmamap.GetRdmaDevicesForPcidev(pciAddrs)
if len(rdmaResources) > 0 {
isSupportRdma = true
for _, res := range rdmaResources {
resRdmaDevices := rdmamap.GetRdmaCharDevices(res)
for _, rdmaDevice := range resRdmaDevices {
deviceSpec = append(deviceSpec, &pluginapi.DeviceSpec{
HostPath: rdmaDevice,
ContainerPath: rdmaDevice,
Permissions: "rwm",
})
}
}
}
return &rdmaSpec{isSupportRdma: isSupportRdma, deviceSpec: deviceSpec}
}
func (r *rdmaSpec) IsRdma() bool {
return r.isSupportRdma
}
func (r *rdmaSpec) GetRdmaDeviceSpec() []*pluginapi.DeviceSpec {
return r.deviceSpec
}