Real-time water quality control