Have you read "The Water Knife" by Paolo Bacigalupi? Mostly set in Phoenix when the water has run out.

https://www.goodreads.com/book/show/23209924-the-water-knife