/
build.go
59 lines (53 loc) · 1.16 KB
/
build.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
package eval
import (
"os"
"os/user"
"path/filepath"
"strings"
"sync"
)
func loadWeightsСached(loadFunc func() (*Weights, error)) func() (*Weights, error) {
var once sync.Once
var weights *Weights
var err error
return func() (*Weights, error) {
once.Do(func() {
weights, err = loadFunc()
})
return weights, err
}
}
var loadDefaultWeightsСached = loadWeightsСached(loadDefaultWeights)
// TODO return err
func NewDefaultEvaluationService() *EvaluationService {
var weights, err = loadDefaultWeightsСached()
if err != nil {
panic(err)
}
return NewEvaluationService(weights)
}
func loadFileWeights(path string) (*Weights, error) {
var f, err = os.Open(path)
if err != nil {
return nil, err
}
defer f.Close()
return LoadWeights(f)
}
func mapPath(path string) string {
if strings.HasPrefix(path, "~/") {
curUser, err := user.Current()
if err != nil {
return path
}
return filepath.Join(curUser.HomeDir, strings.TrimPrefix(path, "~/"))
}
if strings.HasPrefix(path, "./") {
var exePath, err = os.Executable()
if err != nil {
return path
}
return filepath.Join(filepath.Dir(exePath), strings.TrimPrefix(path, "./"))
}
return path
}