I just noticed your code snippet won't work if the layer has over 2000 features (you must loop and paginate).