mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 14:20:23 -04:00
98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
// +build ignore
|
|
|
|
// Simple tool to create an archive stream from an old and new directory
|
|
//
|
|
// By default it will stream the comparison of two temporary directories with junk files
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/docker/docker/pkg/archive"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
flDebug = flag.Bool("D", false, "debugging output")
|
|
flNewDir = flag.String("newdir", "", "")
|
|
flOldDir = flag.String("olddir", "", "")
|
|
log = logrus.New()
|
|
)
|
|
|
|
func main() {
|
|
flag.Usage = func() {
|
|
fmt.Println("Produce a tar from comparing two directory paths. By default a demo tar is created of around 200 files (including hardlinks)")
|
|
fmt.Printf("%s [OPTIONS]\n", os.Args[0])
|
|
flag.PrintDefaults()
|
|
}
|
|
flag.Parse()
|
|
log.Out = os.Stderr
|
|
if (len(os.Getenv("DEBUG")) > 0) || *flDebug {
|
|
logrus.SetLevel(logrus.DebugLevel)
|
|
}
|
|
var newDir, oldDir string
|
|
|
|
if len(*flNewDir) == 0 {
|
|
var err error
|
|
newDir, err = ioutil.TempDir("", "docker-test-newDir")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(newDir)
|
|
if _, err := prepareUntarSourceDirectory(100, newDir, true); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
} else {
|
|
newDir = *flNewDir
|
|
}
|
|
|
|
if len(*flOldDir) == 0 {
|
|
oldDir, err := ioutil.TempDir("", "docker-test-oldDir")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer os.RemoveAll(oldDir)
|
|
} else {
|
|
oldDir = *flOldDir
|
|
}
|
|
|
|
changes, err := archive.ChangesDirs(newDir, oldDir)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
a, err := archive.ExportChanges(newDir, changes)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer a.Close()
|
|
|
|
i, err := io.Copy(os.Stdout, a)
|
|
if err != nil && err != io.EOF {
|
|
log.Fatal(err)
|
|
}
|
|
fmt.Fprintf(os.Stderr, "wrote archive of %d bytes", i)
|
|
}
|
|
|
|
func prepareUntarSourceDirectory(numberOfFiles int, targetPath string, makeLinks bool) (int, error) {
|
|
fileData := []byte("fooo")
|
|
for n := 0; n < numberOfFiles; n++ {
|
|
fileName := fmt.Sprintf("file-%d", n)
|
|
if err := ioutil.WriteFile(path.Join(targetPath, fileName), fileData, 0700); err != nil {
|
|
return 0, err
|
|
}
|
|
if makeLinks {
|
|
if err := os.Link(path.Join(targetPath, fileName), path.Join(targetPath, fileName+"-link")); err != nil {
|
|
return 0, err
|
|
}
|
|
}
|
|
}
|
|
totalSize := numberOfFiles * len(fileData)
|
|
return totalSize, nil
|
|
}
|