ipfs-bitswap/dist/src/wantlist/index.js

JavaScript implementation of the Bitswap data exchange protocol used by IPFS

import { trackedMap } from '@libp2p/utils/tracked-map';
import { base58btc } from 'multiformats/bases/base58';
import { Message } from '../message/message.js';
import { WantListEntry as Entry } from './entry.js';
const WantType = {
    Block: Message.Wantlist.WantType.Block,
    Have: Message.Wantlist.WantType.Have
};
const sortBy = (fn, list) => {
    return Array.prototype.slice.call(list, 0).sort((a, b) => {
        const aa = fn(a);
        const bb = fn(b);
        return aa < bb ? -1 : aa > bb ? 1 : 0;
    });
};
export class Wantlist {
    static Entry = Entry;
    set;
    _stats;
    constructor(stats, libp2p) {
        this.set = (libp2p != null)
            ? trackedMap({
                name: 'ipfs_bitswap_wantlist',
                metrics: libp2p.metrics
            })
            : new Map();
        this._stats = stats;
    }
    get length() {
        return this.set.size;
    }
    add(cid, priority, wantType) {
        const cidStr = cid.toString(base58btc);
        const entry = this.set.get(cidStr);
        if (entry != null) {
            entry.inc();
            entry.priority = priority;
            // We can only overwrite want-have with want-block
            if (entry.wantType === WantType.Have && wantType === WantType.Block) {
                entry.wantType = wantType;
            }
        }
        else {
            this.set.set(cidStr, new Entry(cid, priority, wantType));
            if (this._stats != null) {
                this._stats.push(undefined, 'wantListSize', 1);
            }
        }
    }
    remove(cid) {
        const cidStr = cid.toString(base58btc);
        const entry = this.set.get(cidStr);
        if (entry == null) {
            return;
        }
        entry.dec();
        // only delete when no refs are held
        if (entry.hasRefs()) {
            return;
        }
        this.set.delete(cidStr);
        if (this._stats != null) {
            this._stats.push(undefined, 'wantListSize', -1);
        }
    }
    removeForce(cidStr) {
        if (this.set.has(cidStr)) {
            this.set.delete(cidStr);
        }
    }
    forEach(fn) {
        this.set.forEach(fn);
    }
    entries() {
        return this.set.entries();
    }
    sortedEntries() {
        // TODO: Figure out if this is an actual bug.
        // @ts-expect-error - Property 'key' does not exist on type 'WantListEntry'
        return new Map(sortBy(o => o[1].key, Array.from(this.set.entries())));
    }
    contains(cid) {
        const cidStr = cid.toString(base58btc);
        return this.set.has(cidStr);
    }
    get(cid) {
        const cidStr = cid.toString(base58btc);
        return this.set.get(cidStr);
    }
}
//# sourceMappingURL=index.js.map