pinafore/routes/_actions/addStatusOrNotification.js

101 lines
4.0 KiB
JavaScript
Raw Normal View History

import throttle from 'lodash-es/throttle'
2018-02-17 04:38:21 +01:00
import { mark, stop } from '../_utils/marks'
import { store } from '../_store/store'
import uniqBy from 'lodash-es/uniqBy'
import uniq from 'lodash-es/uniq'
import isEqual from 'lodash-es/isEqual'
import {
insertTimelineItems as insertTimelineItemsInDatabase
} from '../_database/timelines/insertion'
import { runMediumPriorityTask } from '../_utils/runMediumPriorityTask'
2018-02-17 04:38:21 +01:00
2018-04-03 03:02:09 +02:00
const STREAMING_THROTTLE_DELAY = 3000
function getExistingItemIdsSet (instanceName, timelineName) {
2018-02-17 04:38:21 +01:00
let timelineItemIds = store.getForTimeline(instanceName, timelineName, 'timelineItemIds') || []
return new Set(timelineItemIds)
2018-02-17 04:38:21 +01:00
}
function removeDuplicates (instanceName, timelineName, updates) {
2018-02-17 04:38:21 +01:00
// remove duplicates, including duplicates due to reblogs
let existingItemIds = getExistingItemIdsSet(instanceName, timelineName)
2018-02-17 04:38:21 +01:00
return updates.filter(update => !existingItemIds.has(update.id))
}
2018-03-10 07:31:26 +01:00
async function insertUpdatesIntoTimeline (instanceName, timelineName, updates) {
updates = removeDuplicates(instanceName, timelineName, updates)
if (!updates.length) {
return
}
2018-03-10 07:31:26 +01:00
await insertTimelineItemsInDatabase(instanceName, timelineName, updates)
2018-03-10 07:31:26 +01:00
let itemIdsToAdd = store.getForTimeline(instanceName, timelineName, 'itemIdsToAdd') || []
let newItemIdsToAdd = uniq([].concat(itemIdsToAdd).concat(updates.map(_ => _.id)))
if (!isEqual(itemIdsToAdd, newItemIdsToAdd)) {
console.log('adding ', (newItemIdsToAdd.length - itemIdsToAdd.length),
'items to itemIdsToAdd for timeline', timelineName)
store.setForTimeline(instanceName, timelineName, {itemIdsToAdd: newItemIdsToAdd})
}
2018-03-10 07:31:26 +01:00
}
async function insertUpdatesIntoThreads (instanceName, updates) {
if (!updates.length) {
return
}
2018-03-11 01:21:10 +01:00
let threads = store.getThreads(instanceName)
2018-03-10 07:31:26 +01:00
for (let timelineName of Object.keys(threads)) {
let thread = threads[timelineName]
let updatesForThisThread = updates.filter(
status => thread.includes(status.in_reply_to_id) && !thread.includes(status.id)
)
if (!updatesForThisThread.length) {
continue
}
2018-03-10 07:31:26 +01:00
let itemIdsToAdd = store.getForTimeline(instanceName, timelineName, 'itemIdsToAdd') || []
let newItemIdsToAdd = uniq([].concat(itemIdsToAdd).concat(updatesForThisThread.map(_ => _.id)))
if (!isEqual(itemIdsToAdd, newItemIdsToAdd)) {
console.log('adding ', (newItemIdsToAdd.length - itemIdsToAdd.length),
'items to itemIdsToAdd for thread', timelineName)
store.setForTimeline(instanceName, timelineName, {itemIdsToAdd: newItemIdsToAdd})
2018-03-10 07:31:26 +01:00
}
}
}
2018-02-17 04:38:21 +01:00
async function processFreshUpdates (instanceName, timelineName) {
mark('processFreshUpdates')
let freshUpdates = store.getForTimeline(instanceName, timelineName, 'freshUpdates')
if (freshUpdates && freshUpdates.length) {
let updates = freshUpdates.slice()
store.setForTimeline(instanceName, timelineName, {freshUpdates: []})
await Promise.all([
insertUpdatesIntoTimeline(instanceName, timelineName, updates),
insertUpdatesIntoThreads(instanceName, updates.filter(status => status.in_reply_to_id))
])
2018-02-17 04:38:21 +01:00
}
2018-03-10 07:31:26 +01:00
stop('processFreshUpdates')
2018-02-17 04:38:21 +01:00
}
const lazilyProcessFreshUpdates = throttle((instanceName, timelineName) => {
runMediumPriorityTask(() => {
2018-02-17 04:38:21 +01:00
/* no await */ processFreshUpdates(instanceName, timelineName)
})
2018-04-03 03:02:09 +02:00
}, STREAMING_THROTTLE_DELAY)
2018-02-17 04:38:21 +01:00
export function addStatusOrNotification (instanceName, timelineName, newStatusOrNotification) {
2018-03-19 18:09:05 +01:00
addStatusesOrNotifications(instanceName, timelineName, [newStatusOrNotification])
}
export function addStatusesOrNotifications (instanceName, timelineName, newStatusesOrNotifications) {
console.log('addStatusesOrNotifications', Date.now())
2018-02-17 04:38:21 +01:00
let freshUpdates = store.getForTimeline(instanceName, timelineName, 'freshUpdates') || []
freshUpdates = [].concat(freshUpdates).concat(newStatusesOrNotifications)
freshUpdates = uniqBy(freshUpdates, _ => _.id)
2018-02-17 04:38:21 +01:00
store.setForTimeline(instanceName, timelineName, {freshUpdates: freshUpdates})
lazilyProcessFreshUpdates(instanceName, timelineName)
}