Water companies have been warned they may have to implement measures including restrictions in the months ahead