Add keyguardGoingAway to the state controller.

There was previously no method to get callbacks about
goingAway, and it was also inconsistently updated,
which is not ideal for a StateController.

Bug: 229890190
Test: atest SystemUITests
Change-Id: I07257e97f9699ffd522806ab72c7952ed5876b1a