Optimize date formatting and repository/tracker stores

This commit is contained in:
Jon Staab
2024-12-09 13:37:07 -08:00
parent 7982bebb35
commit fba0482ec3
10 changed files with 115 additions and 99 deletions
+27 -67
View File
@@ -1,85 +1,45 @@
import {ctx, isNil} from "@welshman/lib"
import {Repository, Relay, LOCAL_RELAY_URL, getFilterResultCardinality} from "@welshman/util"
import type {TrustedEvent, Filter} from "@welshman/util"
import {Tracker, subscribe as baseSubscribe, SubscriptionEvent} from "@welshman/net"
import type {SubscribeRequestWithHandlers} from "@welshman/net"
import {throttle} from 'throttle-debounce'
import {Repository, Relay} from "@welshman/util"
import type {TrustedEvent} from "@welshman/util"
import {Tracker} from "@welshman/net"
import {custom} from "@welshman/store"
export const repository = new Repository<TrustedEvent>()
export const repositoryStore = custom(setter => {
const onUpdate = () => setter(repository)
onUpdate()
repository.on('update', onUpdate)
return () => repository.off('update', onUpdate)
}, {
set: (other: Repository) => repository.load(other.dump()),
})
export const relay = new Relay(repository)
export const tracker = new Tracker()
export const trackerStore = custom(setter => {
const onUpdate = () => setter(tracker)
// Adapt above objects to stores
onUpdate()
tracker.on('update', onUpdate)
export const makeRepositoryStore = ({throttle: t = 300}: {throttle?: number} = {}) =>
custom(setter => {
let onUpdate = () => setter(repository)
return () => tracker.off('update', onUpdate)
}, {
set: (other: Tracker) => tracker.load(other.relaysById),
})
export type PartialSubscribeRequest = Partial<SubscribeRequestWithHandlers> & {filters: Filter[]}
export const subscribe = (request: PartialSubscribeRequest) => {
const events: TrustedEvent[] = []
// If we already have all results for any filter, don't send the filter to the network
if (request.closeOnEose) {
for (const filter of request.filters.splice(0)) {
const cardinality = getFilterResultCardinality(filter)
if (!isNil(cardinality)) {
const results = repository.query([filter])
if (results.length === cardinality) {
for (const event of results) {
events.push(event)
}
break
}
}
request.filters.push(filter)
if (t) {
onUpdate = throttle(t, onUpdate)
}
}
// Make sure to query our local relay too
const delay = ctx.app.requestDelay
const authTimeout = ctx.app.authTimeout
const timeout = request.closeOnEose ? ctx.app.requestTimeout : 0
const sub = baseSubscribe({delay, timeout, authTimeout, relays: [], ...request})
onUpdate()
repository.on('update', onUpdate)
// Keep cached results async so the caller can set up handlers
setTimeout(() => {
for (const event of events) {
sub.emitter.emit(SubscriptionEvent.Event, LOCAL_RELAY_URL, event)
}
return () => repository.off('update', onUpdate)
}, {
set: (other: Repository) => repository.load(other.dump()),
})
return sub
}
export const makeTrackerStore = ({throttle: t = 300}: {throttle?: number} = {}) =>
custom(setter => {
let onUpdate = () => setter(tracker)
export const load = (request: PartialSubscribeRequest) =>
new Promise<TrustedEvent[]>(resolve => {
const sub = subscribe({closeOnEose: true, timeout: ctx.app.requestTimeout, ...request})
const events: TrustedEvent[] = []
if (t) {
onUpdate = throttle(t, onUpdate)
}
sub.emitter.on(SubscriptionEvent.Event, (url: string, e: TrustedEvent) => events.push(e))
sub.emitter.on(SubscriptionEvent.Complete, () => resolve(events))
onUpdate()
tracker.on('update', onUpdate)
return () => tracker.off('update', onUpdate)
}, {
set: (other: Tracker) => tracker.load(other.relaysById),
})