1
0
forked from Kieran/snort

useSubscribe, handle emitted requests in sqlite

This commit is contained in:
Martti Malmi 2024-01-18 13:47:07 +02:00
parent b7e61ebde5
commit 2ea516e636
14 changed files with 99 additions and 40 deletions

View File

@ -8,7 +8,7 @@ export async function sendEventToRelays(
setResults?: (x: Array<OkResponse>) => void, setResults?: (x: Array<OkResponse>) => void,
) { ) {
if (customRelays) { if (customRelays) {
system.HandleEvent({ ...ev, relays: [] }); system.HandleEvent("*", { ...ev, relays: [] });
return removeUndefined( return removeUndefined(
await Promise.all( await Promise.all(
customRelays.map(async r => { customRelays.map(async r => {

View File

@ -36,10 +36,10 @@ const TimelineFollows = (props: TimelineFollowsProps) => {
const [latest, setLatest] = useHistoryState(unixNow(), "TimelineFollowsLatest"); const [latest, setLatest] = useHistoryState(unixNow(), "TimelineFollowsLatest");
const [limit, setLimit] = useState(50); const [limit, setLimit] = useState(50);
const feed = useFollowsTimelineView(limit); const feed = useFollowsTimelineView(limit);
console.log("feed", feed);
const { muted, isEventMuted } = useModeration(); const { muted, isEventMuted } = useModeration();
const sortedFeed = useMemo(() => orderDescending(feed), [feed]); const oldest = useMemo(() => feed.at(-1)?.created_at, [feed]);
const oldest = useMemo(() => sortedFeed.at(-1)?.created_at, [sortedFeed]);
const postsOnly = useCallback( const postsOnly = useCallback(
(a: NostrEvent) => (props.postsOnly ? !a.tags.some(b => b[0] === "e" || b[0] === "a") : true), (a: NostrEvent) => (props.postsOnly ? !a.tags.some(b => b[0] === "e" || b[0] === "a") : true),
@ -58,8 +58,8 @@ const TimelineFollows = (props: TimelineFollowsProps) => {
const mixin = useHashtagsFeed(); const mixin = useHashtagsFeed();
const mainFeed = useMemo(() => { const mainFeed = useMemo(() => {
return filterPosts((sortedFeed ?? []).filter(a => a.created_at <= latest)); return filterPosts((feed ?? []).filter(a => a.created_at <= latest));
}, [sortedFeed, filterPosts, latest, login.follows.timestamp]); }, [feed, filterPosts, latest, login.follows.timestamp]);
const findHashTagContext = (a: NostrEvent) => { const findHashTagContext = (a: NostrEvent) => {
const tag = a.tags.filter(a => a[0] === "t").find(a => login.tags.item.includes(a[1].toLowerCase()))?.[1]; const tag = a.tags.filter(a => a[0] === "t").find(a => login.tags.item.includes(a[1].toLowerCase()))?.[1];
@ -81,12 +81,12 @@ const TimelineFollows = (props: TimelineFollowsProps) => {
}, [mixin, mainFeed, postsOnly, isEventMuted]); }, [mixin, mainFeed, postsOnly, isEventMuted]);
const latestFeed = useMemo(() => { const latestFeed = useMemo(() => {
return filterPosts((sortedFeed ?? []).filter(a => a.created_at > latest)); return filterPosts((feed ?? []).filter(a => a.created_at > latest));
}, [sortedFeed, latest]); }, [feed, latest]);
const liveStreams = useMemo(() => { const liveStreams = useMemo(() => {
return (sortedFeed ?? []).filter(a => a.kind === EventKind.LiveEvent && findTag(a, "status") === "live"); return (feed ?? []).filter(a => a.kind === EventKind.LiveEvent && findTag(a, "status") === "live");
}, [sortedFeed]); }, [feed]);
const latestAuthors = useMemo(() => { const latestAuthors = useMemo(() => {
return dedupeByPubkey(latestFeed).map(e => e.pubkey); return dedupeByPubkey(latestFeed).map(e => e.pubkey);
@ -120,7 +120,7 @@ const TimelineFollows = (props: TimelineFollowsProps) => {
}} }}
displayAs={displayAs} displayAs={displayAs}
/> />
{sortedFeed.length > 0 && ( {feed.length > 0 && (
<ShowMoreInView <ShowMoreInView
onClick={() => { onClick={() => {
setLimit(s => s + 20); setLimit(s => s + 20);

View File

@ -35,7 +35,7 @@ export default function TrendingNotes({ count = Infinity, small = false }: { cou
console.error(`Event with invalid sig\n\n${ev}\n\nfrom ${trendingNotesUrl}`); console.error(`Event with invalid sig\n\n${ev}\n\nfrom ${trendingNotesUrl}`);
return; return;
} }
System.HandleEvent(ev as TaggedNostrEvent); System.HandleEvent("*", ev as TaggedNostrEvent);
return ev; return ev;
}), }),
); );

View File

@ -4,6 +4,7 @@ import { useRequestBuilder } from "@snort/system-react";
import { useEffect, useMemo, useState } from "react"; import { useEffect, useMemo, useState } from "react";
import useLogin from "@/Hooks/useLogin"; import useLogin from "@/Hooks/useLogin";
import useSubscribe from "@/Hooks/useSubscribe";
import { Relay } from "@/system"; import { Relay } from "@/system";
import { Day } from "@/Utils/Const"; import { Day } from "@/Utils/Const";
@ -77,16 +78,15 @@ export function useFollowsTimelineView(limit = 20) {
const follows = useLogin(s => s.follows.item); const follows = useLogin(s => s.follows.item);
const kinds = [EventKind.TextNote, EventKind.Repost, EventKind.Polls]; const kinds = [EventKind.TextNote, EventKind.Repost, EventKind.Polls];
const filter = useMemo(() => { const filter = useMemo(
return [ () => ({
{ authors: follows,
authors: follows, kinds,
kinds, limit,
limit, }),
}, [follows, limit],
]; );
}, [follows, limit]); return useSubscribe("follows-timeline", filter);
return useWorkerRelayView("follows-timeline", filter, Day * 7);
} }
export function useNotificationsView() { export function useNotificationsView() {
@ -101,7 +101,7 @@ export function useNotificationsView() {
}, },
]; ];
}, [publicKey]); }, [publicKey]);
return useWorkerRelayView("notifications", req, Day * 30); return useSubscribe("notifications", req[0]);
} }
export function useReactionsView(ids: Array<NostrLink>, leaveOpen = true) { export function useReactionsView(ids: Array<NostrLink>, leaveOpen = true) {
@ -123,7 +123,7 @@ export function useReactionsView(ids: Array<NostrLink>, leaveOpen = true) {
return rb.buildRaw(); return rb.buildRaw();
}, [ids]); }, [ids]);
return useWorkerRelayView("reactions", req, undefined); return useSubscribe("reactions", req[0]);
} }
export function useReactionsViewCount(ids: Array<NostrLink>, leaveOpen = true) { export function useReactionsViewCount(ids: Array<NostrLink>, leaveOpen = true) {
@ -160,5 +160,5 @@ export function useFollowsContactListView() {
}, },
]; ];
}, [follows]); }, [follows]);
return useWorkerRelayView("follows-contacts-relays", filter, undefined); return useSubscribe("follows-contacts-relays", filter[0]);
} }

