SPARK-2277: make TaskScheduler track hosts on rack