All files / lib/gossip GossipMemoryStore.ts

98.11% Statements 52/53
90% Branches 9/10
100% Functions 21/21
100% Lines 51/51

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142    1x       1x             1x 334x 334x 334x 334x 334x     249x       369x       367x       157x 157x 157x 146x   157x 157x       499x       499x       499x 499x 499x 256x 254x   256x       1032x       18x       37x           1018x           3x 3x             796x       7x 7x 7x     7x     7x 6x       7x 7x     7x 7x       244x       243x 243x       314x 314x 314x         2844x       2728x       1553x    
/* eslint-disable @typescript-eslint/require-await */
import { ShortChannelId } from "@node-lightning/core";
import { shortChannelIdFromNumber } from "@node-lightning/core";
import { OutPoint } from "@node-lightning/core";
import { ChannelAnnouncementMessage } from "../messages/ChannelAnnouncementMessage";
import { ChannelUpdateMessage } from "../messages/ChannelUpdateMessage";
import { ExtendedChannelAnnouncementMessage } from "../messages/ExtendedChannelAnnouncementMessage";
import { NodeAnnouncementMessage } from "../messages/NodeAnnouncementMessage";
import { IGossipStore } from "./GossipStore";
 
/**
 * In-memory implementation of the IGossipStore.
 */
export class GossipMemoryStore implements IGossipStore {
    private _channelAnn = new Map<bigint, ChannelAnnouncementMessage>();
    private _channelByOutPoint = new Map<string, bigint>();
    private _channelUpd = new Map<bigint, ChannelUpdateMessage>();
    private _nodeAnn = new Map<string, NodeAnnouncementMessage>();
    private _nodeChannels = new Map<string, Set<bigint>>();
 
    public get channelAnnouncementCount() {
        return this._channelAnn.size;
    }
 
    public get channelUpdateCount() {
        return this._channelUpd.size;
    }
 
    public get nodeAnnouncementCount() {
        return this._nodeAnn.size;
    }
 
    public async saveChannelAnnouncement(msg: ChannelAnnouncementMessage): Promise<void> {
        const chanKey = getChanKey(msg.shortChannelId);
        this._channelAnn.set(chanKey, msg);
        if (msg instanceof ExtendedChannelAnnouncementMessage) {
            this._channelByOutPoint.set(msg.outpoint.toString(), msg.shortChannelId.toNumber());
        }
        await this._saveNodeChannel(msg.nodeId1, chanKey);
        await this._saveNodeChannel(msg.nodeId2, chanKey);
    }
 
    public async saveChannelUpdate(msg: ChannelUpdateMessage): Promise<void> {
        this._channelUpd.set(getChanUpdKey(getChanKey(msg.shortChannelId), msg.direction), msg);
    }
 
    public async saveNodeAnnouncement(msg: NodeAnnouncementMessage): Promise<void> {
        this._nodeAnn.set(getNodeKey(msg.nodeId), msg);
    }
 
    public async findChannelsForNode(nodeId: Buffer): Promise<ShortChannelId[]> {
        const scidInts = this._nodeChannels.get(getNodeKey(nodeId));
        const results: ShortChannelId[] = [];
        if (!scidInts) return results;
        for (const scidInt of scidInts) {
            results.push(shortChannelIdFromNumber(scidInt));
        }
        return results;
    }
 
    public async findNodeAnnouncement(nodeId: Buffer): Promise<NodeAnnouncementMessage> {
        return this._nodeAnn.get(getNodeKey(nodeId));
    }
 
    public async findNodeAnnouncements(): Promise<NodeAnnouncementMessage[]> {
        return Array.from(this._nodeAnn.values());
    }
 
    public async findChannelAnnouncemnts(): Promise<ChannelAnnouncementMessage[]> {
        return Array.from(this._channelAnn.values());
    }
 
    public async findChannelAnnouncement(
        scid: ShortChannelId,
    ): Promise<ChannelAnnouncementMessage> {
        return this._channelAnn.get(getChanKey(scid));
    }
 
    public async findChannelAnnouncementByOutpoint(
        outpoint: OutPoint,
    ): Promise<ChannelAnnouncementMessage> {
        const scidNum = this._channelByOutPoint.get(outpoint.toString());
        return this._channelAnn.get(scidNum);
    }
 
    public async findChannelUpdate(
        scid: ShortChannelId,
        dir: number,
    ): Promise<ChannelUpdateMessage> {
        return this._channelUpd.get(getChanUpdKey(getChanKey(scid), dir));
    }
 
    public async deleteChannelAnnouncement(scid: ShortChannelId): Promise<void> {
        const msg = await this.findChannelAnnouncement(scid);
        Iif (!msg) return;
        const chanKey = getChanKey(scid);
 
        // delete channel
        this._channelAnn.delete(getChanKey(scid));
 
        // delete outpoint link
        if (msg instanceof ExtendedChannelAnnouncementMessage) {
            this._channelByOutPoint.delete(msg.outpoint.toString());
        }
 
        // delete channel_updates
        this._channelUpd.delete(getChanUpdKey(scid.toNumber(), 0));
        this._channelUpd.delete(getChanUpdKey(scid.toNumber(), 0));
 
        // delete node links
        this._nodeChannels.get(getNodeKey(msg.nodeId1)).delete(chanKey);
        this._nodeChannels.get(getNodeKey(msg.nodeId2)).delete(chanKey);
    }
 
    public async deleteChannelUpdate(scid: ShortChannelId, dir: number) {
        this._channelUpd.delete(getChanUpdKey(getChanKey(scid), dir));
    }
 
    public async deleteNodeAnnouncement(nodeId: Buffer) {
        this._nodeAnn.delete(getNodeKey(nodeId));
        this._nodeChannels.get(getNodeKey(nodeId));
    }
 
    private async _saveNodeChannel(nodeId: Buffer, chanKey: bigint) {
        const nodeKey = getNodeKey(nodeId);
        this._nodeChannels.set(nodeKey, this._nodeChannels.get(nodeKey) || new Set<bigint>());
        this._nodeChannels.get(nodeKey).add(chanKey);
    }
}
 
function getNodeKey(nodeId: Buffer): string {
    return nodeId.toString("hex");
}
 
function getChanKey(scid: ShortChannelId): bigint {
    return scid.toNumber();
}
 
function getChanUpdKey(chanKey: bigint, direction: number): bigint {
    return (chanKey << BigInt(8)) | BigInt(direction);
}