mirror of
https://github.com/nbd-wtf/nostr-tools.git
synced 2025-12-08 16:28:49 +00:00
remove untilOpen promise that was causing memory leaks when a connection was never opened.
This commit is contained in:
140
magic.ts
Normal file
140
magic.ts
Normal file
@@ -0,0 +1,140 @@
|
|||||||
|
import {Relay, relayInit} from './relay'
|
||||||
|
import {Event} from './event'
|
||||||
|
import {normalizeURL} from './utils'
|
||||||
|
|
||||||
|
export default function (
|
||||||
|
writeableRelays: string[],
|
||||||
|
fallbackRelays: string[],
|
||||||
|
safeRelays: string[]
|
||||||
|
) {
|
||||||
|
return new MagicPool(fallbackRelays, writeableRelays, safeRelays)
|
||||||
|
}
|
||||||
|
|
||||||
|
class MagicPool {
|
||||||
|
private _conn: {[url: string]: Relay}
|
||||||
|
private _fallback: {[url: string]: Relay}
|
||||||
|
private _write: {[url: string]: Relay}
|
||||||
|
private _safe: {[url: string]: Relay}
|
||||||
|
|
||||||
|
private _profileRelays: {[pubkey: string]: RelayTableScore}
|
||||||
|
private _tempCache: {[id: string]: Event}
|
||||||
|
|
||||||
|
constructor(
|
||||||
|
fallbackRelays: string[],
|
||||||
|
writeableRelays: string[],
|
||||||
|
safeRelays: string[] = [
|
||||||
|
'wss://eden.nostr.land',
|
||||||
|
'wss://nostr.milou.lol',
|
||||||
|
'wss://relay.minds.com/nostr/v1/ws'
|
||||||
|
]
|
||||||
|
) {
|
||||||
|
this._conn = {}
|
||||||
|
this._write = {}
|
||||||
|
this._fallback = {}
|
||||||
|
this._profileRelays = {}
|
||||||
|
this._tempCache = {}
|
||||||
|
|
||||||
|
const hasEventId = (id: string): boolean => id in this._tempCache
|
||||||
|
const init = (url: string) => {
|
||||||
|
this._conn[normalizeURL(url)] = relayInit(normalizeURL(url), hasEventId)
|
||||||
|
}
|
||||||
|
|
||||||
|
fallbackRelays.forEach(init)
|
||||||
|
writeableRelays.forEach(init)
|
||||||
|
safeRelays.forEach(init)
|
||||||
|
|
||||||
|
this._write = Object.fromEntries(
|
||||||
|
writeableRelays.map(url => [
|
||||||
|
normalizeURL(url),
|
||||||
|
this._conn[normalizeURL(url)]
|
||||||
|
])
|
||||||
|
)
|
||||||
|
this._fallback = Object.fromEntries(
|
||||||
|
fallbackRelays.map(url => [
|
||||||
|
normalizeURL(url),
|
||||||
|
this._conn[normalizeURL(url)]
|
||||||
|
])
|
||||||
|
)
|
||||||
|
this._safe = Object.fromEntries(
|
||||||
|
safeRelays.map(url => [normalizeURL(url), this._conn[normalizeURL(url)]])
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
publish(event: Event) {
|
||||||
|
return Promise.all(
|
||||||
|
Object.entries(this._write).map(
|
||||||
|
([url, relay]) =>
|
||||||
|
new Promise(async resolve => {
|
||||||
|
await relay.connect()
|
||||||
|
let pub = relay.publish(event)
|
||||||
|
let to = setTimeout(() => {
|
||||||
|
let end = setTimeout(() => {
|
||||||
|
resolve({url, success: false, reason: 'timeout'})
|
||||||
|
}, 2500)
|
||||||
|
pub.on('seen', () => {
|
||||||
|
clearTimeout(end)
|
||||||
|
resolve({url, success: true, reason: 'seen'})
|
||||||
|
})
|
||||||
|
}, 2500)
|
||||||
|
pub.on('ok', () => {
|
||||||
|
clearTimeout(to)
|
||||||
|
resolve({url, success: true, reason: 'ok'})
|
||||||
|
})
|
||||||
|
pub.on('failed', (reason: string) => {
|
||||||
|
clearTimeout(to)
|
||||||
|
resolve({url, success: false, reason})
|
||||||
|
})
|
||||||
|
})
|
||||||
|
)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
profile(
|
||||||
|
pubkey: string,
|
||||||
|
onUpdate: (events: Event[]) => void
|
||||||
|
): {
|
||||||
|
page(n: number): void
|
||||||
|
} {
|
||||||
|
var relays = new Set()
|
||||||
|
let rts = this._profileRelays[pubkey]
|
||||||
|
if (rts) {
|
||||||
|
relays = rts.get(3)
|
||||||
|
}
|
||||||
|
|
||||||
|
let fallback = Object.values(this._fallback)
|
||||||
|
for (let i = 0; i < fallback.length; i++) {
|
||||||
|
if (relays.size < 3) {
|
||||||
|
relays.add(fallback[Math.floor(Math.random() * fallback.length)])
|
||||||
|
} else break
|
||||||
|
}
|
||||||
|
|
||||||
|
// start subscription
|
||||||
|
for (let r in relays) {
|
||||||
|
r.
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
page(n: number) {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class RelayTableScore {
|
||||||
|
seen: string[] = []
|
||||||
|
hinted: string[] = []
|
||||||
|
explicit: string[] = []
|
||||||
|
|
||||||
|
get(n: number): Set<string> {
|
||||||
|
let relays = new Set<string>()
|
||||||
|
for (let i = 0; i < n; i++) {
|
||||||
|
for (let j = 0; j < 3; j++) {
|
||||||
|
let v = [this.seen, this.explicit, this.hinted][j][i]
|
||||||
|
if (v) {
|
||||||
|
relays.add(v)
|
||||||
|
if (relays.size >= n) return relays
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return relays
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "nostr-tools",
|
"name": "nostr-tools",
|
||||||
"version": "1.4.0",
|
"version": "1.4.1",
|
||||||
"description": "Tools for making a Nostr client.",
|
"description": "Tools for making a Nostr client.",
|
||||||
"repository": {
|
"repository": {
|
||||||
"type": "git",
|
"type": "git",
|
||||||
|
|||||||
6
relay.ts
6
relay.ts
@@ -38,10 +38,6 @@ export type SubscriptionOptions = {
|
|||||||
export function relayInit(url: string): Relay {
|
export function relayInit(url: string): Relay {
|
||||||
var ws: WebSocket
|
var ws: WebSocket
|
||||||
var resolveClose: () => void
|
var resolveClose: () => void
|
||||||
var setOpen: (value: PromiseLike<void> | void) => void
|
|
||||||
var untilOpen = new Promise<void>(resolve => {
|
|
||||||
setOpen = resolve
|
|
||||||
})
|
|
||||||
var openSubs: {[id: string]: {filters: Filter[]} & SubscriptionOptions} = {}
|
var openSubs: {[id: string]: {filters: Filter[]} & SubscriptionOptions} = {}
|
||||||
var listeners: {
|
var listeners: {
|
||||||
connect: Array<() => void>
|
connect: Array<() => void>
|
||||||
@@ -74,7 +70,6 @@ export function relayInit(url: string): Relay {
|
|||||||
|
|
||||||
ws.onopen = () => {
|
ws.onopen = () => {
|
||||||
listeners.connect.forEach(cb => cb())
|
listeners.connect.forEach(cb => cb())
|
||||||
setOpen()
|
|
||||||
resolve()
|
resolve()
|
||||||
}
|
}
|
||||||
ws.onerror = () => {
|
ws.onerror = () => {
|
||||||
@@ -171,7 +166,6 @@ export function relayInit(url: string): Relay {
|
|||||||
async function trySend(params: [string, ...any]) {
|
async function trySend(params: [string, ...any]) {
|
||||||
let msg = JSON.stringify(params)
|
let msg = JSON.stringify(params)
|
||||||
|
|
||||||
await untilOpen
|
|
||||||
try {
|
try {
|
||||||
ws.send(msg)
|
ws.send(msg)
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
|
|||||||
Reference in New Issue
Block a user