【javascript】filterMap()をジェネレータ関数を使って作ってみた

#前置き

//これを
array.filter().map()
//こうしたかった
array.filterMap()

ざっくり言いますと
filter()やmap()のcallback関数をジェネレータ関数で処理するようにしただけです。
引数は同様。
最初のyieldでfilter()、2回目のyieldでmap()。

「こんなのがあったらいいな」というものなので、実用に耐え得るか否かはわかりません。

#function

const filterMap = (arr, generator) => {
	try {
		if (!Array.isArray(arr)) {
			throw new Error('1st arg : not Array');
		}
		if (typeof generator!=='function'
			 || generator.constructor.name!=='GeneratorFunction') {
			throw new Error('2nd arg : not GeneratorFunction');
		}
		const array = arr.slice();
		const result = [];
		for (let i=0,m=array.length; i<m; i++) {
			const gen = generator(array[i], i, array);
			if ( gen.next().value ) {
				result.push( gen.next().value );
			}
			gen.return();
		}
		return result;
	} catch(e) {
		console.error(e);
	}
};

const arr = [0,1,7,2,1,3,4,5,9,4,5,0,8,3,5,7,4,4];

//値が5のindexを出力
const idx5 = filterMap(arr, function*(x,i,a) {
	yield x===5;	//filter
	yield i			//map
});
console.log( idx5 );	//  [7, 10, 14]


const chunk = filterMap([...Array(10).keys()], function*(x,i,a) {
	yield i%3===0;			//filter
	yield a.slice(i, i+3);	//map
});
console.log( chunk );	//[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]

#Array.prototype

if (!Array.prototype.filterMap) {
	Object.defineProperty(Array.prototype, 'filterMap', {
		value: function(generator) {
			try {
				if (typeof generator!=='function'
					 || generator.constructor.name!=='GeneratorFunction') {
					throw new Error('arg is not a GeneratorFunction');
				}
				const array = this.slice();
				const result = [];
				for (let i=0,m=array.length; i<m; i++) {
					const gen = generator(array[i], i, array);
					if ( gen.next().value ) {
						result.push( gen.next().value );
					}
					gen.return();
				}
				return result;
			} catch(e) {
				console.error(e);
			}
		},
	});
}

const arr = [0,1,7,2,1,3,4,5,9,4,5,0,8,3,5,7,4,4];

//値が5のindexを出力
const idx5 = arr.filterMap(function*(x,i,a) {
	yield x===5;	//filter
	yield i			//map
});
console.log( idx5 );	//  [7, 10, 14]


const chunk = [...Array(10).keys()].filterMap(function*(x,i,a) {
	yield i%3===0;			//filter
	yield a.slice(i, i+3);	//map
});
console.log( chunk );	//[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]

#補足

const idx5 = arr.filterMap(function*(x,i,a) {
	yield x===5;	//filter
	yield i			//map
});

処理が単純であればyield* [filter, map]でまとめられます。

const idx5 = arr.filterMap(function*(x,i,a) {
	yield* [x===5, i];		// [filter, map]
});

#私感・妄想
ジェネレータ関数にもアロー関数のような略した書き方がほしい。

妄想.js

//これを
function*(x,i) {
    yield* [x===5, i];
});

//こうして
(x,i) *> {
	yield* [x===5, i]
};

//更にこうして
(x,i) **> [x===5, i];

//こうなる
const idx5 = arr.filterMap((x,i) **> [x===5, i]);

妄想2.js

//こういうのも
function* increase(n) {
    while (true) yield n++;
});

//こう…欲張りすぎか
const increase = n **> n++;

なんてね

#参考