59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package persistence
|
|
|
|
import (
|
|
"github.com/astaxie/beego/orm"
|
|
"github.com/cloudsonic/sonic-server/model"
|
|
)
|
|
|
|
type checkSumRepository struct {
|
|
ormer orm.Ormer
|
|
}
|
|
|
|
const checkSumId = "1"
|
|
|
|
type checksum struct {
|
|
ID string `orm:"pk;column(id)"`
|
|
Sum string
|
|
}
|
|
|
|
func NewCheckSumRepository(o orm.Ormer) model.ChecksumRepository {
|
|
r := &checkSumRepository{ormer: o}
|
|
return r
|
|
}
|
|
|
|
func (r *checkSumRepository) GetData() (model.ChecksumMap, error) {
|
|
loadedData := make(map[string]string)
|
|
|
|
var all []checksum
|
|
_, err := r.ormer.QueryTable(&checksum{}).Limit(-1).All(&all)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, cks := range all {
|
|
loadedData[cks.ID] = cks.Sum
|
|
}
|
|
|
|
return loadedData, nil
|
|
}
|
|
|
|
func (r *checkSumRepository) SetData(newSums model.ChecksumMap) error {
|
|
_, err := r.ormer.Raw("delete from checksum").Exec()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var checksums []checksum
|
|
for k, v := range newSums {
|
|
cks := checksum{ID: k, Sum: v}
|
|
checksums = append(checksums, cks)
|
|
}
|
|
_, err = r.ormer.InsertMulti(batchSize, &checksums)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var _ model.ChecksumRepository = (*checkSumRepository)(nil)
|