82 lines
1.5 KiB
Go
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
|
|
}
|