-
Notifications
You must be signed in to change notification settings - Fork 23
/
reload.go
66 lines (60 loc) · 1.49 KB
/
reload.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package main
import (
"errors"
"fmt"
"github.com/MG-RAST/Shock/shock-server/conf"
"github.com/MG-RAST/Shock/shock-server/db"
"github.com/MG-RAST/Shock/shock-server/node"
"os"
"path/filepath"
"regexp"
"strings"
)
var (
dirExp, _ = regexp.Compile("(?i)([a-f0-9]{2}/){3}[a-f0-9]{8}-[a-f0-9]{4}-4[a-f0-9]{3}-[89ab][a-f0-9]{3}-[a-f0-9]{12}$")
)
func reload(source string) (err error) {
fmt.Println("source:", conf.RELOAD)
if strings.HasPrefix(conf.RELOAD, "http://") {
fmt.Println("source-type: url")
if err = reloadDB(); err != nil {
return err
}
return reloadFromUrl(source)
} else {
fmt.Println("source-type: dir")
if source == conf.Conf["data-path"] {
fmt.Println("source dir is the same as conf data dir: db reload only.")
} else {
return errors.New("source dir is not the same as conf data dir: copy not implemented yet")
}
if err = reloadDB(); err != nil {
return err
}
return filepath.Walk(source, reloadFromDir)
}
return
}
func reloadDB() (err error) {
fmt.Printf("dropping & re-initializing database...")
if err = db.Drop(); err != nil {
return err
}
db.Initialize()
fmt.Printf("done\n")
return
}
func reloadFromUrl(url string) error {
return errors.New("reload from url not implemented yet")
}
func reloadFromDir(path string, info os.FileInfo, err error) error {
if dirExp.MatchString(path) {
id := filepath.Base(path)
fmt.Printf(id + "...")
if err := node.ReloadFromDisk(path); err != nil {
return err
}
fmt.Printf("done\n")
}
return nil
}