code review

Author: sergiyvamz

wrapper/src/main/java/software/amazon/jdbc/ConnectionPluginManager.java

@@ -278,7 +278,7 @@ protected <T, E extends Exception> PluginChainJdbcCallable<T, E> makePluginChain
       if (isSubscribed) {
         if (pluginChainFunc == null) {
           // This case is for DefaultConnectionPlugin that always terminates the list of plugins.
-          // Default plugin couldn't be skipped.
+          // Default plugin can't be skipped.
           pluginChainFunc = (pipelineFunc, jdbcFunc, pluginToSkip) ->
               executeWithTelemetry(() -> pipelineFunc.call(plugin, jdbcFunc), pluginName);
         } else {

wrapper/src/main/java/software/amazon/jdbc/plugin/federatedauth/FederatedAuthPlugin.java

@@ -30,6 +30,7 @@
 import org.checkerframework.checker.nullness.qual.NonNull;
 import software.amazon.awssdk.auth.credentials.AwsCredentialsProvider;
 import software.amazon.awssdk.regions.Region;
+import software.amazon.jdbc.AuthenticationConnectionPlugin;
 import software.amazon.jdbc.AwsWrapperProperty;
 import software.amazon.jdbc.HostSpec;
 import software.amazon.jdbc.JdbcCallable;
@@ -47,7 +48,7 @@
 import software.amazon.jdbc.util.telemetry.TelemetryFactory;
 import software.amazon.jdbc.util.telemetry.TelemetryGauge;
 
-public class FederatedAuthPlugin extends AbstractConnectionPlugin {
+public class FederatedAuthPlugin extends AbstractConnectionPlugin implements AuthenticationConnectionPlugin {
 
   static final ConcurrentHashMap<String, TokenInfo> tokenCache = new ConcurrentHashMap<>();
   private final CredentialsProviderFactory credentialsProviderFactory;