Both AuthenticationContext and SecurityIdentity can be captured and restored in a similar manner; they should use the same gamut of run*() methods.

The methods from org.wildfly.common.function.Functions can help with this.