2020-01-13 01:46:45 +01:00
|
|
|
package db_sql
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/astaxie/beego/orm"
|
|
|
|
"github.com/cloudsonic/sonic-server/log"
|
|
|
|
"github.com/cloudsonic/sonic-server/scanner"
|
|
|
|
)
|
|
|
|
|
|
|
|
type checkSumRepository struct {
|
|
|
|
data map[string]string
|
|
|
|
}
|
|
|
|
|
|
|
|
const checkSumId = "1"
|
|
|
|
|
2020-01-13 14:35:12 +01:00
|
|
|
type Checksum struct {
|
|
|
|
ID string `orm:"pk;column(id)"`
|
|
|
|
Sum string
|
2020-01-13 01:46:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func NewCheckSumRepository() scanner.CheckSumRepository {
|
|
|
|
r := &checkSumRepository{}
|
|
|
|
return r
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *checkSumRepository) loadData() error {
|
|
|
|
loadedData := make(map[string]string)
|
|
|
|
|
2020-01-13 14:35:12 +01:00
|
|
|
var all []Checksum
|
|
|
|
_, err := Db().QueryTable(&Checksum{}).All(&all)
|
2020-01-13 01:46:45 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-01-13 06:04:11 +01:00
|
|
|
|
|
|
|
for _, cks := range all {
|
2020-01-13 14:35:12 +01:00
|
|
|
loadedData[cks.ID] = cks.Sum
|
2020-01-13 06:04:11 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
r.data = loadedData
|
2020-01-13 01:46:45 +01:00
|
|
|
log.Debug("Loaded checksums", "total", len(loadedData))
|
2020-01-13 06:04:11 +01:00
|
|
|
return nil
|
2020-01-13 01:46:45 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func (r *checkSumRepository) Get(id string) (string, error) {
|
|
|
|
if r.data == nil {
|
|
|
|
err := r.loadData()
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return r.data[id], nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *checkSumRepository) SetData(newSums map[string]string) error {
|
2020-01-13 06:04:11 +01:00
|
|
|
err := WithTx(func(o orm.Ormer) error {
|
2020-01-13 16:24:44 +01:00
|
|
|
_, err := Db().Raw("delete from checksum").Exec()
|
2020-01-13 06:04:11 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2020-01-13 14:35:12 +01:00
|
|
|
var checksums []Checksum
|
2020-01-13 06:04:11 +01:00
|
|
|
for k, v := range newSums {
|
2020-01-13 14:35:12 +01:00
|
|
|
cks := Checksum{ID: k, Sum: v}
|
|
|
|
checksums = append(checksums, cks)
|
2020-01-13 06:04:11 +01:00
|
|
|
}
|
2020-01-13 14:35:12 +01:00
|
|
|
_, err = Db().InsertMulti(100, &checksums)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2020-01-13 06:04:11 +01:00
|
|
|
return nil
|
|
|
|
})
|
2020-01-13 01:46:45 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
r.data = newSums
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
var _ scanner.CheckSumRepository = (*checkSumRepository)(nil)
|