update snapshot processing

box.go

@@ -17,28 +17,165 @@ type Box struct {
 	zfs  *ZFSConfig
 }
 
-func (b *Box) ZFSGetLastSnapshot(path string) (s Snapshot, err error) {
-	return b.ssh.getLastSnapshot(path)
+func (b *Box) ZFSGetLastSnapshot(path string) (last Snapshot, err error) {
+	err = b.SnapshotInitialize()
+	if err != nil {
+		return
+	}
+
+	if *debugFlag {
+		log.Printf("Box.ZFSGetLastSnapshot : %s : Start %s (%d snapshots)", b.Name, path, len(b.zfs.SnapshotList))
+	}
+
+	for _, v := range b.zfs.SnapshotList {
+		if v.Path() == path {
+			last = v
+		}
+	}
+	if len(string(last)) == 0 {
+		err = fmt.Errorf("no snapshot")
+	}
+	return
 }
 
-func (b *Box) ZFSIsLastSnapshot(s Snapshot) bool {
-	return b.ssh.isLastSnapshot(s)
+func (b *Box) ZFSIsLastSnapshot(src Snapshot) (is bool, err error) {
+	err = b.SnapshotInitialize()
+	if err != nil {
+		return
+	}
+
+	if *debugFlag {
+		log.Printf("SSHConfig.isLastSnapshot : %s : Start %s", b.Name, string(src))
+	}
+
+	_, err = b.ZFSGetNextSnapshot(src)
+	if err != nil {
+		if err.Error() == "no snapshot" {
+			is = true
+			err = nil
+		}
+	} else {
+		is = false
+	}
+	return
 }
 
-func (b *Box) ZFSGetFirstSnapshot(path string) (s Snapshot, err error) {
-	return b.ssh.getFirstSnapshot(path)
+func (b *Box) ZFSGetFirstSnapshot(path string) (first Snapshot, err error) {
+	err = b.SnapshotInitialize()
+	if err != nil {
+		return
+	}
+
+	if *debugFlag {
+		log.Printf("SSHConfig.getFirstSnapshot : Start %s:%s", b.Name, path)
+	}
+
+	for _, v := range b.zfs.SnapshotList {
+		if v.Path() == path {
+			first = v
+			return
+		}
+	}
+	err = fmt.Errorf("no snapshot")
+	return
 }
 
 func (b *Box) ZFSGetNextSnapshot(src Snapshot) (next Snapshot, err error) {
-	return b.ssh.getNextSnapshot(src)
+	err = b.SnapshotInitialize()
+	if err != nil {
+		return
+	}
+
+	if *debugFlag {
+		log.Printf("Box.ZFSGetNextSnapshot : Start %s:%s", b.Name, string(src))
+	}
+
+	for id, v := range b.zfs.SnapshotList {
+		if v == src {
+			if len(b.zfs.SnapshotList) > id+1 {
+				next = b.zfs.SnapshotList[id+1]
+				if next.Path() == src.Path() {
+					return
+				} else {
+					err = fmt.Errorf("no snapshot")
+					return
+				}
+			} else {
+				err = fmt.Errorf("no snapshot")
+				return
+			}
+		}
+	}
+	err = fmt.Errorf("no snapshot")
+	return
 }
 
 func (b *Box) ZFSUpdateSnapshotList() (err error) {
-	return b.ssh.getSnapshotList()
+	b.zfs.M.Lock()
+	if b.zfs.SnapshotDeleted || b.zfs.SnapshotAdded {
+		b.zfs.SnapshotInitialized = false
+	}
+	b.zfs.M.Unlock()
+
+	err = b.SnapshotInitialize()
+	return
 }
 
-func (b *Box) ZFSGetSnapshotList() []Snapshot {
-	return b.ssh.snapshot
+func (b *Box) ZFSGetSnapshotList() (snaps []Snapshot, err error) {
+	err = b.SnapshotInitialize()
+	if err != nil {
+		return
+	}
+
+	b.zfs.M.Lock()
+	defer b.zfs.M.Unlock()
+
+	snaps = b.zfs.SnapshotList
+	return
+}
+
+func (b *Box) SnapshotInitialize() (err error) {
+	b.zfs.M.Lock()
+	defer b.zfs.M.Unlock()
+
+	if b.zfs.SnapshotInitialized {
+		return nil
+	}
+
+	if *debugFlag {
+		log.Printf("Box.SnapshotInitialize : %s : Start", b.Name)
+	}
+
+	b.zfs.SnapshotList = make([]Snapshot, 0)
+
+	var buf *bytes.Buffer
+	buf, err = b.SSHExec("zfs list -H -t snapshot -o name")
+
+	csvReader := csv.NewReader(buf)
+	csvReader.Comma = '\t'
+	csvReader.FieldsPerRecord = 1
+
+	csvData, err := csvReader.ReadAll()
+	if err != nil {
+		if *debugFlag {
+			log.Printf("Box.SnapshotInitialize : %s : csvReader.ReadAll() : %s", b.Name, err)
+		}
+		return err
+	}
+
+	for _, rec := range csvData {
+		b.zfs.SnapshotList = append(b.zfs.SnapshotList, Snapshot(rec[0]))
+	}
+
+	if *debugFlag {
+		log.Printf("Box.SnapshotInitialize : %s : read %d zfs snapshots", b.Name, len(b.zfs.SnapshotList))
+	}
+
+	b.zfs.SnapshotInitialized = true
+	b.zfs.SnapshotAdded = false
+	b.zfs.SnapshotDeleted = false
+
+	return nil
 }
 
 func (b *Box) ZFSUpdateList() (err error) {