Wi-Fi disconnects I go to lockscreen

Yes, that's correct. Unless something is actively happening (like things being downloaded), WiFi disconnects when the phone locks.

It saves power. And it's been that way probably since the first iPhone. But, you should always buy the tool that best meets your needs.