relynolli-server/storage/catalog.go

86 lines
2.6 KiB
Go

package storage
import (
"context"
"fmt"
"relynolli-server/internal"
"relynolli-server/models/catalog"
filters2 "relynolli-server/models/filters"
"slices"
cmap "github.com/orcaman/concurrent-map/v2"
"github.com/uptrace/bun"
)
type StorageCatalog interface {
GetCatalogItem(ctx context.Context, id *int64) (*catalog.DBCatalog, error)
GetCatalogItems(ctx context.Context, filters cmap.ConcurrentMap[string, []string], limit int, offset int) (int, *[]catalog.DBCatalog, error)
GetFilters(ctx context.Context) (int, *[]filters2.DBFilter, error)
}
func NewStorageCatalog() StorageCatalog {
if instance == nil {
instance = &storage{
db: internal.InitDatabase().GetInstance(),
}
}
return instance
}
func (s *storage) GetCatalogItem(ctx context.Context, id *int64) (*catalog.DBCatalog, error) {
db := internal.InitDatabase().GetInstance()
model := new(catalog.DBCatalog)
err := db.NewSelect().Model(model).Where("id = ?", id).Where("available_quantity > 0").Where("is_available = 1").Scan(ctx)
if err != nil {
return nil, err
}
return model, nil
}
func buildFilterGroup(ctx context.Context, q *bun.SelectQuery, filters *cmap.ConcurrentMap[string, []string]) *bun.SelectQuery {
db := internal.InitDatabase().GetInstance()
availableFilters := new([]filters2.DBFilter)
//Get filters
db.NewSelect().Model(availableFilters).Scan(ctx)
for _, filter := range filters.Keys() {
filter = filter[0 : len(filter)-2]
if !slices.ContainsFunc(*availableFilters, func(elem filters2.DBFilter) bool {
return elem.Code == filter
}) {
continue
}
q = q.WhereGroup(" AND ", func(query *bun.SelectQuery) *bun.SelectQuery {
values, _ := filters.Get(filter + "[]")
for _, val := range values {
query = q.WhereOr(fmt.Sprintf("properties->>'$.%s' = ?", filter), val)
}
return query
})
}
return q
}
func (s *storage) GetCatalogItems(ctx context.Context, filters cmap.ConcurrentMap[string, []string], limit int, offset int) (int, *[]catalog.DBCatalog, error) {
db := internal.InitDatabase().GetInstance()
model := new([]catalog.DBCatalog)
filterQuery := db.NewSelect().Model(model).Where("is_active = 1").Where("available_quantity > 0")
count, _ := buildFilterGroup(ctx, filterQuery, &filters).Limit(limit).Offset(offset).Order("code").ScanAndCount(ctx)
return count, model, nil
}
func (s *storage) GetFilters(ctx context.Context) (int, *[]filters2.DBFilter, error) {
models := new([]filters2.DBFilter)
db := internal.InitDatabase().GetInstance()
count, err := db.NewSelect().Model(models).ScanAndCount(ctx)
if err != nil {
return 0, nil, err
}
return count, models, nil
}