mirror of https://github.com/gabehf/Koito.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
2.7 KiB
110 lines
2.7 KiB
package psql
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/gabehf/koito/internal/db"
|
|
"github.com/gabehf/koito/internal/repository"
|
|
)
|
|
|
|
func (p *Psql) CountListens(ctx context.Context, period db.Period) (int64, error) {
|
|
t2 := time.Now()
|
|
t1 := db.StartTimeFromPeriod(period)
|
|
count, err := p.q.CountListens(ctx, repository.CountListensParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
func (p *Psql) CountTracks(ctx context.Context, period db.Period) (int64, error) {
|
|
t2 := time.Now()
|
|
t1 := db.StartTimeFromPeriod(period)
|
|
count, err := p.q.CountTopTracks(ctx, repository.CountTopTracksParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
func (p *Psql) CountAlbums(ctx context.Context, period db.Period) (int64, error) {
|
|
t2 := time.Now()
|
|
t1 := db.StartTimeFromPeriod(period)
|
|
count, err := p.q.CountTopReleases(ctx, repository.CountTopReleasesParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
func (p *Psql) CountArtists(ctx context.Context, period db.Period) (int64, error) {
|
|
t2 := time.Now()
|
|
t1 := db.StartTimeFromPeriod(period)
|
|
count, err := p.q.CountTopArtists(ctx, repository.CountTopArtistsParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
func (p *Psql) CountTimeListened(ctx context.Context, period db.Period) (int64, error) {
|
|
t2 := time.Now()
|
|
t1 := db.StartTimeFromPeriod(period)
|
|
count, err := p.q.CountTimeListened(ctx, repository.CountTimeListenedParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
func (p *Psql) CountTimeListenedToItem(ctx context.Context, opts db.TimeListenedOpts) (int64, error) {
|
|
t2 := time.Now()
|
|
t1 := db.StartTimeFromPeriod(opts.Period)
|
|
|
|
if opts.ArtistID > 0 {
|
|
count, err := p.q.CountTimeListenedToArtist(ctx, repository.CountTimeListenedToArtistParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
ArtistID: opts.ArtistID,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
} else if opts.AlbumID > 0 {
|
|
count, err := p.q.CountTimeListenedToRelease(ctx, repository.CountTimeListenedToReleaseParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
ReleaseID: opts.AlbumID,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
|
|
} else if opts.TrackID > 0 {
|
|
count, err := p.q.CountTimeListenedToTrack(ctx, repository.CountTimeListenedToTrackParams{
|
|
ListenedAt: t1,
|
|
ListenedAt_2: t2,
|
|
ID: opts.TrackID,
|
|
})
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
return 0, errors.New("an id must be provided")
|
|
}
|