"use strict"; import { once, EventEmitter } from 'events'; import log from '../../lib/io/NamespacedLog.mjs'; const l = log("systemquery"); import Agent from './agent/Agent.mjs'; import InfoBroker from './core/InfoBroker.mjs'; import ItemQueue from './async/ItemQueue.mjs'; class SystemQuery extends EventEmitter { constructor(config, mode = "agent") { super(); // The operating mode. Possible values: agent [default], query_client // TODO: Is this the best way of doing this? Maybe we should have a separate class for this? I'm not sure. this.mode = mode; this.config = config; this.info = new InfoBroker(); } async init() { /// // 1: Create agent /// this.agent = new Agent(this.config); await this.agent.init(); /// // 2: Attach listeners /// this.agent.on("message-query", this.handle_query.bind(this)); this.agent.on("message-query-response", this.handle_query_response.bind(this)); } async handle_query(peer, msg) { // 1: Validate input if(typeof msg.name !== "string" || !this.info.is_valid_table(msg.name)) return; // 2: Fetch system info let table = await this.info.fetch_table(msg.name); if(table === null) return; // 3: Return to requester await peer.send("query-response", { name: msg.name, table }); } async handle_query_response(peer, msg) { l.log(`query-response from ${peer.id_short}`, msg); } async capture_query_response(event_name, ac) { return await once(this.agent, `message-${event_name}`, { signal: ac }); } *fetch_table(name) { // If it isn't valid for us, it ain't gonna be valid for anyone else.... if(!this.info.is_valid_table(name)) return null; const queue = new ItemQueue(); const handle_response = (peer, msg) => { // TODO: Validate response. Is it the right table? Is it even a table? Note that multiple fetch_table calls may be running in parallel, so we should not make too much of a fuss if we get the wrong table by accident. // TODO: It would be seriously cool to have fetch_table() be an async generator that yields pairs of peer ids and tables as they come in. if(typeof msg !== "object" || typeof msg.table !== "object" || typeof msg.name !== "string") { l.debug(`Discarding invalid table from peer ${peer.id_short}`); return; } // If it's not the right table, ignore it if(msg.name !== name) return; queue.push({ peer, table: msg.table }); }; this.on("message-query-response", handle_response); // Only *after* we have our listeners in place do we then broadcast the // query. Note that despite not having entered the below while loop yet, // we do not drop any messages due to the use of the ItemQueue. this.agent.broadcast(`query`, { name }); while(peers_seen.length < this.agent.connected_peers.length) { let next = queue.pop(this.config.net.table_config); if(typeof next === "undefined") // We timed out break; yield next; } this.off("message-query-response", handle_response); // FUTURE: Add a cache here? Note that we also do not listen for query responses unless we've asked for a table. } } SystemQuery.Create = async function(config) { let result = new SystemQuery(config); await result.init(); return result; } export default SystemQuery;