mirror of
https://github.com/TECHNOFAB11/zfs-localpv.git
synced 2025-12-12 14:30:12 +01:00
a10dedbd5e
This is an initial scheduler implementation for ZFS Local PV.
* adding scheduler as a configurable option
* adding volumeWeightedScheduler as scheduling logic
The volumeWeightedScheduler will go through all the nodes as per
topology information and it will pick the node which has less
volume provisioned in the given pool.
lets say there are 2 nodes node1 and node2 with below pool configuration :-
```
node1
|
|-----> pool1
| |
| |------> pvc1
| |------> pvc2
|-----> pool2
|------> pvc3
node2
|
|-----> pool1
| |
| |------> pvc4
|-----> pool2
|------> pvc5
|------> pvc6
```
So if application is using pool1 as shown in the below storage class, then ZFS driver will schedule it on node2 as it has one volume as compared to node1 which has 2 volumes in pool1.
```yaml
kind: StorageClass
apiVersion: storage.k8s.io/v1
metadata:
name: openebs-zfspv
provisioner: zfs.csi.openebs.io
parameters:
blocksize: "4k"
compression: "on"
dedup: "on"
thinprovision: "yes"
poolname: "pool1"
```
So if application is using pool2 as shown in the below storage class, then ZFS driver will schedule it on node1 as it has one volume only as compared node2 which has 2 volumes in pool2.
```yaml
kind: StorageClass
apiVersion: storage.k8s.io/v1
metadata:
name: openebs-zfspv
provisioner: zfs.csi.openebs.io
parameters:
blocksize: "4k"
compression: "on"
dedup: "on"
thinprovision: "yes"
poolname: "pool2"
```
In case of same number of volumes on all the nodes for the given pool, it can pick any node and schedule the PV on that.
Signed-off-by: Pawan <pawan@mayadata.io>
96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
/*
|
|
Copyright © 2019 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 driver
|
|
|
|
import (
|
|
"github.com/Sirupsen/logrus"
|
|
"math"
|
|
|
|
"github.com/container-storage-interface/spec/lib/go/csi"
|
|
"github.com/openebs/zfs-localpv/pkg/builder"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
|
|
zvol "github.com/openebs/zfs-localpv/pkg/zfs"
|
|
)
|
|
|
|
// scheduling algorithm constants
|
|
const (
|
|
// pick the node where less volumes are provisioned for the given pool
|
|
// this will be the default scheduler when none provided
|
|
VolumeWeighted = "VolumeWeighted"
|
|
)
|
|
|
|
// volumeWeightedScheduler goes through all the pools on the nodes mentioned
|
|
// in the topology and picks the node which has less volume on
|
|
// the given zfs pool.
|
|
func volumeWeightedScheduler(topo *csi.TopologyRequirement, pool string) string {
|
|
var selected string
|
|
|
|
zvlist, err := builder.NewKubeclient().
|
|
WithNamespace(zvol.OpenEBSNamespace).
|
|
List(metav1.ListOptions{})
|
|
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
volmap := map[string]int{}
|
|
|
|
// create the map of the volume count
|
|
// for the given pool
|
|
for _, zv := range zvlist.Items {
|
|
if zv.Spec.PoolName == pool {
|
|
volmap[zv.Spec.OwnerNodeID]++
|
|
}
|
|
}
|
|
|
|
var numVol int = math.MaxInt32
|
|
|
|
// schedule it on the node which has less
|
|
// number of volume for the given pool
|
|
for _, prf := range topo.Preferred {
|
|
node := prf.Segments[zvol.ZFSTopologyKey]
|
|
if volmap[node] < numVol {
|
|
selected = node
|
|
numVol = volmap[node]
|
|
}
|
|
}
|
|
return selected
|
|
}
|
|
|
|
// scheduler schedules the PV as per topology constraints for
|
|
// the given zfs pool.
|
|
func scheduler(topo *csi.TopologyRequirement, schld string, pool string) string {
|
|
|
|
if len(topo.Preferred) == 0 {
|
|
logrus.Errorf("topology information not provided")
|
|
return ""
|
|
}
|
|
// if there is a single node, schedule it on that
|
|
if len(topo.Preferred) == 1 {
|
|
return topo.Preferred[0].Segments[zvol.ZFSTopologyKey]
|
|
}
|
|
|
|
switch schld {
|
|
case VolumeWeighted:
|
|
return volumeWeightedScheduler(topo, pool)
|
|
default:
|
|
return volumeWeightedScheduler(topo, pool)
|
|
}
|
|
|
|
return ""
|
|
}
|