Browse Source

Polishing

pull/31496/head
Juergen Hoeller 2 years ago
parent
commit
86650d1a39
  1. 10
      spring-r2dbc/src/main/java/org/springframework/r2dbc/connection/ConnectionFactoryUtils.java
  2. 4
      spring-r2dbc/src/main/java/org/springframework/r2dbc/connection/SingleConnectionFactory.java
  3. 2
      spring-r2dbc/src/main/java/org/springframework/r2dbc/core/DefaultDatabaseClient.java
  4. 2
      spring-r2dbc/src/test/java/org/springframework/r2dbc/connection/R2dbcTransactionManagerUnitTests.java
  5. 2
      spring-r2dbc/src/test/java/org/springframework/r2dbc/connection/TransactionAwareConnectionFactoryProxyUnitTests.java
  6. 20
      spring-r2dbc/src/test/java/org/springframework/r2dbc/connection/lookup/MapConnectionFactoryLookupUnitTests.java
  7. 10
      spring-r2dbc/src/test/java/org/springframework/r2dbc/core/NamedParameterUtilsUnitTests.java
  8. 8
      spring-r2dbc/src/test/java/org/springframework/r2dbc/core/binding/IndexedBindMarkersUnitTests.java
  9. 2
      spring-r2dbc/src/test/java/org/springframework/r2dbc/core/binding/NamedBindMarkersUnitTests.java

10
spring-r2dbc/src/main/java/org/springframework/r2dbc/connection/ConnectionFactoryUtils.java