View File

@ -0,0 +1,31 @@
import { ReqFilter, RequestBuilder, TaggedNostrEvent } from "@snort/system";
import inMemoryDB from "@snort/system/src/InMemoryDB";
import { useRequestBuilder } from "@snort/system-react";
import { useEffect, useMemo, useState } from "react";
import { System } from "@/system";
export default function useSubscribe(id: string, filter: ReqFilter): TaggedNostrEvent[] {
const getEvents = () => inMemoryDB.findArray(filter);
const [events, setEvents] = useState(getEvents());
const rb = useMemo(() => {
const rb = new RequestBuilder(id);
rb.withBareFilter(filter);
return rb;
}, [id, filter]);
useRequestBuilder(rb);
useEffect(() => {
const cb = (subId: string) => {
if (subId === id) {
setEvents(getEvents());
}
};
System.on("event", cb);
return () => {
System.off("event", cb);
};
}, [id, filter]);
return events as Array<TaggedNostrEvent>;
}

View File

@ -91,7 +91,7 @@ export class Nip29ChatSystem extends ExternalStore<Array<Chat>> implements ChatS
}, },
sendMessage: async (ev, system: SystemInterface) => { sendMessage: async (ev, system: SystemInterface) => {
ev.forEach(async a => { ev.forEach(async a => {
system.HandleEvent({ ...a, relays: [] }); system.HandleEvent("*", { ...a, relays: [] });
await system.WriteOnceToRelay(`wss://${relay}`, a); await system.WriteOnceToRelay(`wss://${relay}`, a);
}); });
}, },

