Is my iPhone XR okay in water

They are water resistant and not waterproof. This is so that it is protected against incidental water getting on it. Water damage is not covered by the warranty and so you should not intentionally be getting it wet and definitely not submerging it.

It could severely damage the device and is very expensive to fix.

No. It isn't waterproof.