[JS] 콘솔로 값 입력 받기, readline 모듈

(동기) node.js 배운 거를 좀 연습하는 목적에서 백준 문제를 풀어보려고 했다. 그런데 console 출력만 할 줄 알지 console 입력에 대해서는 별도로 배운게 없다는 사실을 깨달아 버렸다!! ㅋㅋㅋㅋㅋ

 

그래서 이번 기회에 console 입력에 대해서 정리해본다.

인터넷에 몇가지 방법이 제시되어 있던데, 내 기준으로 그나마 좀 쉬워보이는 방법을 제시해보려고 한다.

 

---

우리는 JS의 내장 모듈인 'readline' 모듈을 사용할 것이다. 

처음에는 아래의 가장 기본이 되는 코드를 작성한다.

const readline = require('readline');

// 인터페이스 객체를 만들자.
const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
});

이후에는 상황에 따라서 코드의 내용이나 디테일이 달라진다.

그래서 본인이 어떤 입력값을 받는지에 따라서 코드의 내용이 달라질 수 있다.

---

• 정수 하나 입력 받기 (1회)
• 무한히 입력 받기
• 정수 두 개 입력 받기 (1회)
• 배열의 크기 N을 입력 받고, N개 정수 입력 받기

여러분이 처음 부분을 뛰어넘을 수도 있으니 미리 말해두자면, 'readline' 모듈의 'rl' 객체는 event-driven 방식으로 동작한다. event-driven에 대해서는 별도로 설정하지는 않겠다.

 

"line"은 한 줄이 입력되는 이벤트이고, "close"는 EOF 또는 직접 close() 함수를 호출 했을 때 발생하는 이벤트이다. 

'readline' 모듈을 잘 쓰려면, 어떤 이벤트들이 있는지 잘 알고 있는 것이 좋겠지만, 가장 기본이 되는 것은 "line"와 "close"이다. 

---

1. 정수 하나 입력 받기(1회)

let N; // 정수를 저장할 변수
rl.on("line", function(line){
    console.log(`print input string ${line}`);
    N = Number(line);
    console.log(`print input number ${N}`);
    rl.close();
})

말 그대로 정수 하나를 입력 받아서 그대로 출력하는 코드다.

 

마지막에 rl.close()가 있다는 점에 주목하자. 이 한줄이 입력을 무한히 받는지 아닌지를 결정한다.

 

2. 무한히 입력 받기

let N; // 정수를 저장할 변수
rl.on("line", function(line){
    console.log(`print input string ${line}`);
    N = Number(line);
    console.log(`print input number ${N}`);
})

정수 하나를 입력하고 엔터를 쳤더라도 프로그램이 계속해서 입력을 기다린다.(또는 입력을 받을 수 있도록 유지한다.)

 

close() 함수는 rl 객체의 "close" 이벤트를 주는 만큼 아주 중요한 함수이다.

 

3. 정수 2개 입력받기(1회)

let input = []
rl.on("line", function(line){
    input = line.split(' ').map((el) => parseInt(el));
    rl.close();
})

rl.on("close", function(){
    let A = input[0];
    let B = input[1];
    console.log(A+B);
})

백준 1000번 A+B를 해결할 수 있는 코드를 살펴보자.

 

두 정수가 공백을 기준으로 입력되기 때문에, split(' ') 부분을 넣어줬다.

그리고 입력된 정수를 바로 Int 형으로 변환하기 위해서 parseInt(el) 부분을 넣어줬다.

map(...)은 리스트 자료형인 line.split(' ')의 각 원소에 콜백 함수를 적용해주는 역할을 한다.

 

이후의 기능은 모두 rl.on("close", ....)에서 처리가 된다. 

참고로 본인은 처음에 아래의 코드를 구성하였다.

let input = []
rl.on("line", function(line){
    input = line.split(' ').map((el) => parseInt(el));
    rl.close();
})


let A = input[0];
let B = input[1];
console.log(A+B);

 

하지만, 문제는 이렇게 코드를 구성하면, A, B에 적절한 값이 들어가지 않기 때문에 NaN을 출력하게 된다.

 

비단 console.log(A+B) 말고, 그냥 console.log("test")로 출력을 해봐도, rl.on("line", ...) 이전에 콘솔 출력이 되는 것을 알 수 있다.

 

이런 일이 일어나는 이유는 글의 앞에서 제시하였듯이 rl 객체가 event-driven 방식으로 작동하기 때문이다!!

 

그래서 우리가 입력을 받고 rl.close()를 호출하여 입력이 끝났다는 것을 알려줘야 한다.

 

---

4. 배열의 크기 N을 입력 받고, N개 정수 입력 받기

백준 10818번 최소, 최대 문제의 입력을 node.js로 어떻게 받을 수 있는지 살펴보자.

 

먼저 입력은 아래와 같다.

5
20 10 35 30 7

 

자, 이제 node.js 코드를 살펴보자.

let N;

let count = 0; // 몇번째 라인을 입력 받는지를 저장하는 변수
rl.on("line", function(line){
    if(count == 0)
        N = Number(line.trim());
    else if(count == 1){
        get_Min_Max(line.trim());
        rl.close();
    }
    count++;
})

function get_Min_Max(input){
    let arr = input.split(' ').map((num) => parseInt(num));
    console.log(arr); // 테스트용

    // min, max를 출력은 스스로 해보시기 바랍니다.
}

간단한 트릭을 이용해서 한줄 한줄 입력될 때마다 다른 작업을 수행하도록 코드를 구성하였습니다.

 

우리가 매번 입력을 받을 때, 우리는 항상 단 하나의 rl.on("line", ...)를 가지고, 그곳에 기록된 내용을 수행할 수 밖에 없습니다. 즉, 아래와 같은 코드는 우리의 목적에 부합하지 않는다는 거죠.

/* 잘못된 사용예 1 */
rl.on("line", function(line){
    N = Number(line.trim());
})

rl.on("line", function(line){
    get_Min_Max(line.trim());
    rl.close();
})

 

/* 잘못된 사용예 2 */
rl.on("line", function(line){
    N = Number(line.trim());
    rl.on("line", function(line){
        get_Min_Max(line.trim());
        rl.close();
    })
})

그래서 저는 아예 현재 몇번째 줄의 입력을 받는지를 기록하여 n-th 라인의 입력 때마다 다른 작업을 수행하도록 코드를 구성하였습니다.

 

이분의 블로그 글을 보면, 저와 비슷한 방식으로 rl.on("line", ...)를 사용하고 계십니다.

다만, 다른 점은 이분은 rl.on("line", ...)의 입력을 모두 받은 후에, rl.on("close", ...) 함수에서 작업을 처리한다는 거죠.

 

어느 방법이 더 좋은지는 저도 잘 모르겠습니다. (저도 아직 node.js PS를 많이 안 해봐서요)

 

그래도 C/C++, python 문법에 익숙한 저는 입력을 모두 받고, 작업을 처리하는 것보다 입력을 한줄한줄 받을 때 처리하는 것이 더 익숙하기 때문에 위와 같은 방식으로 코드를 구성해서 PS를 하고 있습니다.

 

---

혹시 더 좋은 코드가 있다면, 제시해주세요!! 

함께 배웁시다 ㅎㅎ