mirror of
https://github.com/TECHNOFAB11/zfs-localpv.git
synced 2025-12-12 14:30:12 +01:00
de9b302083
This commit adds the support for use to specify custom labels to the kubernetes nodes and use them in the allowedToplogoies section of the StorageClass. Few notes: - This PR depends on the CSI driver's capability to support custom topology keys. - label on the nodes should be added first and then deploy the driver to make it aware of all the labels that node has. If labels are added after ZFS-LocalPV driver has been deployed, a restart all the node csi driver agents is required so that the driver can pick the labels and add them as supported topology keys. - if storageclass is using Immediate binding mode and topology key is not mentioned then all the nodes should be labeled using same key, that means: - same key should be present on all nodes, nodes can have different values for those keys. - If nodes are labeled with different keys i.e. some nodes are having different keys, then ZFSPV's default scheduler can not effictively do the volume count based scheduling. In this case the CSI provisioner will pick keys from any random node and then prepare the preferred topology list using the nodes which has those keys defined. And ZFSPV scheduler will schedule the PV among those nodes only. Signed-off-by: Pawan <pawan@mayadata.io>
103 lines
2.9 KiB
Go
103 lines
2.9 KiB
Go
/*
|
|
Copyright 2020 The OpenEBS Authors
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package v1alpha1
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
// NodeGetter abstracts fetching of Node details from kubernetes cluster
|
|
type NodeGetter interface {
|
|
Get(name string, options metav1.GetOptions) (*corev1.Node, error)
|
|
}
|
|
|
|
// NodeLister abstracts fetching of Nodes from kubernetes cluster
|
|
type NodeLister interface {
|
|
List(options metav1.ListOptions) (*corev1.NodeList, error)
|
|
}
|
|
type node struct{}
|
|
|
|
func Node() *node {
|
|
return &node{}
|
|
}
|
|
|
|
// Get returns a node instance from kubernetes cluster
|
|
func (n *node) Get(name string, options metav1.GetOptions) (*corev1.Node, error) {
|
|
cs, err := Clientset().Get()
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get node: %s", name)
|
|
} else {
|
|
return cs.CoreV1().Nodes().Get(name, options)
|
|
}
|
|
}
|
|
|
|
// List returns a slice of Nodes registered in a Kubernetes cluster
|
|
func (n *node) List(options metav1.ListOptions) (*corev1.NodeList, error) {
|
|
cs, err := Clientset().Get()
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get nodes")
|
|
} else {
|
|
return cs.CoreV1().Nodes().List(options)
|
|
}
|
|
}
|
|
|
|
// NumberOfNodes returns the number of nodes registered in a Kubernetes cluster
|
|
func NumberOfNodes() (int, error) {
|
|
n := Node()
|
|
nodes, err := n.List(metav1.ListOptions{})
|
|
if err != nil {
|
|
return 0, errors.Wrapf(err, "failed to get the number of nodes")
|
|
} else {
|
|
return len(nodes.Items), nil
|
|
}
|
|
}
|
|
|
|
// GetNode returns a node instance from kubernetes cluster
|
|
func GetNode(name string) (*corev1.Node, error) {
|
|
n := Node()
|
|
node, err := n.Get(name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get node")
|
|
} else {
|
|
return node, nil
|
|
}
|
|
}
|
|
|
|
// ListNodes returns list of node instance from kubernetes cluster
|
|
func ListNodes(options metav1.ListOptions) (*corev1.NodeList, error) {
|
|
n := Node()
|
|
nodelist, err := n.List(options)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to list node")
|
|
} else {
|
|
return nodelist, nil
|
|
}
|
|
}
|
|
|
|
// GetOSAndKernelVersion gets us the OS,Kernel version
|
|
func GetOSAndKernelVersion() (string, error) {
|
|
nodes := Node()
|
|
// get a single node
|
|
firstNode, err := nodes.List(metav1.ListOptions{Limit: 1})
|
|
if err != nil {
|
|
return "unknown, unknown", errors.Wrapf(err, "failed to get the os kernel/arch")
|
|
}
|
|
nodedetails := firstNode.Items[0].Status.NodeInfo
|
|
return nodedetails.OSImage + ", " + nodedetails.KernelVersion, nil
|
|
}
|