laantungir/nostr-tools
1
0
Fork 0
mirror of https://github.com/nbd-wtf/nostr-tools.git synced 2025-12-08 16:28:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

get rid of RelayTrackingPool, merge it into SimplePool.

Browse Source
This commit is contained in:
fiatjaf
2023-12-17 19:15:27 -03:00
parent 2a7fd83be8
commit a0cb2eecae
1 changed files with 13 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
pool.ts
View File

@@ -11,6 +11,8 @@ export type SubscribeManyParams = Omit<SubscriptionParams, 'onclose'> & {
export class SimplePool { export class SimplePool {
private relays = new Map<string, Relay>() private relays = new Map<string, Relay>()
public seenOn = new Map<string, Set<Relay>>()
public trackRelays: boolean = false
async ensureRelay(url: string): Promise<Relay> { async ensureRelay(url: string): Promise<Relay> {
url = normalizeURL(url) url = normalizeURL(url)
@@ -29,6 +31,17 @@ export class SimplePool {
filters: Filter[], filters: Filter[],
params: SubscribeManyParams, params: SubscribeManyParams,
): Promise<{ close: () => void }> { ): Promise<{ close: () => void }> {
if (this.trackRelays) {
params.receivedEvent = (relay: Relay, id: string) => {
let set = this.seenOn.get(id)
if (!set) {
set = new Set()
this.seenOn.set(id, set)
}
set.add(relay)
}
}
const _knownIds = new Set<string>() const _knownIds = new Set<string>()
params.alreadyHaveEvent = (id: string) => { params.alreadyHaveEvent = (id: string) => {
if (params.alreadyHaveEvent?.(id)) { if (params.alreadyHaveEvent?.(id)) {
@@ -148,20 +161,3 @@ export class SimplePool {
}) })
} }
} }
export class RelayTrackingPool extends SimplePool {
public seenOn = new Map<string, Set<Relay>>()
subscribeMany(relays: string[], filters: Filter[], params: SubscribeManyParams): Promise<{ close: () => void }> {
params.receivedEvent = (relay: Relay, id: string) => {
let set = this.seenOn.get(id)
if (!set) {
set = new Set()
this.seenOn.set(id, set)
}
set.add(relay)
}
return super.subscribeMany(relays, filters, params)
}
}