Is there any kind of callback or event the lambda function can receive just before the runtime is frozen/shutdown?

I am running java based lambda function and using java, it is triggered by apigw and it uses AmazonSQSAsync client for sending msg to sqs asynchronously using sendMessageAsync() call. The concern is that, after the handleRequest() method returns the response, will the jvm / runtime environment immediately be frozen for later reuse? if yes, what happens to msg in flight as the call to sqs is async? Is there any kind of callback or event the lambda function can receive just before the runtime is frozen/shutdown? I don’t think so based on the lambda docs but wanted to check if anyone knows of any workaround for that.