readline trong Node.js: đọc dữ liệu nhập từ terminal

Để đọc dữ liệu người dùng nhập trong Node.js, hãy tạo một interface readline trên process.stdin và đặt câu hỏi bằng rl.question(). Node.js không có prompt(): phiên bản của trình duyệt mở một hộp thoại, còn chương trình terminal thì không có hộp thoại nào. Cách viết hiện đại dùng readline/promises, trong đó await rl.question('Name? ') trả về dòng mà người dùng đã gõ. Hãy đóng interface khi xong việc, nếu không chương trình sẽ cứ chờ thêm dữ liệu.

// input: 36
import * as readline from 'node:readline/promises';
import { stdin as input, stdout as output } from 'node:process';

const rl = readline.createInterface({ input, output });

const age = await rl.question('How old are you? ');
rl.close();

console.log(`Next year you will be ${Number(age) + 1}.`);

question() in ra lời nhắc, chờ một dòng dữ liệu rồi trả về dòng đó (đã bỏ ký tự xuống dòng), nên chương trình in ra Next year you will be 37. Với nhiều câu hỏi, hãy lần lượt await từng question().

question() của readline với callback

API gốc, có trong mọi phiên bản Node.js, truyền câu trả lời cho một callback. Câu hỏi lồng nhau thì callback cũng lồng nhau:

// input: blue
const readline = require('readline');

const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout,
});

rl.question('Favorite color? ', (answer) => {
    console.log(`You picked ${answer.trim().toLowerCase()}.`);
    rl.close();
});

Chương trình in ra lời nhắc, rồi You picked blue. Câu trả lời là một chuỗi bình thường, nên hãy trim trước khi so sánh: nếu người dùng gõ thêm một dấu cách sau từ đó, bạn sẽ nhận được 'blue ', không bằng 'blue'.

Đọc dữ liệu từng dòng

Để xử lý từng dòng cho đến khi hết dữ liệu, hãy duyệt interface bằng for await. Đây là cách một chương trình đọc danh sách người dùng gõ vào (kết thúc bằng Ctrl+D) hoặc một tệp được pipe vào bằng node index.js < numbers.txt. Code cũ làm việc tương tự bằng sự kiện, rl.on('line', fn) cho mỗi dòng và rl.on('close', fn) ở cuối; vòng lặp chính là cặp sự kiện đó được viết từ trên xuống dưới:

// input: 12
// input: 7
// input: 30
const readline = require('readline');

async function main() {
    const rl = readline.createInterface({ input: process.stdin });
    let total = 0;
    let count = 0;

    for await (const line of rl) {
        total += Number(line);
        count += 1;
    }

    console.log(`${count} numbers, total ${total}`);
}

main();

Kết quả in ra là 3 numbers, total 49, và vòng lặp tự kết thúc khi hết dữ liệu, nên không cần close(). Thêm dòng dữ liệu thứ tư là 1 thì kết quả thành 4 numbers, total 50.

Đọc toàn bộ stdin một lần

Với dữ liệu được pipe từ một chương trình khác (cat data.txt | node index.js), đọc tất cả trong một lần gọi sẽ đơn giản hơn. File descriptor 0 là đầu vào chuẩn (standard input). Một tệp được lưu trên Windows có thể kết thúc mỗi dòng bằng \r\n, nên hãy tách theo /\r?\n/ khi dữ liệu có thể đến từ đó:

// input: apples 3
// input: bread 1
// input: cheese 2
const fs = require('fs');

const text = fs.readFileSync(0, 'utf8');
const lines = text.trim().split(/\r?\n/);

for (const line of lines) {
    const [item, qty] = line.split(' ');
    console.log(item.padEnd(8), 'x', qty);
}

Chương trình in ba mặt hàng thành một cột. readFileSync(0) chặn cho đến khi nhận đủ dữ liệu, điều này ổn với script đọc từ pipe và với việc kiểm thử, nhưng không phù hợp cho công cụ tương tác, vì người dùng sẽ phải gõ hết mọi thứ trước khi thấy bất kỳ lời nhắc nào.

Nên dùng cách nào

Cả bốn cách đều đọc cùng một đầu vào chuẩn. Khác biệt nằm ở thời điểm code của bạn nhận được văn bản: từng câu trả lời, từng dòng, hay tất cả ở cuối.

Lỗi thường gặp

Quên rl.close(). Interface giữ đầu vào chuẩn luôn mở, nên chương trình không bao giờ thoát sau câu hỏi cuối cùng. Hãy đóng nó khi đã có đủ câu trả lời.

Dùng câu trả lời như một số. Giống mọi thứ được gõ trong terminal, câu trả lời là chuỗi. '36' + 1 là '361'; hãy chuyển đổi bằng Number() trước.

Pipe nhiều câu trả lời vào nhiều câu hỏi. Khi dữ liệu đến từ pipe thay vì bàn phím, readline đọc tất cả cùng lúc, và dòng nào đến trước khi question() tương ứng được hỏi sẽ bị bỏ mất. Với dữ liệu từ pipe, hãy dùng vòng lặp for await hoặc readFileSync(0).

Bỏ sót tùy chọn output. Không có output: process.stdout, question() sẽ không bao giờ in ra lời nhắc, và người dùng chỉ nhìn chằm chằm vào một terminal trống. Vòng lặp đọc từng dòng thì không cần tùy chọn này, vì nó không in gì cả.