Optimize Checksums saving
This commit is contained in:
@@ -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)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
_, err = Db().InsertMulti(100, &checksums)
|
||||||
|
if err != nil {
|
||||||
|
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",
|
||||||
|
|||||||
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user