LotteryAutoScript/lib/helper/d_storage.js

const { log, readDyidFile, writeDyidFile } = require('../utils');

const d_storage = {
    /**
     * 搜索dyid
     * @param {string} dyid
     * @returns {Promise<boolean>}
     */
    searchDyid: async (dyid) => {
        let buffer = '';
        let found = false;

        const stream = readDyidFile(Number(process.env.NUMBER));

        for await (const chunk of stream) {
            buffer += chunk;

            while ((buffer.indexOf(dyid)) !== -1) {
                found = true;
                stream.destroy();
                return found;
            }

            const lastCommaIdx = buffer.lastIndexOf(',');
            if (lastCommaIdx !== -1) {
                buffer = buffer.slice(lastCommaIdx);
            }
        }

        if (!found && buffer.length > 0) {
            if (buffer.includes(dyid)) {
                found = true;
            }
        }

        stream.destroy();
        return found;
    },
    /**
     * 更新dyid
     * @param {string} dyid
     */
    updateDyid: (dyid) => {
        log.info('更新dyid', `写入${dyid}`);
        return new Promise((resolve) => {
            const ws = writeDyidFile(Number(process.env.NUMBER));
            ws.write(dyid + ',', () => {
                ws.destroy();
                resolve();
            });
            ws.on('error', err => {
                log.error('更新dyid', err);
                resolve();
            });
        });
    }
};


module.exports = d_storage;