javascript groupby 억지구현

정의

Array.prototype.groupBy = function(mapper = util.mandatory("mapper")) {
    let keySet = new Set(this.map(mapper));
    let groupObj = {};
    
    for(const keyItem of keySet) {
        groupObj[keyItem] = this.filter(item => mapper(item) == keyItem);
    }
    
    return groupObj;
};

IE에서는 사용할 수 없도록 mandatory를 사용한 소스코드. IE는 생각조차도 하지 말것.

 

 

사용

const arr = [{
    "color": "red", 
    "gender": "female"
}, {
    "color": "red", 
    "gender": "female"
}, {
    "color": "blue", 
    "gender": "male"
}, {
    "color": "blue", 
    "gender": "male"
}];

arr.groupBy(item => item["color"]);
/*
{
    "red": [
        {
            "color": "red",
            "gender": "female"
        },
        {
            "color": "red",
            "gender": "female"
        }
    ],
    "blue": [
        {
            "color": "blue",
            "gender": "male"
        },
        {
            "color": "blue",
            "gender": "male"
        }
    ]
}
*/

arr.groupBy(item => item["gender"]);
/*
{
    "female": [
        {
            "color": "red",
            "gender": "female"
        },
        {
            "color": "red",
            "gender": "female"
        }
    ],
    "male": [
        {
            "color": "blue",
            "gender": "male"
        },
        {
            "color": "blue",
            "gender": "male"
        }
    ]
}
*/