Tengo el siguiente código:

 function* getPossibleCollidersWith(shape) { for (let square of retrieve(shape)) { for (let neighbor of square) { yield neighbor; } } }

Esto funciona, pero usa generadores y rendimiento, que es muy lento. Lo he comparado con el generador de perfiles de Chrome y tiene un rendimiento significativamente menor que eliminar el generador y simplemente iterar a través del bucle anidado en el código del cliente.

Sin embargo, es muy repetitivo hacer un bucle anidado cada vez que quiero obtener los vecinos de una forma. Me gustaría crear un iterador personalizado para ver si tiene un mejor rendimiento que la función del generador, pero todos los ejemplos que veo son bastante complejos e implican realizar un seguimiento de los índices, los valores nulos, etc.

¿Existe una forma sencilla de convertir el código anterior en un iterador personalizado? Idealmente, me gustaría seguir usando bucles for of para devolver cada elemento, pero todos los ejemplos que veo en línea implican un seguimiento de índice propenso a errores.

Me gustaría que el código del cliente se viera así, si es posible:

 for (let neighbor of getPossibleCollidersWith(shape)) { // handle neighbor }