This puzzle is Example 40 from Capablanca's "Chess Fundamentals" book.
There is a complete training database for the Queen and Rook endgame. It's on eBay.