Skip to content

Commit 1c9c30c

Browse files
committed
Polish
1 parent dca726a commit 1c9c30c

File tree

2 files changed

+21
-26
lines changed

2 files changed

+21
-26
lines changed

spring-batch-infrastructure/src/main/java/org/springframework/batch/item/database/AbstractCursorItemReader.java

+11-16
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,7 @@
102102
* @author Peter Zozom
103103
* @author Robert Kasanicky
104104
* @author Thomas Risberg
105+
* @author Michael Minella
105106
*/
106107
public abstract class AbstractCursorItemReader<T> extends AbstractItemCountingItemStreamItemReader<T>
107108
implements InitializingBean {
@@ -134,9 +135,9 @@ public abstract class AbstractCursorItemReader<T> extends AbstractItemCountingIt
134135

135136
private boolean useSharedExtendedConnection = false;
136137

137-
private Boolean connectionAutoCommit;
138+
private boolean connectionAutoCommit;
138139

139-
private Boolean initialConnectionAutoCommit;
140+
private boolean initialConnectionAutoCommit;
140141

141142
public AbstractCursorItemReader() {
142143
super();
@@ -364,20 +365,12 @@ public boolean isUseSharedExtendedConnection() {
364365
* Connection / Datasource default configuration.
365366
*
366367
* @param autoCommit value used for {@link Connection#setAutoCommit(boolean)}.
368+
* @since 4.0
367369
*/
368370
public void setConnectionAutoCommit(boolean autoCommit) {
369371
this.connectionAutoCommit = autoCommit;
370372
}
371373

372-
/**
373-
* Return whether "autoCommit" should be overridden for the connection used by the cursor.
374-
*
375-
* @return the "autoCommit" value, or {@code null} if none to be applied.
376-
*/
377-
public Boolean getConnectionAutoCommit() {
378-
return this.connectionAutoCommit;
379-
}
380-
381374
public abstract String getSql();
382375

383376
/**
@@ -402,9 +395,9 @@ protected void doClose() throws Exception {
402395
JdbcUtils.closeResultSet(this.rs);
403396
rs = null;
404397
cleanupOnClose();
405-
if (this.initialConnectionAutoCommit != null) {
406-
this.con.setAutoCommit(initialConnectionAutoCommit);
407-
}
398+
399+
this.con.setAutoCommit(initialConnectionAutoCommit);
400+
408401
if (useSharedExtendedConnection && dataSource instanceof ExtendedConnectionDataSourceProxy) {
409402
((ExtendedConnectionDataSourceProxy)dataSource).stopCloseSuppression(this.con);
410403
if (!TransactionSynchronizationManager.isActualTransactionActive()) {
@@ -449,8 +442,10 @@ protected void initializeConnection() {
449442
else {
450443
this.con = dataSource.getConnection();
451444
}
452-
if (this.connectionAutoCommit != null && this.con.getAutoCommit() != this.connectionAutoCommit) {
453-
this.initialConnectionAutoCommit = this.con.getAutoCommit();
445+
446+
this.initialConnectionAutoCommit = this.con.getAutoCommit();
447+
448+
if (this.con.getAutoCommit() != this.connectionAutoCommit) {
454449
this.con.setAutoCommit(this.connectionAutoCommit);
455450
}
456451
}

spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/JdbcCursorItemReaderConfigTests.java

+10-10
Original file line numberDiff line numberDiff line change
@@ -15,30 +15,30 @@
1515
*/
1616
package org.springframework.batch.item.database;
1717

18-
import static org.junit.Assert.assertEquals;
19-
import static org.mockito.Mockito.mock;
20-
import static org.mockito.Mockito.reset;
21-
import static org.mockito.Mockito.times;
22-
import static org.mockito.Mockito.verify;
23-
import static org.mockito.Mockito.when;
24-
2518
import java.sql.Connection;
2619
import java.sql.PreparedStatement;
2720
import java.sql.ResultSet;
28-
2921
import javax.sql.DataSource;
3022

3123
import org.junit.Test;
3224
import org.junit.runner.RunWith;
3325
import org.junit.runners.JUnit4;
3426
import org.mockito.ArgumentCaptor;
27+
3528
import org.springframework.batch.item.ExecutionContext;
3629
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
3730
import org.springframework.transaction.PlatformTransactionManager;
3831
import org.springframework.transaction.TransactionStatus;
3932
import org.springframework.transaction.support.TransactionCallback;
4033
import org.springframework.transaction.support.TransactionTemplate;
4134

35+
import static org.junit.Assert.assertEquals;
36+
import static org.mockito.Mockito.mock;
37+
import static org.mockito.Mockito.reset;
38+
import static org.mockito.Mockito.times;
39+
import static org.mockito.Mockito.verify;
40+
import static org.mockito.Mockito.when;
41+
4242
@RunWith(JUnit4.class)
4343
public class JdbcCursorItemReaderConfigTests {
4444

@@ -107,8 +107,8 @@ public Void doInTransaction(TransactionStatus status) {
107107

108108
@Test
109109
public void testOverrideConnectionAutoCommit() throws Exception {
110-
boolean initialAutoCommit= false;
111-
boolean neededAutoCommit = true;
110+
boolean initialAutoCommit= false;
111+
boolean neededAutoCommit = true;
112112

113113
DataSource ds = mock(DataSource.class);
114114
Connection con = mock(Connection.class);

0 commit comments

Comments
 (0)