Fix some performance/volume issues

This commit is contained in:
Deluan
2020-01-13 00:04:11 -05:00
committed by Deluan Quintão
parent 973f8869eb
commit fddd0dc353
9 changed files with 94 additions and 45 deletions
+1 -1
View File
@@ -35,7 +35,7 @@ type albumRepository struct {
func NewAlbumRepository() domain.AlbumRepository { func NewAlbumRepository() domain.AlbumRepository {
r := &albumRepository{} r := &albumRepository{}
r.entityName = "album" r.tableName = "album"
return r return r
} }
+1 -1
View File
@@ -18,7 +18,7 @@ type artistRepository struct {
func NewArtistRepository() domain.ArtistRepository { func NewArtistRepository() domain.ArtistRepository {
r := &artistRepository{} r := &artistRepository{}
r.entityName = "artist" r.tableName = "artist"
return r return r
} }
+26 -21
View File
@@ -1,8 +1,6 @@
package db_sql package db_sql
import ( import (
"encoding/json"
"github.com/astaxie/beego/orm" "github.com/astaxie/beego/orm"
"github.com/cloudsonic/sonic-server/log" "github.com/cloudsonic/sonic-server/log"
"github.com/cloudsonic/sonic-server/scanner" "github.com/cloudsonic/sonic-server/scanner"
@@ -15,8 +13,8 @@ type checkSumRepository struct {
const checkSumId = "1" const checkSumId = "1"
type CheckSums struct { type CheckSums struct {
ID string `orm:"pk;column(id)"` ID string `orm:"pk;column(id)"`
Data string `orm:"type(text)"` Value string
} }
func NewCheckSumRepository() scanner.CheckSumRepository { func NewCheckSumRepository() scanner.CheckSumRepository {
@@ -26,20 +24,20 @@ func NewCheckSumRepository() scanner.CheckSumRepository {
func (r *checkSumRepository) loadData() error { func (r *checkSumRepository) loadData() error {
loadedData := make(map[string]string) loadedData := make(map[string]string)
r.data = loadedData
cks := CheckSums{ID: checkSumId} var all []CheckSums
err := Db().Read(&cks) _, err := Db().QueryTable(&CheckSums{}).All(&all)
if err == orm.ErrNoRows {
_, err = Db().Insert(&cks)
return err
}
if err != nil { if err != nil {
return err return err
} }
_ = json.Unmarshal([]byte(cks.Data), &loadedData)
for _, cks := range all {
loadedData[cks.ID] = cks.Value
}
r.data = loadedData
log.Debug("Loaded checksums", "total", len(loadedData)) log.Debug("Loaded checksums", "total", len(loadedData))
return err return nil
} }
func (r *checkSumRepository) Get(id string) (string, error) { func (r *checkSumRepository) Get(id string) (string, error) {
@@ -53,14 +51,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 {
data, _ := json.Marshal(&newSums) err := WithTx(func(o orm.Ormer) error {
cks := CheckSums{ID: checkSumId, Data: string(data)} _, err := Db().Raw("delete from check_sums").Exec()
var err error if err != nil {
if Db().QueryTable(&CheckSums{}).Filter("id", checkSumId).Exist() { return err
_, err = Db().Update(&cks) }
} else {
_, err = Db().Insert(&cks) for k, v := range newSums {
} cks := CheckSums{ID: k, Value: v}
_, err := Db().Insert(&cks)
if err != nil {
return err
}
}
return nil
})
if err != nil { if err != nil {
return err return err
} }
+2 -2
View File
@@ -21,7 +21,7 @@ type artistIndexRepository struct {
func NewArtistIndexRepository() domain.ArtistIndexRepository { func NewArtistIndexRepository() domain.ArtistIndexRepository {
r := &artistIndexRepository{} r := &artistIndexRepository{}
r.entityName = "artist_info" r.tableName = "artist_info"
return r return r
} }
@@ -106,7 +106,7 @@ func (r *artistIndexRepository) GetAll() (domain.ArtistIndexes, error) {
} }
func (r *artistIndexRepository) DeleteAll() error { func (r *artistIndexRepository) DeleteAll() error {
_, err := r.newQuery(Db()).Exclude("idx", ".PHONY").Delete() _, err := Db().Raw("delete from artist_info").Exec()
return err return err
} }
+1 -1
View File
@@ -41,7 +41,7 @@ type mediaFileRepository struct {
func NewMediaFileRepository() domain.MediaFileRepository { func NewMediaFileRepository() domain.MediaFileRepository {
r := &mediaFileRepository{} r := &mediaFileRepository{}
r.entityName = "media_file" r.tableName = "media_file"
return r return r
} }
+1 -1
View File
@@ -24,7 +24,7 @@ type playlistRepository struct {
func NewPlaylistRepository() domain.PlaylistRepository { func NewPlaylistRepository() domain.PlaylistRepository {
r := &playlistRepository{} r := &playlistRepository{}
r.entityName = "playlist" r.tableName = "playlist"
return r return r
} }
+1 -1
View File
@@ -16,7 +16,7 @@ type propertyRepository struct {
func NewPropertyRepository() domain.PropertyRepository { func NewPropertyRepository() domain.PropertyRepository {
r := &propertyRepository{} r := &propertyRepository{}
r.entityName = "property" r.tableName = "property"
return r return r
} }
@@ -10,7 +10,7 @@ var _ = Describe("PropertyRepository", func() {
var repo domain.PropertyRepository var repo domain.PropertyRepository
BeforeEach(func() { BeforeEach(func() {
Db().QueryTable("property").Exclude("id", ".PHONY").Delete() Db().Raw("delete from property").Exec()
repo = NewPropertyRepository() repo = NewPropertyRepository()
}) })
+60 -16
View File
@@ -3,15 +3,16 @@ package db_sql
import ( import (
"github.com/astaxie/beego/orm" "github.com/astaxie/beego/orm"
"github.com/cloudsonic/sonic-server/domain" "github.com/cloudsonic/sonic-server/domain"
"github.com/cloudsonic/sonic-server/log"
"github.com/cloudsonic/sonic-server/persistence" "github.com/cloudsonic/sonic-server/persistence"
) )
type sqlRepository struct { type sqlRepository struct {
entityName string tableName string
} }
func (r *sqlRepository) newQuery(o orm.Ormer, options ...domain.QueryOptions) orm.QuerySeter { func (r *sqlRepository) newQuery(o orm.Ormer, options ...domain.QueryOptions) orm.QuerySeter {
q := o.QueryTable(r.entityName) q := o.QueryTable(r.tableName)
if len(options) > 0 { if len(options) > 0 {
opts := options[0] opts := options[0]
q = q.Offset(opts.Offset) q = q.Offset(opts.Offset)
@@ -69,23 +70,66 @@ func (r *sqlRepository) put(id string, a interface{}) error {
}) })
} }
func (r *sqlRepository) purgeInactive(activeList interface{}, getId func(item interface{}) string) ([]string, error) { func paginateSlice(slice []string, skip int, size int) []string {
ids := persistence.CollectValue(activeList, getId) if skip > len(slice) {
var values []orm.Params skip = len(slice)
err := WithTx(func(o orm.Ormer) error { }
qs := r.newQuery(o).Exclude("id__in", ids)
num, err := qs.Values(&values, "id") end := skip + size
if num > 0 { if end > len(slice) {
_, err = qs.Delete() end = len(slice)
}
return slice[skip:end]
}
func difference(slice1 []string, slice2 []string) []string {
var diffStr []string
m := map[string]int{}
for _, s1Val := range slice1 {
m[s1Val] = 1
}
for _, s2Val := range slice2 {
m[s2Val] = m[s2Val] + 1
}
for mKey, mVal := range m {
if mVal == 1 {
diffStr = append(diffStr, mKey)
} }
return err }
})
return diffStr
}
func (r *sqlRepository) purgeInactive(activeList interface{}, getId func(item interface{}) string) ([]string, error) {
allIds, err := r.GetAllIds()
if err != nil { if err != nil {
return nil, err return nil, err
} }
result := make([]string, len(values)) activeIds := persistence.CollectValue(activeList, getId)
for i, v := range values { idsToDelete := difference(allIds, activeIds)
result[i] = v["ID"].(string) if len(idsToDelete) == 0 {
return nil, nil
} }
return result, nil log.Debug("Purging inactive records", "table", r.tableName, "total", len(idsToDelete))
err = WithTx(func(o orm.Ormer) error {
var offset int
for {
var subset = paginateSlice(idsToDelete, offset, 100)
if len(subset) == 0 {
break
}
log.Trace("-- Purging inactive records", "table", r.tableName, "num", len(subset), "from", offset)
offset += len(subset)
_, err := r.newQuery(o).Filter("id__in", subset).Delete()
if err != nil {
return err
}
}
return nil
})
return idsToDelete, err
} }