Fix missing calls to onCallbackFinished

Adjust code to remove error-prone early return + some refactoring.

Bug: 383556228
Flag: android.chre.flags.offload_implementation
Test: Confirm desired wakelock behavior through dumpsys power
Change-Id: I2b6e2c466cfcb5e85d221ad02aa79cf3ed904e4e