82 lines
1.5 KiB
Go

package arn
import (
"sort"
"sync"
"time"
"github.com/aerogo/nano"
)
// Activity is a user activity that appears in the follower's feeds.
type Activity interface {
Creator() *User
TypeName() string
GetID() ID
GetCreated() string
GetCreatedBy() UserID
GetCreatedTime() time.Time
}
// SortActivitiesLatestFirst puts the latest entries on top.
func SortActivitiesLatestFirst(entries []Activity) {
sort.Slice(entries, func(i, j int) bool {
return entries[i].GetCreated() > entries[j].GetCreated()
})
}
// StreamActivities returns a stream of all activities.
func StreamActivities() <-chan Activity {
channel := make(chan Activity, nano.ChannelBufferSize)
wg := sync.WaitGroup{}
wg.Add(2)
go func() {
for obj := range DB.All("ActivityCreate") {
channel <- obj.(Activity)
}
wg.Done()
}()
go func() {
for obj := range DB.All("ActivityConsumeAnime") {
channel <- obj.(Activity)
}
wg.Done()
}()
go func() {
wg.Wait()
close(channel)
}()
return channel
}
// AllActivities returns a slice of all activities.
func AllActivities() []Activity {
all := make([]Activity, 0, DB.Collection("ActivityCreate").Count()+DB.Collection("ActivityConsumeAnime").Count())
stream := StreamActivities()
for obj := range stream {
all = append(all, obj)
}
return all
}
// FilterActivities filters all Activities by a custom function.
func FilterActivities(filter func(Activity) bool) []Activity {
var filtered []Activity
for obj := range StreamActivities() {
if filter(obj) {
filtered = append(filtered, obj)
}
}
return filtered
}