go fmt
This commit is contained in:
parent
54a0359897
commit
61d5532e33
12
asink.go
12
asink.go
@ -1,11 +1,11 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"flag"
|
|
||||||
"path"
|
|
||||||
"os/user"
|
|
||||||
"code.google.com/p/goconf/conf"
|
"code.google.com/p/goconf/conf"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"os/user"
|
||||||
|
"path"
|
||||||
)
|
)
|
||||||
|
|
||||||
var configFileName string
|
var configFileName string
|
||||||
@ -61,6 +61,8 @@ func ProcessEvent(storage Storage, event *Event) {
|
|||||||
|
|
||||||
if event.IsUpdate() {
|
if event.IsUpdate() {
|
||||||
err := storage.Put(event.Path, event.Hash)
|
err := storage.Put(event.Path, event.Hash)
|
||||||
if err != nil { panic(err) }
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
12
hash.go
12
hash.go
@ -1,21 +1,25 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/sha256"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
"os"
|
||||||
"fmt"
|
|
||||||
"crypto/sha256"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func HashFile(filename string) (string, error) {
|
func HashFile(filename string) (string, error) {
|
||||||
hashfn := sha256.New()
|
hashfn := sha256.New()
|
||||||
|
|
||||||
infile, err := os.Open(filename)
|
infile, err := os.Open(filename)
|
||||||
if err != nil { return "", err }
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
defer infile.Close()
|
defer infile.Close()
|
||||||
|
|
||||||
_, err = io.Copy(hashfn, infile)
|
_, err = io.Copy(hashfn, infile)
|
||||||
if err != nil { return "", err }
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
return fmt.Sprintf("%x", hashfn.Sum(nil)), nil
|
return fmt.Sprintf("%x", hashfn.Sum(nil)), nil
|
||||||
}
|
}
|
||||||
|
@ -1,8 +1,8 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"code.google.com/p/goconf/conf"
|
"code.google.com/p/goconf/conf"
|
||||||
|
"errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Storage interface {
|
type Storage interface {
|
||||||
|
@ -1,12 +1,12 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
"code.google.com/p/goconf/conf"
|
||||||
|
"errors"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"errors"
|
|
||||||
"code.google.com/p/goconf/conf"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type LocalStorage struct {
|
type LocalStorage struct {
|
||||||
@ -18,9 +18,13 @@ func ensureDirExists(dir string) error {
|
|||||||
_, err := os.Lstat(dir)
|
_, err := os.Lstat(dir)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fi, err := os.Lstat(path.Dir(dir))
|
fi, err := os.Lstat(path.Dir(dir))
|
||||||
if err != nil { return err }
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
err = os.Mkdir(dir, fi.Mode().Perm())
|
err = os.Mkdir(dir, fi.Mode().Perm())
|
||||||
if err != nil { return err }
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@ -37,36 +41,50 @@ func NewLocalStorage(config *conf.ConfigFile) (*LocalStorage, error) {
|
|||||||
|
|
||||||
//make sure the base directory and tmp subdir exist
|
//make sure the base directory and tmp subdir exist
|
||||||
err = ensureDirExists(ls.storageDir)
|
err = ensureDirExists(ls.storageDir)
|
||||||
if err != nil { return nil, err}
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
err = ensureDirExists(ls.tmpSubdir)
|
err = ensureDirExists(ls.tmpSubdir)
|
||||||
if err != nil { return nil, err}
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
return ls, nil
|
return ls, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ls *LocalStorage) copyToTmp(src string) (string, error) {
|
func (ls *LocalStorage) copyToTmp(src string) (string, error) {
|
||||||
infile, err := os.Open(src)
|
infile, err := os.Open(src)
|
||||||
if err != nil { return "", err }
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
defer infile.Close()
|
defer infile.Close()
|
||||||
|
|
||||||
outfile, err := ioutil.TempFile(ls.tmpSubdir, "asink")
|
outfile, err := ioutil.TempFile(ls.tmpSubdir, "asink")
|
||||||
if err != nil { return "", err }
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
defer outfile.Close()
|
defer outfile.Close()
|
||||||
|
|
||||||
_, err = io.Copy(outfile, infile)
|
_, err = io.Copy(outfile, infile)
|
||||||
if err != nil { return "", err }
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
return outfile.Name(), nil
|
return outfile.Name(), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (ls *LocalStorage) Put(filename string, hash string) (e error) {
|
func (ls *LocalStorage) Put(filename string, hash string) (e error) {
|
||||||
tmpfile, err := ls.copyToTmp(filename)
|
tmpfile, err := ls.copyToTmp(filename)
|
||||||
if err != nil { return err }
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
err = os.Rename(tmpfile, path.Join(ls.storageDir, hash))
|
err = os.Rename(tmpfile, path.Join(ls.storageDir, hash))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err := os.Remove(tmpfile)
|
err := os.Remove(tmpfile)
|
||||||
if err != nil { return err }
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
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 {
|
func (ls *LocalStorage) Get(filename string, hash string) error {
|
||||||
infile, err := os.Open(path.Join(ls.storageDir, hash))
|
infile, err := os.Open(path.Join(ls.storageDir, hash))
|
||||||
if err != nil { return err }
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
defer infile.Close()
|
defer infile.Close()
|
||||||
|
|
||||||
outfile, err := os.Open(filename)
|
outfile, err := os.Open(filename)
|
||||||
if err != nil { return err }
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
defer outfile.Close()
|
defer outfile.Close()
|
||||||
|
|
||||||
_, err = io.Copy(outfile, infile)
|
_, err = io.Copy(outfile, infile)
|
||||||
|
@ -30,7 +30,9 @@ func StartWatching(watchDir string, fileUpdates chan *Event) {
|
|||||||
event.Path = ev.Name
|
event.Path = ev.Name
|
||||||
if event.IsUpdate() {
|
if event.IsUpdate() {
|
||||||
event.Hash, err = HashFile(ev.Name)
|
event.Hash, err = HashFile(ev.Name)
|
||||||
if err != nil { continue }
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
event.Hash = ""
|
event.Hash = ""
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user