fix wrong calls in wrapper class (#1413)

Author: sergiyvamz

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

@@ -36,7 +36,6 @@
 import software.amazon.jdbc.util.Messages;
 import software.amazon.jdbc.util.PropertyUtils;
 import software.amazon.jdbc.util.SlidingExpirationCache;
-import software.amazon.jdbc.wrapper.HighestWeightHostSelector;
 
 public class C3P0PooledConnectionProvider implements PooledConnectionProvider, CanReleaseResources {
 

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

@@ -32,14 +32,12 @@
 import org.checkerframework.checker.nullness.qual.Nullable;
 import software.amazon.jdbc.dialect.Dialect;
 import software.amazon.jdbc.exceptions.SQLLoginException;
-import software.amazon.jdbc.targetdriverdialect.ConnectInfo;
 import software.amazon.jdbc.targetdriverdialect.TargetDriverDialect;
 import software.amazon.jdbc.util.Messages;
 import software.amazon.jdbc.util.PropertyUtils;
 import software.amazon.jdbc.util.RdsUtils;
 import software.amazon.jdbc.util.SqlState;
 import software.amazon.jdbc.util.WrapperUtils;
-import software.amazon.jdbc.wrapper.HighestWeightHostSelector;
 
 /**
  * This class is a basic implementation of {@link ConnectionProvider} interface. It creates and

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

@@ -35,7 +35,6 @@
 import software.amazon.jdbc.util.Messages;
 import software.amazon.jdbc.util.PropertyUtils;
 import software.amazon.jdbc.util.RdsUtils;
-import software.amazon.jdbc.wrapper.HighestWeightHostSelector;
 
 /**
  * This class is a basic implementation of {@link ConnectionProvider} interface. It creates and

wrapper/src/main/java/software/amazon/jdbc/wrapper/HighestWeightHostSelector.java renamed to wrapper/src/main/java/software/amazon/jdbc/HighestWeightHostSelector.java

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-package software.amazon.jdbc.wrapper;
+package software.amazon.jdbc;
 
 import java.sql.SQLException;
 import java.util.Comparator;
@@ -23,9 +23,6 @@
 import java.util.stream.Collectors;
 import org.checkerframework.checker.nullness.qual.NonNull;
 import org.checkerframework.checker.nullness.qual.Nullable;
-import software.amazon.jdbc.HostRole;
-import software.amazon.jdbc.HostSelector;
-import software.amazon.jdbc.HostSpec;
 import software.amazon.jdbc.hostavailability.HostAvailability;
 import software.amazon.jdbc.util.Messages;
 

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

@@ -44,7 +44,6 @@
 import software.amazon.jdbc.util.RdsUrlType;
 import software.amazon.jdbc.util.RdsUtils;
 import software.amazon.jdbc.util.SlidingExpirationCache;
-import software.amazon.jdbc.wrapper.HighestWeightHostSelector;
 
 public class HikariPooledConnectionProvider implements PooledConnectionProvider,
     CanReleaseResources {

wrapper/src/main/java/software/amazon/jdbc/plugin/limitless/LimitlessRouterServiceImpl.java

@@ -26,6 +26,7 @@
 import java.util.logging.Logger;
 import org.checkerframework.checker.nullness.qual.NonNull;
 import software.amazon.jdbc.AwsWrapperProperty;
+import software.amazon.jdbc.HighestWeightHostSelector;
 import software.amazon.jdbc.HostRole;
 import software.amazon.jdbc.HostSpec;
 import software.amazon.jdbc.PluginService;
@@ -35,7 +36,6 @@
 import software.amazon.jdbc.util.Messages;
 import software.amazon.jdbc.util.SlidingExpirationCacheWithCleanupThread;
 import software.amazon.jdbc.util.Utils;
-import software.amazon.jdbc.wrapper.HighestWeightHostSelector;
 
 public class LimitlessRouterServiceImpl implements LimitlessRouterService {
   private static final Logger LOGGER =

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

@@ -75,7 +75,8 @@ public Object getArray(Map<String, Class<?>> map) throws SQLException {
         this.pluginManager,
         this.array,
         "Array.getArray",
-        () -> this.array.getArray());
+        () -> this.array.getArray(map),
+        map);
   }
 
   @Override
@@ -86,7 +87,9 @@ public Object getArray(long index, int count) throws SQLException {
         this.pluginManager,
         this.array,
         "Array.getArray",
-        () -> this.array.getArray());
+        () -> this.array.getArray(index, count),
+        index,
+        count);
   }
 
   @Override
@@ -97,7 +100,10 @@ public Object getArray(long index, int count, Map<String, Class<?>> map) throws
         this.pluginManager,
         this.array,
         "Array.getArray",
-        () -> this.array.getArray());
+        () -> this.array.getArray(index, count, map),
+        index,
+        count,
+        map);
   }
 
   @Override
@@ -119,7 +125,8 @@ public ResultSet getResultSet(Map<String, Class<?>> map) throws SQLException {
         this.pluginManager,
         this.array,
         "Array.getResultSet",
-        () -> this.array.getResultSet());
+        () -> this.array.getResultSet(map),
+        map);
   }
 
   @Override
@@ -130,7 +137,9 @@ public ResultSet getResultSet(long index, int count) throws SQLException {
         this.pluginManager,
         this.array,
         "Array.getResultSet",
-        () -> this.array.getResultSet());
+        () -> this.array.getResultSet(index, count),
+        index,
+        count);
   }
 
   @Override

wrapper/src/test/java/software/amazon/jdbc/plugin/HighestWeightHostSelectorTest.java

@@ -26,11 +26,11 @@
 import java.util.Properties;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import software.amazon.jdbc.HighestWeightHostSelector;
 import software.amazon.jdbc.HostRole;
 import software.amazon.jdbc.HostSpec;
 import software.amazon.jdbc.HostSpecBuilder;
 import software.amazon.jdbc.hostavailability.SimpleHostAvailabilityStrategy;
-import software.amazon.jdbc.wrapper.HighestWeightHostSelector;
 
 public class HighestWeightHostSelectorTest {
 

wrapper/src/test/java/software/amazon/jdbc/plugin/limitless/LimitlessRouterServiceImplTest.java

@@ -38,6 +38,7 @@
 import org.junit.jupiter.api.Test;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
+import software.amazon.jdbc.HighestWeightHostSelector;
 import software.amazon.jdbc.HostListProvider;
 import software.amazon.jdbc.HostRole;
 import software.amazon.jdbc.HostSpec;
@@ -47,7 +48,6 @@
 import software.amazon.jdbc.RoundRobinHostSelector;
 import software.amazon.jdbc.hostavailability.HostAvailability;
 import software.amazon.jdbc.hostavailability.SimpleHostAvailabilityStrategy;
-import software.amazon.jdbc.wrapper.HighestWeightHostSelector;
 
 public class LimitlessRouterServiceImplTest {