js-wrench/dist/modules/remove.js

JS函数库

import isFn from './isFn';
import isNumber from './isNumber';
import isArray from './isArray';
/**
 * @description 删除数组元素
 *
 * @param {Array<any>} arr  需要操作的数组
 * @param {(Function|number)} iteratee 回调
 * @return {*}  {Array<any>} 返回新的数组 注意并不是原数组的引用
 * @example remove([1,2,3], (v, i, arr) => v === 1) => [2,3];
 */
const remove = (arr, iteratee) => {
    if (!isArray(arr))
        return [];
    const result = [];
    const indexs = []; //存放索引, 因为splice后 length会有变化
    let index = 0;
    const length = arr.length;
    while (index < length) {
        let value = arr[index];
        if (isNumber(iteratee)) {
            index === iteratee ? indexs.push(index) : result.push(value);
        }
        else if (isFn(iteratee)) {
            if (iteratee(value, index, arr)) {
                indexs.push(index);
            }
            else {
                result.push(value);
            }
        }
        index++;
    }
    indexs.forEach(i => arr.splice(i, 1));
    return result;
};
export default remove;