Simply press the power button to lock immediately, quite easy and available presently as a feature now. Your screen doesn’t need to auto-lock so quickly on your important calls.
You do have a valid requirement for an improvement, submit your change request to the support team.