-
Notifications
You must be signed in to change notification settings - Fork 0
/
commits.go
121 lines (93 loc) · 2.54 KB
/
commits.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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package main
import (
"bytes"
"encoding/gob"
"path"
"github.com/boltdb/bolt"
"github.com/yosiat/npm-downloader/models"
)
type CommitsRepository struct {
Db *bolt.DB
}
func CreateCommitsRepository(baseDir string) (CommitsRepository, error) {
db, err := bolt.Open(path.Join(baseDir, "status"), 0600, nil)
if err != nil {
return CommitsRepository{}, err
}
tx, err := db.Begin(true)
if err != nil {
db.Close()
return CommitsRepository{}, err
}
defer tx.Rollback()
_, err = tx.CreateBucketIfNotExists([]byte("Success"))
if err != nil {
db.Close()
return CommitsRepository{}, err
}
_, err = tx.CreateBucketIfNotExists([]byte("Error"))
if err != nil {
db.Close()
return CommitsRepository{}, err
}
if err := tx.Commit(); err != nil {
db.Close()
return CommitsRepository{}, err
}
return CommitsRepository{Db: db}, nil
}
// Sucess - add success entry to the db
func (repo *CommitsRepository) Sucess(pkg models.Package) {
repo.Db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte("Success"))
packageCommit := models.CreatePackageCommit(pkg)
var pkgCommitBuffer bytes.Buffer
enc := gob.NewEncoder(&pkgCommitBuffer)
err := enc.Encode(packageCommit)
if err != nil {
return err
}
return b.Put([]byte(pkg.Id), pkgCommitBuffer.Bytes())
})
}
func (repo *CommitsRepository) Error(pkgID string, err error) {
repo.Db.Update(func(tx *bolt.Tx) error {
b := tx.Bucket([]byte("Error"))
return b.Put([]byte(pkgID), []byte(err.Error()))
})
}
// ErrorsCount - returns how much errors we have
func (repo *CommitsRepository) ErrorsCount() int {
errorsCount := 0
repo.Db.View(func(tx *bolt.Tx) error {
successBucket := tx.Bucket([]byte("Error"))
successBucket.ForEach(func(_, _ []byte) error {
errorsCount++
return nil
})
return nil
})
return errorsCount
}
// AllSucessfullPackages - returns all succesful packages download
func (repo *CommitsRepository) AllSucessfullPackages() map[string]models.PackageCommit {
packageCommitByID := make(map[string]models.PackageCommit)
repo.Db.View(func(tx *bolt.Tx) error {
successBucket := tx.Bucket([]byte("Success"))
successBucket.ForEach(func(packageId, packageCommitBuffer []byte) error {
var pkgCommitStatus models.PackageCommit
decoder := gob.NewDecoder(bytes.NewBuffer(packageCommitBuffer))
err := decoder.Decode(&pkgCommitStatus)
if err != nil {
return err
}
packageCommitByID[pkgCommitStatus.Id] = pkgCommitStatus
return nil
})
return nil
})
return packageCommitByID
}
func (repo *CommitsRepository) Close() {
repo.Db.Close()
}