View File

@ -1,4 +1,5 @@
import { removeUndefined, throwIfOffline } from "@snort/shared"; import { removeUndefined, throwIfOffline } from "@snort/shared";
import LRUSet from "@snort/shared/src/LRUSet";
import { mapEventToProfile, NostrEvent, NostrSystem, ProfileLoaderService, socialGraphInstance } from "@snort/system"; import { mapEventToProfile, NostrEvent, NostrSystem, ProfileLoaderService, socialGraphInstance } from "@snort/system";
import { WorkerRelayInterface } from "@snort/worker-relay"; import { WorkerRelayInterface } from "@snort/worker-relay";
import WorkerRelayPath from "@snort/worker-relay/dist/worker?worker&url"; import WorkerRelayPath from "@snort/worker-relay/dist/worker?worker&url";
@ -70,9 +71,21 @@ export async function initRelayWorker() {
if (await Relay.init()) { if (await Relay.init()) {
if (await Relay.open()) { if (await Relay.open()) {
await Relay.migrate(); await Relay.migrate();
const seen = new LRUSet<string>(100);
System.on("event", async (_, ev) => { System.on("event", async (_, ev) => {
if (seen.has(ev.id)) return;
seen.add(ev.id);
await Relay.event(ev); await Relay.event(ev);
}); });
System.on("request", async (subId, f) => {
const evs = await Relay.req(["REQ", "", ...f.filters]);
evs.forEach(ev => {
seen.add(ev.id);
queueMicrotask(() => {
System.HandleEvent(subId, { ...ev, relays: [] });
});
});
});
} }
} }
} catch (e) { } catch (e) {

View File

@ -131,6 +131,10 @@ class InMemoryDB {
} }
} }
count(filter: Filter): number {
return this.findArray(filter).length;
}
find(filter: Filter, callback: (event: TaggedNostrEvent) => void): void { find(filter: Filter, callback: (event: TaggedNostrEvent) => void): void {
this.findArray(filter).forEach(event => { this.findArray(filter).forEach(event => {
callback(event); callback(event);

View File

@ -10,6 +10,8 @@ import { NostrEvent, ReqCommand, ReqFilter, TaggedNostrEvent, u256 } from "./nos
import { RelayInfo } from "./relay-info"; import { RelayInfo } from "./relay-info";
import EventKind from "./event-kind"; import EventKind from "./event-kind";
import { EventExt } from "./event-ext"; import { EventExt } from "./event-ext";
import { getHex64 } from "./utils";
import inMemoryDB from "./InMemoryDB";
/** /**
* Relay settings * Relay settings
@ -210,6 +212,12 @@ export class Connection extends EventEmitter<ConnectionEvents> {
} }
*/ */
const id = getHex64(e.data as string, "id");
if (inMemoryDB.has(id)) {
this.#log("Already have, skip processing %s", id);
return;
}
const msg = JSON.parse(e.data as string) as Array<string | NostrEvent | boolean>; const msg = JSON.parse(e.data as string) as Array<string | NostrEvent | boolean>;
const tag = msg[0] as string; const tag = msg[0] as string;
switch (tag) { switch (tag) {
@ -232,7 +240,7 @@ export class Connection extends EventEmitter<ConnectionEvents> {
} as TaggedNostrEvent; } as TaggedNostrEvent;
if (!EventExt.isValid(ev)) { if (!EventExt.isValid(ev)) {
//this.#log("Rejecting invalid event %O", ev); this.#log("Rejecting invalid event %O", ev);
return; return;
} }
this.emit("event", msg[1] as string, ev); this.emit("event", msg[1] as string, ev);

View File

@ -109,7 +109,7 @@ export interface SystemInterface {
/** /**
* Push an event into the system from external source * Push an event into the system from external source
*/ */
HandleEvent(ev: TaggedNostrEvent): void; HandleEvent(subId: string, ev: TaggedNostrEvent): void;
/** /**
* Send an event to all permanent connections * Send an event to all permanent connections

View File

@ -24,12 +24,13 @@ import { RelayMetadataLoader } from "./outbox-model";
import { Optimizer, DefaultOptimizer } from "./query-optimizer"; import { Optimizer, DefaultOptimizer } from "./query-optimizer";
import { ConnectionPool, DefaultConnectionPool } from "./connection-pool"; import { ConnectionPool, DefaultConnectionPool } from "./connection-pool";
import { QueryManager } from "./query-manager"; import { QueryManager } from "./query-manager";
import inMemoryDB from "./InMemoryDB";
export interface NostrSystemEvents { export interface NostrSystemEvents {
change: (state: SystemSnapshot) => void; change: (state: SystemSnapshot) => void;
auth: (challenge: string, relay: string, cb: (ev: NostrEvent) => void) => void; auth: (challenge: string, relay: string, cb: (ev: NostrEvent) => void) => void;
event: (subId: string, ev: TaggedNostrEvent) => void; event: (subId: string, ev: TaggedNostrEvent) => void;
filters: (filter: BuiltRawReqFilter) => void; request: (subId: string, filter: BuiltRawReqFilter) => void;
} }
export interface NostrsystemProps { export interface NostrsystemProps {
@ -122,6 +123,7 @@ export class NostrSystem extends EventEmitter<NostrSystemEvents> implements Syst
this.pool.on("event", (_, sub, ev) => { this.pool.on("event", (_, sub, ev) => {
ev.relays?.length && this.relayMetricsHandler.onEvent(ev.relays[0]); ev.relays?.length && this.relayMetricsHandler.onEvent(ev.relays[0]);
this.emit("event", sub, ev); this.emit("event", sub, ev);
inMemoryDB.handleEvent(ev);
}); });
this.pool.on("disconnect", (id, code) => { this.pool.on("disconnect", (id, code) => {
const c = this.pool.getConnection(id); const c = this.pool.getConnection(id);
@ -148,7 +150,7 @@ export class NostrSystem extends EventEmitter<NostrSystemEvents> implements Syst
this.#queryManager.on("trace", t => { this.#queryManager.on("trace", t => {
this.relayMetricsHandler.onTraceReport(t); this.relayMetricsHandler.onTraceReport(t);
}); });
this.#queryManager.on("filters", (f: BuiltRawReqFilter) => this.emit("filters", f)); this.#queryManager.on("request", (subId: string, f: BuiltRawReqFilter) => this.emit("request", subId, f));
} }
get Sockets(): ConnectionStateSnapshot[] { get Sockets(): ConnectionStateSnapshot[] {
@ -189,13 +191,14 @@ export class NostrSystem extends EventEmitter<NostrSystemEvents> implements Syst
return this.#queryManager.query(req); return this.#queryManager.query(req);
} }
HandleEvent(ev: TaggedNostrEvent) { HandleEvent(subId: string, ev: TaggedNostrEvent) {
this.emit("event", "*", ev); inMemoryDB.handleEvent(ev);
this.emit("event", subId, ev);
this.#queryManager.handleEvent(ev); this.#queryManager.handleEvent(ev);
} }
async BroadcastEvent(ev: NostrEvent, cb?: (rsp: OkResponse) => void): Promise<OkResponse[]> { async BroadcastEvent(ev: NostrEvent, cb?: (rsp: OkResponse) => void): Promise<OkResponse[]> {
this.HandleEvent({ ...ev, relays: [] }); this.HandleEvent("*", { ...ev, relays: [] });
return await this.pool.broadcast(this, ev, cb); return await this.pool.broadcast(this, ev, cb);
} }

View File

@ -8,7 +8,7 @@ import { trimFilters } from "./request-trim";
interface QueryManagerEvents { interface QueryManagerEvents {
change: () => void; change: () => void;
trace: (report: TraceReport) => void; trace: (report: TraceReport) => void;
filters: (req: BuiltRawReqFilter) => void; request: (subId: string, req: BuiltRawReqFilter) => void;
} }
/** /**
@ -57,9 +57,9 @@ export class QueryManager extends EventEmitter<QueryManagerEvents> {
} else { } else {
const q = new Query(this.#system, req); const q = new Query(this.#system, req);
q.on("trace", r => this.emit("trace", r)); q.on("trace", r => this.emit("trace", r));
q.on("filters", fx => { q.on("request", (id, fx) => {
this.#send(q, fx); this.#send(q, fx);
this.emit("filters", fx); this.emit("request", id, fx);
}); });
this.#queries.set(req.id, q); this.#queries.set(req.id, q);
@ -78,7 +78,7 @@ export class QueryManager extends EventEmitter<QueryManagerEvents> {
async fetch(req: RequestBuilder, cb?: (evs: Array<TaggedNostrEvent>) => void) { async fetch(req: RequestBuilder, cb?: (evs: Array<TaggedNostrEvent>) => void) {
const q = new Query(this.#system, req); const q = new Query(this.#system, req);
q.on("trace", r => this.emit("trace", r)); q.on("trace", r => this.emit("trace", r));
q.on("filters", fx => { q.on("request", (subId, fx) => {
this.#send(q, fx); this.#send(q, fx);
}); });
if (cb) { if (cb) {

View File

@ -100,7 +100,7 @@ export interface TraceReport {
export interface QueryEvents { export interface QueryEvents {
loading: (v: boolean) => void; loading: (v: boolean) => void;
trace: (report: TraceReport) => void; trace: (report: TraceReport) => void;
filters: (req: BuiltRawReqFilter) => void; request: (subId: string, req: BuiltRawReqFilter) => void;
event: (evs: Array<TaggedNostrEvent>) => void; event: (evs: Array<TaggedNostrEvent>) => void;
end: () => void; end: () => void;
} }
@ -329,11 +329,11 @@ export class Query extends EventEmitter<QueryEvents> {
if (!(this.request.options?.skipDiff ?? false) && existing.length > 0) { if (!(this.request.options?.skipDiff ?? false) && existing.length > 0) {
const filters = this.request.buildDiff(this.#system, existing); const filters = this.request.buildDiff(this.#system, existing);
this.#log("Build %s %O", this.id, filters); this.#log("Build %s %O", this.id, filters);
filters.forEach(f => this.emit("filters", f)); filters.forEach(f => this.emit("request", this.id, f));
} else { } else {
const filters = this.request.build(this.#system); const filters = this.request.build(this.#system);
this.#log("Build %s %O", this.id, filters); this.#log("Build %s %O", this.id, filters);
filters.forEach(f => this.emit("filters", f)); filters.forEach(f => this.emit("request", this.id, f));
} }
} }

View File

@ -124,7 +124,7 @@ export class SystemWorker extends EventEmitter<NostrSystemEvents> implements Sys
this.#workerRpc(WorkerCommand.DisconnectRelay, address); this.#workerRpc(WorkerCommand.DisconnectRelay, address);
} }
HandleEvent(ev: TaggedNostrEvent): void { HandleEvent(subId: string, ev: TaggedNostrEvent): void {
throw new Error("Method not implemented."); throw new Error("Method not implemented.");
} }