Water management in energy industry