@ -302,13 +302,13 @@ public abstract class ConnectionFactoryUtils {
* @return the innermost target Connection, or the passed-in one if not wrapped * @return the innermost target Connection, or the passed-in one if not wrapped
* @see Wrapped#unwrap() * @see Wrapped#unwrap()
*/ */
@SuppressWarnings({"rawtypes", "unchecked"}) @SuppressWarnings("unchecked")
public static Connection getTargetConnection(Connection con) { public static Connection getTargetConnection(Connection con) {
Object conToUse = con; Connection conToUse = con;
while (conToUse instanceof Wrapped wrapped) { while (conToUse instanceof Wrapped<?>) {
conToUse = wrapped.unwrap(); conToUse = ((Wrapped<Connection>) conToUse).unwrap();
} }
return (Connection) conToUse; return conToUse;
} }
/** /**

4
spring-r2dbc/src/main/java/org/springframework/r2dbc/connection/SingleConnectionFactory.java

@ -1,5 +1,5 @@
/* /*
* Copyright 2002-2021 the original author or authors. * Copyright 2002-2023 the original author or authors.
* *
* Licensed under the Apache License, Version 2.0 (the "License"); * Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. * you may not use this file except in compliance with the License.
@ -231,7 +231,7 @@ public class SingleConnectionFactory extends DelegatingConnectionFactory
*/ */
protected Connection getCloseSuppressingConnectionProxy(Connection target) { protected Connection getCloseSuppressingConnectionProxy(Connection target) {
return (Connection) Proxy.newProxyInstance(SingleConnectionFactory.class.getClassLoader(), return (Connection) Proxy.newProxyInstance(SingleConnectionFactory.class.getClassLoader(),
new Class<?>[] { Connection.class, Wrapped.class }, new CloseSuppressingInvocationHandler(target)); new Class<?>[] {Connection.class, Wrapped.class}, new CloseSuppressingInvocationHandler(target));
} }

2
spring-r2dbc/src/main/java/org/springframework/r2dbc/core/DefaultDatabaseClient.java

@ -186,7 +186,7 @@ final class DefaultDatabaseClient implements DatabaseClient {
*/ */
private static Connection createConnectionProxy(Connection con) { private static Connection createConnectionProxy(Connection con) {
return (Connection) Proxy.newProxyInstance(DatabaseClient.class.getClassLoader(), return (Connection) Proxy.newProxyInstance(DatabaseClient.class.getClassLoader(),
new Class<?>[] { Connection.class, Wrapped.class }, new Class<?>[] {Connection.class, Wrapped.class},
new CloseSuppressingInvocationHandler(con)); new CloseSuppressingInvocationHandler(con));
} }

2
spring-r2dbc/src/test/java/org/springframework/r2dbc/connection/R2dbcTransactionManagerUnitTests.java

@ -70,7 +70,7 @@ class R2dbcTransactionManagerUnitTests {
@BeforeEach @BeforeEach
@SuppressWarnings({"unchecked", "rawtypes"}) @SuppressWarnings({"rawtypes", "unchecked"})
void before() { void before() {
when(connectionFactoryMock.create()).thenReturn((Mono) Mono.just(connectionMock)); when(connectionFactoryMock.create()).thenReturn((Mono) Mono.just(connectionMock));
when(connectionMock.beginTransaction(any(io.r2dbc.spi.TransactionDefinition.class))).thenReturn(Mono.empty()); when(connectionMock.beginTransaction(any(io.r2dbc.spi.TransactionDefinition.class))).thenReturn(Mono.empty());

2
spring-r2dbc/src/test/java/org/springframework/r2dbc/connection/TransactionAwareConnectionFactoryProxyUnitTests.java

@ -57,7 +57,7 @@ class TransactionAwareConnectionFactoryProxyUnitTests {
@BeforeEach @BeforeEach
@SuppressWarnings({ "rawtypes", "unchecked" }) @SuppressWarnings({"rawtypes", "unchecked"})
void before() { void before() {
when(connectionFactoryMock.create()).thenReturn((Mono) Mono.just(connectionMock1), when(connectionFactoryMock.create()).thenReturn((Mono) Mono.just(connectionMock1),
(Mono) Mono.just(connectionMock2), (Mono) Mono.just(connectionMock3)); (Mono) Mono.just(connectionMock2), (Mono) Mono.just(connectionMock3));

20
spring-r2dbc/src/test/java/org/springframework/r2dbc/connection/lookup/MapConnectionFactoryLookupUnitTests.java

@ -34,13 +34,14 @@ public class MapConnectionFactoryLookupUnitTests {
private static final String CONNECTION_FACTORY_NAME = "connectionFactory"; private static final String CONNECTION_FACTORY_NAME = "connectionFactory";
@Test @Test
public void getConnectionFactoriesReturnsUnmodifiableMap() { public void getConnectionFactoriesReturnsUnmodifiableMap() {
MapConnectionFactoryLookup lookup = new MapConnectionFactoryLookup(); MapConnectionFactoryLookup lookup = new MapConnectionFactoryLookup();
Map<String, ConnectionFactory> connectionFactories = lookup.getConnectionFactories(); Map<String, ConnectionFactory> connectionFactories = lookup.getConnectionFactories();
assertThatThrownBy(() -> connectionFactories.put("", assertThatThrownBy(() -> connectionFactories.put("", new DummyConnectionFactory()))
new DummyConnectionFactory())).isInstanceOf(UnsupportedOperationException.class); .isInstanceOf(UnsupportedOperationException.class);
} }
@Test @Test
@ -52,8 +53,8 @@ public class MapConnectionFactoryLookupUnitTests {
MapConnectionFactoryLookup lookup = new MapConnectionFactoryLookup(); MapConnectionFactoryLookup lookup = new MapConnectionFactoryLookup();
lookup.setConnectionFactories(connectionFactories); lookup.setConnectionFactories(connectionFactories);
ConnectionFactory connectionFactory = lookup.getConnectionFactory(CONNECTION_FACTORY_NAME); assertThat(lookup.getConnectionFactory(CONNECTION_FACTORY_NAME))
assertThat(connectionFactory).isNotNull().isSameAs(expectedConnectionFactory); .isNotNull().isSameAs(expectedConnectionFactory);
} }
@Test @Test
@ -67,12 +68,12 @@ public class MapConnectionFactoryLookupUnitTests {
lookup.setConnectionFactories(connectionFactories); lookup.setConnectionFactories(connectionFactories);
lookup.addConnectionFactory(CONNECTION_FACTORY_NAME, expectedConnectionFactory); lookup.addConnectionFactory(CONNECTION_FACTORY_NAME, expectedConnectionFactory);
ConnectionFactory connectionFactory = lookup.getConnectionFactory(CONNECTION_FACTORY_NAME); assertThat(lookup.getConnectionFactory(CONNECTION_FACTORY_NAME))
assertThat(connectionFactory).isNotNull().isSameAs(expectedConnectionFactory); .isNotNull().isSameAs(expectedConnectionFactory);
} }
@Test @Test
@SuppressWarnings({ "unchecked", "rawtypes" }) @SuppressWarnings({"rawtypes", "unchecked"})
public void getConnectionFactoryWhereSuppliedMapHasNonConnectionFactoryTypeUnderSpecifiedKey() { public void getConnectionFactoryWhereSuppliedMapHasNonConnectionFactoryTypeUnderSpecifiedKey() {
Map connectionFactories = new HashMap<>(); Map connectionFactories = new HashMap<>();
connectionFactories.put(CONNECTION_FACTORY_NAME, new Object()); connectionFactories.put(CONNECTION_FACTORY_NAME, new Object());
@ -86,8 +87,9 @@ public class MapConnectionFactoryLookupUnitTests {
public void getConnectionFactoryWhereSuppliedMapHasNoEntryForSpecifiedKey() { public void getConnectionFactoryWhereSuppliedMapHasNoEntryForSpecifiedKey() {
MapConnectionFactoryLookup lookup = new MapConnectionFactoryLookup(); MapConnectionFactoryLookup lookup = new MapConnectionFactoryLookup();
assertThatThrownBy(() -> lookup.getConnectionFactory(CONNECTION_FACTORY_NAME)) assertThatThrownBy(
.isInstanceOf(ConnectionFactoryLookupFailureException.class); () -> lookup.getConnectionFactory(CONNECTION_FACTORY_NAME)).isInstanceOf(
ConnectionFactoryLookupFailureException.class);
} }
} }

10
spring-r2dbc/src/test/java/org/springframework/r2dbc/core/NamedParameterUtilsUnitTests.java

@ -84,8 +84,9 @@ public class NamedParameterUtilsUnitTests {
@Test @Test
public void substituteObjectArray() { public void substituteObjectArray() {
MapBindParameterSource namedParams = new MapBindParameterSource(new HashMap<>()); MapBindParameterSource namedParams = new MapBindParameterSource(new HashMap<>());
namedParams.addValue("a", Arrays.asList(new Object[] { "Walter", "Heisenberg" }, namedParams.addValue("a",
new Object[] { "Walt Jr.", "Flynn" })); Arrays.asList(new Object[] {"Walter", "Heisenberg"},
new Object[] {"Walt Jr.", "Flynn"}));
PreparedOperation<?> operation = NamedParameterUtils.substituteNamedParameters( PreparedOperation<?> operation = NamedParameterUtils.substituteNamedParameters(
"xxx :a", BIND_MARKERS, namedParams); "xxx :a", BIND_MARKERS, namedParams);
@ -96,8 +97,9 @@ public class NamedParameterUtilsUnitTests {
@Test @Test
public void shouldBindObjectArray() { public void shouldBindObjectArray() {
MapBindParameterSource namedParams = new MapBindParameterSource(new HashMap<>()); MapBindParameterSource namedParams = new MapBindParameterSource(new HashMap<>());
namedParams.addValue("a", Arrays.asList(new Object[] { "Walter", "Heisenberg" }, namedParams.addValue("a",
new Object[] { "Walt Jr.", "Flynn" })); Arrays.asList(new Object[] {"Walter", "Heisenberg"},
new Object[] {"Walt Jr.", "Flynn"}));
BindTarget bindTarget = mock(); BindTarget bindTarget = mock();

8
spring-r2dbc/src/test/java/org/springframework/r2dbc/core/binding/IndexedBindMarkersUnitTests.java

@ -32,7 +32,6 @@ class IndexedBindMarkersUnitTests {
@Test @Test
void shouldCreateNewBindMarkers() { void shouldCreateNewBindMarkers() {
BindMarkersFactory factory = BindMarkersFactory.indexed("$", 0); BindMarkersFactory factory = BindMarkersFactory.indexed("$", 0);
BindMarkers bindMarkers1 = factory.create(); BindMarkers bindMarkers1 = factory.create();
BindMarkers bindMarkers2 = factory.create(); BindMarkers bindMarkers2 = factory.create();
@ -43,7 +42,6 @@ class IndexedBindMarkersUnitTests {
@Test @Test
void shouldCreateNewBindMarkersWithOffset() { void shouldCreateNewBindMarkersWithOffset() {
BindTarget bindTarget = mock(); BindTarget bindTarget = mock();
BindMarkers bindMarkers = BindMarkersFactory.indexed("$", 1).create(); BindMarkers bindMarkers = BindMarkersFactory.indexed("$", 1).create();
BindMarker first = bindMarkers.next(); BindMarker first = bindMarkers.next();
@ -60,10 +58,9 @@ class IndexedBindMarkersUnitTests {
@Test @Test
void nextShouldIncrementBindMarker() { void nextShouldIncrementBindMarker() {
String[] prefixes = { "$", "?" }; String[] prefixes = {"$", "?"};
for (String prefix : prefixes) { for (String prefix : prefixes) {
BindMarkers bindMarkers = BindMarkersFactory.indexed(prefix, 0).create(); BindMarkers bindMarkers = BindMarkersFactory.indexed(prefix, 0).create();
BindMarker marker1 = bindMarkers.next(); BindMarker marker1 = bindMarkers.next();
@ -76,9 +73,7 @@ class IndexedBindMarkersUnitTests {
@Test @Test
void bindValueShouldBindByIndex() { void bindValueShouldBindByIndex() {
BindTarget bindTarget = mock(); BindTarget bindTarget = mock();
BindMarkers bindMarkers = BindMarkersFactory.indexed("$", 0).create(); BindMarkers bindMarkers = BindMarkersFactory.indexed("$", 0).create();
bindMarkers.next().bind(bindTarget, "foo"); bindMarkers.next().bind(bindTarget, "foo");
@ -91,7 +86,6 @@ class IndexedBindMarkersUnitTests {
@Test @Test
void bindNullShouldBindByIndex() { void bindNullShouldBindByIndex() {
BindTarget bindTarget = mock(); BindTarget bindTarget = mock();
BindMarkers bindMarkers = BindMarkersFactory.indexed("$", 0).create(); BindMarkers bindMarkers = BindMarkersFactory.indexed("$", 0).create();
bindMarkers.next(); // ignore bindMarkers.next(); // ignore

2
spring-r2dbc/src/test/java/org/springframework/r2dbc/core/binding/NamedBindMarkersUnitTests.java

@ -43,7 +43,7 @@ class NamedBindMarkersUnitTests {
} }
@ParameterizedTest @ParameterizedTest
@ValueSource(strings = { "$", "?" }) @ValueSource(strings = {"$", "?"})
void nextShouldIncrementBindMarker(String prefix) { void nextShouldIncrementBindMarker(String prefix) {
BindMarkers bindMarkers = BindMarkersFactory.named(prefix, "p", 32).create(); BindMarkers bindMarkers = BindMarkersFactory.named(prefix, "p", 32).create();

Loading…
Cancel
Save