Hey, it is likely a side effect of Battery Optimisation, that you are experiencing.
If you follow: Settings > Battery and then uncheck “Sleep standby optimisation”, this should fix the issues you see with notifications not being pushed to your device when it is locked and not being used.
Any further issues, it may be worth sending Feedback directly to the team, either via Settings > System > Feedback or Settings > Tips & Feedback > Feedback.