This commit is contained in:
Aaron Lindsay 2013-02-11 23:17:12 -05:00
parent 54a0359897
commit 61d5532e33
6 changed files with 69 additions and 39 deletions

View File

@ -1,11 +1,11 @@
package main
import (
"fmt"
"flag"
"path"
"os/user"
"code.google.com/p/goconf/conf"
"flag"
"fmt"
"os/user"
"path"
)
var configFileName string
@ -61,6 +61,8 @@ func ProcessEvent(storage Storage, event *Event) {
if event.IsUpdate() {
err := storage.Put(event.Path, event.Hash)
if err != nil { panic(err) }
if err != nil {
panic(err)
}
}
}

12
hash.go
View File

@ -1,21 +1,25 @@
package main
import (
"crypto/sha256"
"fmt"
"io"
"os"
"fmt"
"crypto/sha256"
)
func HashFile(filename string) (string, error) {
hashfn := sha256.New()
infile, err := os.Open(filename)
if err != nil { return "", err }
if err != nil {
return "", err
}
defer infile.Close()
_, err = io.Copy(hashfn, infile)
if err != nil { return "", err }
if err != nil {
return "", err
}
return fmt.Sprintf("%x", hashfn.Sum(nil)), nil
}

View File

@ -1,8 +1,8 @@
package main
import (
"errors"
"code.google.com/p/goconf/conf"
"errors"
)
type Storage interface {

View File

@ -1,12 +1,12 @@
package main
import (
"os"
"code.google.com/p/goconf/conf"
"errors"
"io"
"io/ioutil"
"os"
"path"
"errors"
"code.google.com/p/goconf/conf"
)
type LocalStorage struct {
@ -18,9 +18,13 @@ func ensureDirExists(dir string) error {
_, err := os.Lstat(dir)
if err != nil {
fi, err := os.Lstat(path.Dir(dir))
if err != nil { return err }
if err != nil {
return err
}
err = os.Mkdir(dir, fi.Mode().Perm())
if err != nil { return err }
if err != nil {
return err
}
}
return nil
}
@ -37,36 +41,50 @@ func NewLocalStorage(config *conf.ConfigFile) (*LocalStorage, error) {
//make sure the base directory and tmp subdir exist
err = ensureDirExists(ls.storageDir)
if err != nil { return nil, err}
if err != nil {
return nil, err
}
err = ensureDirExists(ls.tmpSubdir)
if err != nil { return nil, err}
if err != nil {
return nil, err
}
return ls, nil
}
func (ls *LocalStorage) copyToTmp(src string) (string, error) {
infile, err := os.Open(src)
if err != nil { return "", err }
if err != nil {
return "", err
}
defer infile.Close()
outfile, err := ioutil.TempFile(ls.tmpSubdir, "asink")
if err != nil { return "", err }
if err != nil {
return "", err
}
defer outfile.Close()
_, err = io.Copy(outfile, infile)
if err != nil { return "", err }
if err != nil {
return "", err
}
return outfile.Name(), nil
}
func (ls *LocalStorage) Put(filename string, hash string) (e error) {
tmpfile, err := ls.copyToTmp(filename)
if err != nil { return err }
if err != nil {
return err
}
err = os.Rename(tmpfile, path.Join(ls.storageDir, hash))
if err != nil {
err := os.Remove(tmpfile)
if err != nil { return err }
if err != nil {
return err
}
}
return nil
@ -74,11 +92,15 @@ func (ls *LocalStorage) Put(filename string, hash string) (e error) {
func (ls *LocalStorage) Get(filename string, hash string) error {
infile, err := os.Open(path.Join(ls.storageDir, hash))
if err != nil { return err }
if err != nil {
return err
}
defer infile.Close()
outfile, err := os.Open(filename)
if err != nil { return err }
if err != nil {
return err
}
defer outfile.Close()
_, err = io.Copy(outfile, infile)

View File

@ -30,7 +30,9 @@ func StartWatching(watchDir string, fileUpdates chan *Event) {
event.Path = ev.Name
if event.IsUpdate() {
event.Hash, err = HashFile(ev.Name)
if err != nil { continue }
if err != nil {
continue
}
} else {
event.Hash = ""
}