Add fault tolerance for dataflow workers