Optimize Checksums saving

This commit is contained in:
Deluan
2020-01-13 08:35:12 -05:00
committed by Deluan Quintão
parent b07642a8d1
commit 87ca885b5e
3 changed files with 17 additions and 15 deletions
+12 -10
View File
@@ -12,9 +12,9 @@ type checkSumRepository struct {
const checkSumId = "1" const checkSumId = "1"
type CheckSums struct { type Checksum struct {
ID string `orm:"pk;column(id)"` ID string `orm:"pk;column(id)"`
Value string Sum string
} }
func NewCheckSumRepository() scanner.CheckSumRepository { func NewCheckSumRepository() scanner.CheckSumRepository {
@@ -25,14 +25,14 @@ func NewCheckSumRepository() scanner.CheckSumRepository {
func (r *checkSumRepository) loadData() error { func (r *checkSumRepository) loadData() error {
loadedData := make(map[string]string) loadedData := make(map[string]string)
var all []CheckSums var all []Checksum
_, err := Db().QueryTable(&CheckSums{}).All(&all) _, err := Db().QueryTable(&Checksum{}).All(&all)
if err != nil { if err != nil {
return err return err
} }
for _, cks := range all { for _, cks := range all {
loadedData[cks.ID] = cks.Value loadedData[cks.ID] = cks.Sum
} }
r.data = loadedData r.data = loadedData
@@ -52,19 +52,21 @@ func (r *checkSumRepository) Get(id string) (string, error) {
func (r *checkSumRepository) SetData(newSums map[string]string) error { func (r *checkSumRepository) SetData(newSums map[string]string) error {
err := WithTx(func(o orm.Ormer) error { err := WithTx(func(o orm.Ormer) error {
_, err := Db().Raw("delete from check_sums").Exec() _, err := Db().Raw("delete from checksum").Exec()
if err != nil { if err != nil {
return err return err
} }
var checksums []Checksum
for k, v := range newSums { for k, v := range newSums {
cks := CheckSums{ID: k, Value: v} cks := Checksum{ID: k, Sum: v}
// TODO Use InsertMulti checksums = append(checksums, cks)
_, err := Db().Insert(&cks) }
_, err = Db().InsertMulti(100, &checksums)
if err != nil { if err != nil {
return err return err
} }
}
return nil return nil
}) })
if err != nil { if err != nil {
@@ -10,7 +10,7 @@ var _ = Describe("ChecksumRepository", func() {
var repo scanner.CheckSumRepository var repo scanner.CheckSumRepository
BeforeEach(func() { BeforeEach(func() {
Db().Delete(&CheckSums{ID: checkSumId}) Db().Delete(&Checksum{ID: checkSumId})
repo = NewCheckSumRepository() repo = NewCheckSumRepository()
err := repo.SetData(map[string]string{ err := repo.SetData(map[string]string{
"a": "AAA", "b": "BBB", "a": "AAA", "b": "BBB",
+1 -1
View File
@@ -58,7 +58,7 @@ func initORM(dbPath string) error {
orm.RegisterModel(new(Album)) orm.RegisterModel(new(Album))
orm.RegisterModel(new(MediaFile)) orm.RegisterModel(new(MediaFile))
orm.RegisterModel(new(ArtistInfo)) orm.RegisterModel(new(ArtistInfo))
orm.RegisterModel(new(CheckSums)) orm.RegisterModel(new(Checksum))
orm.RegisterModel(new(Property)) orm.RegisterModel(new(Property))
orm.RegisterModel(new(Playlist)) orm.RegisterModel(new(Playlist))
err := orm.RegisterDataBase("default", "sqlite3", dbPath) err := orm.RegisterDataBase("default", "sqlite3", dbPath)