132 lines
2.7 KiB
Go
132 lines
2.7 KiB
Go
package persistence
|
|
|
|
import (
|
|
"github.com/Masterminds/squirrel"
|
|
"github.com/astaxie/beego/orm"
|
|
"github.com/deluan/navidrome/model"
|
|
)
|
|
|
|
type sqlRepository struct {
|
|
tableName string
|
|
ormer orm.Ormer
|
|
}
|
|
|
|
func (r *sqlRepository) newQuery(options ...model.QueryOptions) orm.QuerySeter {
|
|
q := r.ormer.QueryTable(r.tableName)
|
|
if len(options) > 0 {
|
|
opts := options[0]
|
|
q = q.Offset(opts.Offset)
|
|
if opts.Max > 0 {
|
|
q = q.Limit(opts.Max)
|
|
}
|
|
if opts.Sort != "" {
|
|
if opts.Order == "desc" {
|
|
q = q.OrderBy("-" + opts.Sort)
|
|
} else {
|
|
q = q.OrderBy(opts.Sort)
|
|
}
|
|
}
|
|
for field, value := range opts.Filters {
|
|
q = q.Filter(field, value)
|
|
}
|
|
}
|
|
return q
|
|
}
|
|
|
|
func (r *sqlRepository) newRawQuery(options ...model.QueryOptions) squirrel.SelectBuilder {
|
|
sq := squirrel.Select("*").From(r.tableName)
|
|
if len(options) > 0 {
|
|
if options[0].Max > 0 {
|
|
sq = sq.Limit(uint64(options[0].Max))
|
|
}
|
|
if options[0].Offset > 0 {
|
|
sq = sq.Offset(uint64(options[0].Max))
|
|
}
|
|
if options[0].Sort != "" {
|
|
if options[0].Order == "desc" {
|
|
sq = sq.OrderBy(options[0].Sort + " desc")
|
|
} else {
|
|
sq = sq.OrderBy(options[0].Sort)
|
|
}
|
|
}
|
|
}
|
|
return sq
|
|
}
|
|
|
|
func (r *sqlRepository) CountAll() (int64, error) {
|
|
return r.newQuery().Count()
|
|
}
|
|
|
|
func (r *sqlRepository) Exists(id string) (bool, error) {
|
|
c, err := r.newQuery().Filter("id", id).Count()
|
|
return c == 1, err
|
|
}
|
|
|
|
// "Hack" to bypass Postgres driver limitation
|
|
func (r *sqlRepository) insert(record interface{}) error {
|
|
_, err := r.ormer.Insert(record)
|
|
if err != nil && err.Error() != "LastInsertId is not supported by this driver" {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *sqlRepository) put(id string, a interface{}) error {
|
|
c, err := r.newQuery().Filter("id", id).Count()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if c == 0 {
|
|
err = r.insert(a)
|
|
if err != nil && err.Error() == "LastInsertId is not supported by this driver" {
|
|
err = nil
|
|
}
|
|
return err
|
|
}
|
|
_, err = r.ormer.Update(a)
|
|
return err
|
|
}
|
|
|
|
func paginateSlice(slice []string, skip int, size int) []string {
|
|
if skip > len(slice) {
|
|
skip = len(slice)
|
|
}
|
|
|
|
end := skip + size
|
|
if end > len(slice) {
|
|
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 diffStr
|
|
}
|
|
|
|
func (r *sqlRepository) Delete(id string) error {
|
|
_, err := r.newQuery().Filter("id", id).Delete()
|
|
return err
|
|
}
|
|
|
|
func (r *sqlRepository) DeleteAll() error {
|
|
_, err := r.newQuery().Filter("id__isnull", false).Delete()
|
|
return err
|
|
}
|