인프런 커뮤니티 질문&답변

안녕하세요 선생님~~

지금까지 모르는거 있어도 그냥 계속 반복해서 보고, log 찍어보면서 해결했는데

flatMap 부분에서 도저히 이해 안 가는 부분이 있어서 이렇게 질문드립니다 ㅠㅠ.. 

우선 왜 map을 먼저하는지 모르겠습니다..

중첩배열을 먼저 평탄화 시켜주고 그 다음 map을 했을때에는 예상치 못한 결과나 에러가 발생하지 않았는데요.

map을 먼저 하게 되면 중첩 배열을 받아서 전달된 함수를 실행하는 부분에서 계속 결과값이 이상해 집니다... ㅠㅠ

사실 당연한 결과인데 map이든 L.map이든 [1,2,[3,4],[5]] 이런 중첩배열을 받아서 순회를 하고

a => a+10 이런 함수를 실행하면 당연히 결과 값이 이상해지는데 선생님 코드에서는 어떻게 제대로 작동 되는 건지 궁금합니다...;;

만약 L.flatMap(a => a, [1, 2, [3, 4], [5]]) 이런식으로 하면 그냥 a를 반환만 하니까 중첩배열이 반환 되고 L.flatten 에서 평탄화 되니까 별문제 없는데 a =>a +10 이런식의 함수를 넘겨주면 결과값이 이상해집니다...ㅠ 

뭐가 문제 인지 힌트 부탁드립니다~ 

const curry = fn => (a, ...args) => args.length ? fn(a, ...args) : (...args) => fn(a, ...args);





const reduce = (fn, iter, acc) => {
    if (!acc) {
        iter = iter[Symbol.iterator]();
        acc = iter.next().value;
    }

    for (const a of iter) {
        acc = fn(acc, a);
    }
    return acc;
}







const L = {};


L.map = curry(function* (fn, iter) {
    for (const a of iter) {
        yield fn(a)
    }
});



const isIterable = (a) => a && a[Symbol.iterator]

L.flatten = function* (iter) {

    for (const a of iter) {
        if (isIterable(a)) {
            yield* a
        } else yield a
    }
}




const map = curry((fn, iter) => {
    const res = [];
    for (const a of iter)
        res.push(fn(a))
    return res;
})






const take = curry((limit, iter) => {
    const res = [];
    for (const a of iter) {
        res.push(a)
        if (res.length === limit) return res;
    }
    return res;
});





const go = (...args) => reduce((a, fn) => fn(a), args);

const pipe = (f, ...fn) => (...args) => go(f(...args), ...fn);



// L.flatMap

L.flatMap = curry(pipe(
    L.map,
    L.flatten
));

const iter = L.flatMap(a => a + 10, [1, 2, [3, 4], [5]])

console.log(...iter) // 결과값 11 12 3 , 4 1 0 5 1 0