Added -newer argument, which takes a date/time in RFC3339 format and

exports only files newer than this time
This commit is contained in:
Denis Yeldandi 2015-05-20 15:11:12 +03:00
parent 99aee22e08
commit 4e1c30b60a

View file

@ -21,10 +21,11 @@ func init() {
const ( const (
defaultFnFormat = `{{.Mime}}/{{.Id}}:{{.Name}}` defaultFnFormat = `{{.Mime}}/{{.Id}}:{{.Name}}`
timeFormat = "2006-01-02T15:04:05"
) )
var cmdExport = &Command{ var cmdExport = &Command{
UsageLine: "export -dir=/tmp -volumeId=234 -o=/dir/name.tar -fileNameFormat={{.Name}}", UsageLine: "export -dir=/tmp -volumeId=234 -o=/dir/name.tar -fileNameFormat={{.Name}} -newer='"+timeFormat+"'",
Short: "list or export files from one volume data file", Short: "list or export files from one volume data file",
Long: `List all files in a volume, or Export all files in a volume to a tar file if the output is specified. Long: `List all files in a volume, or Export all files in a volume to a tar file if the output is specified.
@ -39,19 +40,32 @@ var (
exportVolumeId = cmdExport.Flag.Int("volumeId", -1, "a volume id. The volume .dat and .idx files should already exist in the dir.") exportVolumeId = cmdExport.Flag.Int("volumeId", -1, "a volume id. The volume .dat and .idx files should already exist in the dir.")
dest = cmdExport.Flag.String("o", "", "output tar file name, must ends with .tar, or just a \"-\" for stdout") dest = cmdExport.Flag.String("o", "", "output tar file name, must ends with .tar, or just a \"-\" for stdout")
format = cmdExport.Flag.String("fileNameFormat", defaultFnFormat, "filename format, default to {{.Mime}}/{{.Id}}:{{.Name}}") format = cmdExport.Flag.String("fileNameFormat", defaultFnFormat, "filename format, default to {{.Mime}}/{{.Id}}:{{.Name}}")
newer = cmdExport.Flag.String("newer", "", "export only files newer than this time, default is all files. Must be specified in RFC3339 without timezone")
tarFh *tar.Writer tarFh *tar.Writer
tarHeader tar.Header tarHeader tar.Header
fnTmpl *template.Template fnTmpl *template.Template
fnTmplBuf = bytes.NewBuffer(nil) fnTmplBuf = bytes.NewBuffer(nil)
newerThan time.Time
newerThanUnix int64 = -1
localLocation, _ = time.LoadLocation("Local")
) )
func runExport(cmd *Command, args []string) bool { func runExport(cmd *Command, args []string) bool {
var err error
if *newer != "" {
if newerThan, err = time.ParseInLocation(timeFormat, *newer, localLocation); err != nil {
fmt.Println("cannot parse 'newer' argument: " + err.Error())
return false
}
newerThanUnix = newerThan.Unix()
}
if *exportVolumeId == -1 { if *exportVolumeId == -1 {
return false return false
} }
var err error
if *dest != "" { if *dest != "" {
if *dest != "-" && !strings.HasSuffix(*dest, ".tar") { if *dest != "-" && !strings.HasSuffix(*dest, ".tar") {
fmt.Println("the output file", *dest, "should be '-' or end with .tar") fmt.Println("the output file", *dest, "should be '-' or end with .tar")
@ -109,6 +123,11 @@ func runExport(cmd *Command, args []string) bool {
glog.V(3).Infof("key %d offset %d size %d disk_size %d gzip %v ok %v nv %+v", glog.V(3).Infof("key %d offset %d size %d disk_size %d gzip %v ok %v nv %+v",
n.Id, offset, n.Size, n.DiskSize(), n.IsGzipped(), ok, nv) n.Id, offset, n.Size, n.DiskSize(), n.IsGzipped(), ok, nv)
if ok && nv.Size > 0 && int64(nv.Offset)*8 == offset { if ok && nv.Size > 0 && int64(nv.Offset)*8 == offset {
if newerThanUnix >= 0 && n.HasLastModifiedDate() && n.LastModified < uint64(newerThanUnix) {
glog.V(3).Infof("Skipping this file, as it's old enough: LastModified %d vs %d",
n.LastModified, newerThanUnix)
return nil
}
return walker(vid, n, version) return walker(vid, n, version)
} }
if !ok { if !ok {
@ -151,6 +170,12 @@ func walker(vid storage.VolumeId, n *storage.Needle, version storage.Version) (e
} }
tarHeader.Name, tarHeader.Size = nm, int64(len(n.Data)) tarHeader.Name, tarHeader.Size = nm, int64(len(n.Data))
if n.HasLastModifiedDate() {
tarHeader.ModTime = time.Unix(int64(n.LastModified), 0)
} else {
tarHeader.ModTime = time.Unix(0,0)
}
tarHeader.ChangeTime = tarHeader.ModTime
if err = tarFh.WriteHeader(&tarHeader); err != nil { if err = tarFh.WriteHeader(&tarHeader); err != nil {
return err return err
} }