mirror of https://github.com/k3s-io/k3s
110 lines
3.7 KiB
Go
110 lines
3.7 KiB
Go
/*
|
|
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 vclib
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/property"
|
|
"github.com/vmware/govmomi/vim25/mo"
|
|
"github.com/vmware/govmomi/vim25/soap"
|
|
"github.com/vmware/govmomi/vim25/types"
|
|
"k8s.io/klog/v2"
|
|
)
|
|
|
|
// Datastore extends the govmomi Datastore object
|
|
type Datastore struct {
|
|
*object.Datastore
|
|
Datacenter *Datacenter
|
|
}
|
|
|
|
// DatastoreInfo is a structure to store the Datastore and it's Info.
|
|
type DatastoreInfo struct {
|
|
*Datastore
|
|
Info *types.DatastoreInfo
|
|
}
|
|
|
|
func (di DatastoreInfo) String() string {
|
|
return fmt.Sprintf("Datastore: %+v, datastore URL: %s", di.Datastore, di.Info.Url)
|
|
}
|
|
|
|
// CreateDirectory creates the directory at location specified by directoryPath.
|
|
// If the intermediate level folders do not exist, and the parameter createParents is true, all the non-existent folders are created.
|
|
// directoryPath must be in the format "[vsanDatastore] kubevols"
|
|
func (ds *Datastore) CreateDirectory(ctx context.Context, directoryPath string, createParents bool) error {
|
|
fileManager := object.NewFileManager(ds.Client())
|
|
err := fileManager.MakeDirectory(ctx, directoryPath, ds.Datacenter.Datacenter, createParents)
|
|
if err != nil {
|
|
if soap.IsSoapFault(err) {
|
|
soapFault := soap.ToSoapFault(err)
|
|
if _, ok := soapFault.VimFault().(types.FileAlreadyExists); ok {
|
|
return ErrFileAlreadyExist
|
|
}
|
|
}
|
|
return err
|
|
}
|
|
klog.V(LogLevel).Infof("Created dir with path as %+q", directoryPath)
|
|
return nil
|
|
}
|
|
|
|
// GetType returns the type of datastore
|
|
func (ds *Datastore) GetType(ctx context.Context) (string, error) {
|
|
var dsMo mo.Datastore
|
|
pc := property.DefaultCollector(ds.Client())
|
|
err := pc.RetrieveOne(ctx, ds.Datastore.Reference(), []string{"summary"}, &dsMo)
|
|
if err != nil {
|
|
klog.Errorf("Failed to retrieve datastore summary property. err: %v", err)
|
|
return "", err
|
|
}
|
|
return dsMo.Summary.Type, nil
|
|
}
|
|
|
|
// IsCompatibleWithStoragePolicy returns true if datastore is compatible with given storage policy else return false
|
|
// for not compatible datastore, fault message is also returned
|
|
func (ds *Datastore) IsCompatibleWithStoragePolicy(ctx context.Context, storagePolicyID string) (bool, string, error) {
|
|
pbmClient, err := NewPbmClient(ctx, ds.Client())
|
|
if err != nil {
|
|
klog.Errorf("Failed to get new PbmClient Object. err: %v", err)
|
|
return false, "", err
|
|
}
|
|
return pbmClient.IsDatastoreCompatible(ctx, storagePolicyID, ds)
|
|
}
|
|
|
|
// GetDatastoreHostMounts gets the host names mounted on given datastore
|
|
func (ds *Datastore) GetDatastoreHostMounts(ctx context.Context) ([]types.ManagedObjectReference, error) {
|
|
var dsMo mo.Datastore
|
|
pc := property.DefaultCollector(ds.Client())
|
|
err := pc.RetrieveOne(ctx, ds.Datastore.Reference(), []string{"host"}, &dsMo)
|
|
if err != nil {
|
|
klog.Errorf("Failed to retrieve datastore host mount property. err: %v", err)
|
|
return nil, err
|
|
}
|
|
hosts := make([]types.ManagedObjectReference, len(dsMo.Host))
|
|
for i, dsHostMount := range dsMo.Host {
|
|
hosts[i] = dsHostMount.Key
|
|
}
|
|
return hosts, nil
|
|
}
|
|
|
|
// Exists returns whether the given file exists in this datastore
|
|
func (ds *Datastore) Exists(ctx context.Context, file string) bool {
|
|
_, err := ds.Datastore.Stat(ctx, file)
|
|
return err == nil
|
|
}
|