Respect max rowgroup size in Arrow writer (#381) (#430)

* Respect max rowgroup size in Arrow writer

* simplify while loop

* address review feedback

Co-authored-by: Wakahisa <nevilledips@gmail.com>

Author: alamb