Skip to content

Commit 4b5dab6

Browse files
committed
fix
1 parent 671ab33 commit 4b5dab6

2 files changed

Lines changed: 8 additions & 9 deletions

File tree

benchmarks/tpcc/src/main/java/com/google/cloud/pgadapter/tpcc/SessionHelper.java

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ public SessionHelper(SessionFactory sessionFactory) {
1212
}
1313

1414
public Session createSession(boolean readOnly, boolean autoBatchDml, String transactionTag) {
15-
if (readOnly) return createReadOnlySession(transactionTag);
15+
if (readOnly) return createReadOnlySession();
1616
if (autoBatchDml) return createAutoBatchDmlSession(transactionTag);
1717
return createReadWriteSession(transactionTag);
1818
}
@@ -21,10 +21,9 @@ public Session createReadWriteSession(String transactionTag) {
2121
Session session = sessionFactory.openSession();
2222
session.doWork(
2323
conn -> {
24+
conn.createStatement().execute("RESET ALL");
2425
conn.setReadOnly(false);
25-
conn.createStatement().execute("set auto_batch_dml=false");
26-
conn.createStatement().execute("set auto_batch_dml_update_count_verification=true");
27-
conn.createStatement().execute("set transaction_tag='" + transactionTag + "'");
26+
//conn.createStatement().execute("set transaction_tag='" + transactionTag + "'");
2827
});
2928
return session;
3029
}
@@ -33,22 +32,21 @@ public Session createAutoBatchDmlSession(String transactionTag) {
3332
Session session = sessionFactory.openSession();
3433
session.doWork(
3534
conn -> {
35+
conn.createStatement().execute("RESET ALL");
3636
conn.setReadOnly(false);
3737
conn.createStatement().execute("set auto_batch_dml=true");
3838
conn.createStatement().execute("set auto_batch_dml_update_count_verification=false");
39-
conn.createStatement().execute("set transaction_tag='" + transactionTag + "'");
39+
//conn.createStatement().execute("set transaction_tag='" + transactionTag + "'");
4040
});
4141
return session;
4242
}
4343

44-
public Session createReadOnlySession(String transactionTag) {
44+
public Session createReadOnlySession() {
4545
Session session = sessionFactory.openSession();
4646
session.doWork(
4747
conn -> {
48+
conn.createStatement().execute("RESET ALL");
4849
conn.setReadOnly(true);
49-
conn.createStatement().execute("set auto_batch_dml=false");
50-
conn.createStatement().execute("set auto_batch_dml_update_count_verification=false");
51-
conn.createStatement().execute("set transaction_tag='" + transactionTag + "'");
5250
});
5351
return session;
5452
}

benchmarks/tpcc/src/main/java/com/google/cloud/pgadapter/tpcc/entities/Customer.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
import jakarta.persistence.Table;
2626
import java.math.BigDecimal;
2727
import java.sql.Timestamp;
28+
import org.hibernate.annotations.DynamicUpdate;
2829

2930
@Entity
3031
@Table(name = "customer")

0 commit comments

Comments
 (0)