1 changed files with 152 additions and 0 deletions
@ -0,0 +1,152 @@ |
|||
/* |
|||
Copyright 2016 The Kubernetes 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 main |
|||
|
|||
import ( |
|||
"errors" |
|||
"flag" |
|||
"os" |
|||
"path" |
|||
"time" |
|||
|
|||
"github.com/golang/glog" |
|||
"github.com/kubernetes-incubator/external-storage/lib/controller" |
|||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" |
|||
"k8s.io/apimachinery/pkg/util/wait" |
|||
"k8s.io/client-go/kubernetes" |
|||
"k8s.io/client-go/pkg/api/v1" |
|||
"k8s.io/client-go/rest" |
|||
"syscall" |
|||
) |
|||
|
|||
const ( |
|||
resyncPeriod = 15 * time.Second |
|||
provisionerName = "example.com/hostpath" |
|||
exponentialBackOffOnError = false |
|||
failedRetryThreshold = 5 |
|||
leasePeriod = controller.DefaultLeaseDuration |
|||
retryPeriod = controller.DefaultRetryPeriod |
|||
renewDeadline = controller.DefaultRenewDeadline |
|||
termLimit = controller.DefaultTermLimit |
|||
) |
|||
|
|||
type hostPathProvisioner struct { |
|||
// The directory to create PV-backing directories in
|
|||
pvDir string |
|||
|
|||
// Identity of this hostPathProvisioner, set to node's name. Used to identify
|
|||
// "this" provisioner's PVs.
|
|||
identity string |
|||
} |
|||
|
|||
func NewHostPathProvisioner() controller.Provisioner { |
|||
nodeName := os.Getenv("NODE_NAME") |
|||
if nodeName == "" { |
|||
glog.Fatal("env variable NODE_NAME must be set so that this provisioner can identify itself") |
|||
} |
|||
return &hostPathProvisioner{ |
|||
pvDir: "/tmp/hostpath-provisioner", |
|||
identity: nodeName, |
|||
} |
|||
} |
|||
|
|||
var _ controller.Provisioner = &hostPathProvisioner{} |
|||
|
|||
// Provision creates a storage asset and returns a PV object representing it.
|
|||
func (p *hostPathProvisioner) Provision(options controller.VolumeOptions) (*v1.PersistentVolume, error) { |
|||
path := path.Join(p.pvDir, options.PVName) |
|||
|
|||
if err := os.MkdirAll(path, 0777); err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
pv := &v1.PersistentVolume{ |
|||
ObjectMeta: metav1.ObjectMeta{ |
|||
Name: options.PVName, |
|||
Annotations: map[string]string{ |
|||
"hostPathProvisionerIdentity": p.identity, |
|||
}, |
|||
}, |
|||
Spec: v1.PersistentVolumeSpec{ |
|||
PersistentVolumeReclaimPolicy: options.PersistentVolumeReclaimPolicy, |
|||
AccessModes: options.PVC.Spec.AccessModes, |
|||
Capacity: v1.ResourceList{ |
|||
v1.ResourceName(v1.ResourceStorage): options.PVC.Spec.Resources.Requests[v1.ResourceName(v1.ResourceStorage)], |
|||
}, |
|||
PersistentVolumeSource: v1.PersistentVolumeSource{ |
|||
HostPath: &v1.HostPathVolumeSource{ |
|||
Path: path, |
|||
}, |
|||
}, |
|||
}, |
|||
} |
|||
|
|||
return pv, nil |
|||
} |
|||
|
|||
// Delete removes the storage asset that was created by Provision represented
|
|||
// by the given PV.
|
|||
func (p *hostPathProvisioner) Delete(volume *v1.PersistentVolume) error { |
|||
ann, ok := volume.Annotations["hostPathProvisionerIdentity"] |
|||
if !ok { |
|||
return errors.New("identity annotation not found on PV") |
|||
} |
|||
if ann != p.identity { |
|||
return &controller.IgnoredError{"identity annotation on PV does not match ours"} |
|||
} |
|||
|
|||
path := path.Join(p.pvDir, volume.Name) |
|||
if err := os.RemoveAll(path); err != nil { |
|||
return err |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
func main() { |
|||
syscall.Umask(0) |
|||
|
|||
flag.Parse() |
|||
flag.Set("logtostderr", "true") |
|||
|
|||
// Create an InClusterConfig and use it to create a client for the controller
|
|||
// to use to communicate with Kubernetes
|
|||
config, err := rest.InClusterConfig() |
|||
if err != nil { |
|||
glog.Fatalf("Failed to create config: %v", err) |
|||
} |
|||
clientset, err := kubernetes.NewForConfig(config) |
|||
if err != nil { |
|||
glog.Fatalf("Failed to create client: %v", err) |
|||
} |
|||
|
|||
// The controller needs to know what the server version is because out-of-tree
|
|||
// provisioners aren't officially supported until 1.5
|
|||
serverVersion, err := clientset.Discovery().ServerVersion() |
|||
if err != nil { |
|||
glog.Fatalf("Error getting server version: %v", err) |
|||
} |
|||
|
|||
// Create the provisioner: it implements the Provisioner interface expected by
|
|||
// the controller
|
|||
hostPathProvisioner := NewHostPathProvisioner() |
|||
|
|||
// Start the provision controller which will dynamically provision hostPath
|
|||
// PVs
|
|||
pc := controller.NewProvisionController(clientset, resyncPeriod, provisionerName, hostPathProvisioner, serverVersion.GitVersion, exponentialBackOffOnError, failedRetryThreshold, leasePeriod, renewDeadline, retryPeriod, termLimit) |
|||
pc.Run(wait.NeverStop) |
|||
} |
|||
Loading…
Reference in new issue