discover existing snapshots on disk

Signed-off-by: Ashish Amarnath <[email protected]>

Author: ashish-amarnath

pkg/hostpath/hostpath.go

@@ -22,6 +22,7 @@ import (
 	"io"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"github.com/golang/glog"
 	"google.golang.org/grpc/codes"
@@ -127,12 +128,42 @@ func NewHostPathDriver(driverName, nodeID, endpoint string, ephemeral bool, maxV
 	}, nil
 }
 
+func discoverExistingSnapshots() {
+	glog.V(4).Infof("discovering existing snapshots in %s", dataRoot)
+	snapshots := []string{}
+	err := filepath.Walk(dataRoot, func(path string, info os.FileInfo, err error) error {
+		glog.V(6).Infof("path: %s", path)
+		if filepath.Ext(path) == ".tgz" {
+			snapshots = append(snapshots, path)
+		}
+		return nil
+	})
+
+	if err != nil {
+		glog.Errorf("failed to discover snapshots under %s", dataRoot)
+		return
+	}
+
+	for _, snap := range snapshots {
+		pathSplit := strings.Split(snap, ".")[0]
+		split := strings.Split(pathSplit, "/")
+		id := split[len(split)-1]
+		glog.V(4).Infof("snapshot: %s, id: %s", snap, id)
+		hostPathVolumeSnapshots[id] = hostPathSnapshot{
+			Id:         id,
+			Path:       snap,
+			ReadyToUse: true,
+		}
+	}
+}
+
 func (hp *hostPath) Run() {
 	// Create GRPC servers
 	hp.ids = NewIdentityServer(hp.name, hp.version)
 	hp.ns = NewNodeServer(hp.nodeID, hp.ephemeral, hp.maxVolumesPerNode)
 	hp.cs = NewControllerServer(hp.ephemeral, hp.nodeID)
 
+	discoverExistingSnapshots()
 	s := NewNonBlockingGRPCServer()
 	s.Start(hp.endpoint, hp.ids, hp.cs, hp.ns)
 	s.Wait()