Optimizing water management in hydraulic fracturing