Add CDC to backup and restore

tree.go

@@ -1,6 +1,7 @@
 package khepri
 
 import (
+	"bytes"
 	"encoding/json"
 	"fmt"
 	"io"
@@ -10,6 +11,8 @@ import (
 	"strconv"
 	"syscall"
 	"time"
+
+	"github.com/fd0/khepri/chunker"
 )
 
 type Tree struct {
@@ -32,7 +35,7 @@ type Node struct {
 	Links      uint64      `json:"links,omitempty"`
 	LinkTarget string      `json:"linktarget,omitempty"`
 	Device     uint64      `json:"device,omitempty"`
-	Content    ID          `json:"content,omitempty"`
+	Content    []ID        `json:"content,omitempty"`
 	Subtree    ID          `json:"subtree,omitempty"`
 	Tree       *Tree       `json:"-"`
 	repo       *Repository
@@ -44,6 +47,21 @@ func NewTree() *Tree {
 	}
 }
 
+func store_chunk(repo *Repository, rd io.Reader) (ID, error) {
+	wr, idch, err := repo.Create(TYPE_BLOB)
+	if err != nil {
+		return nil, err
+	}
+
+	io.Copy(wr, rd)
+	err = wr.Close()
+	if err != nil {
+		return nil, err
+	}
+
+	return <-idch, nil
+}
+
 func NewTreeFromPath(repo *Repository, dir string) (*Tree, error) {
 	fd, err := os.Open(dir)
 	defer fd.Close()
@@ -85,18 +103,38 @@ func NewTreeFromPath(repo *Repository, dir string) (*Tree, error) {
 				return nil, err
 			}
 
-			wr, idch, err := repo.Create(TYPE_BLOB)
+			if node.Size < chunker.MinSize {
+				// if the file is small enough, store it directly
+				id, err := store_chunk(repo, file)
+
 				if err != nil {
 					return nil, err
 				}
 
-			io.Copy(wr, file)
-			err = wr.Close()
+				node.Content = []ID{id}
+
+			} else {
+				// else store chunks
+				node.Content = []ID{}
+				ch := chunker.New(file)
+
+				for {
+					chunk, err := ch.Next()
+
+					if err == io.EOF {
+						break
+					}
+
 					if err != nil {
 						return nil, err
 					}
 
-			node.Content = <-idch
+					id, err := store_chunk(repo, bytes.NewBuffer(chunk.Data))
+
+					node.Content = append(node.Content, id)
+				}
+
+			}
 		}
 	}
 
@@ -297,13 +335,14 @@ func (node *Node) CreateAt(path string) error {
 	switch node.Type {
 	case "file":
 		// TODO: handle hard links
-		rd, err := node.repo.Get(TYPE_BLOB, node.Content)
+		f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0600)
+		defer f.Close()
 		if err != nil {
 			return err
 		}
 
-		f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY, 0600)
-		defer f.Close()
+		for _, blobid := range node.Content {
+			rd, err := node.repo.Get(TYPE_BLOB, blobid)
 			if err != nil {
 				return err
 			}
@@ -312,6 +351,7 @@ func (node *Node) CreateAt(path string) error {
 			if err != nil {
 				return err
 			}
+		}
 
 		f.Close()
 	case "symlink":