Refactor transient state in Java client (#24219)

* Refactor transient state in Java client

* some fb

* fix nullref

* fixup rebase

* fb

* apply some fb

* fix flaky

* lock

* check previous state

Author: BrennanConroy