- public final Class<?> getIdentityClass(final QName input) {
- return identityClasses.getUnchecked(input);
+ public final Class<? extends BaseIdentity> getIdentityClass(final QName input) {
+ try {
+ return identityClasses.get(requireNonNull(input));
+ } catch (ExecutionException e) {
+ Throwables.throwIfUnchecked(e.getCause());
+ throw new IllegalStateException("Unexpected error looking up " + input, e);
+ }