Justice Department to investigate US prosecutor’s sudden departure from Atlanta following Trump criticism