From 8c369f1998c95df6c9f7570dc1e2141d081c3e26 Mon Sep 17 00:00:00 2001
From: Fabio Massimo Ercoli <fabiomassimo.ercoli@gmail.com>
Date: Tue, 23 Nov 2021 19:59:15 +0100
Subject: [PATCH] Test concurrent queries with multiple params

---
 .../query/sqm/ConcurrentQueriesByIdsTest.java | 103 ++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 hibernate-core/src/test/java/org/hibernate/orm/test/query/sqm/ConcurrentQueriesByIdsTest.java

diff --git a/hibernate-core/src/test/java/org/hibernate/orm/test/query/sqm/ConcurrentQueriesByIdsTest.java b/hibernate-core/src/test/java/org/hibernate/orm/test/query/sqm/ConcurrentQueriesByIdsTest.java
new file mode 100644
index 0000000000..976b921b05
--- /dev/null
+++ b/hibernate-core/src/test/java/org/hibernate/orm/test/query/sqm/ConcurrentQueriesByIdsTest.java
@@ -0,0 +1,103 @@
+/*
+ * Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * License: GNU Lesser General Public License (LGPL), version 2.1 or later
+ * See the lgpl.txt file in the root directory or http://www.gnu.org/licenses/lgpl-2.1.html
+ */
+package org.hibernate.orm.test.query.sqm;
+
+import java.util.Arrays;
+import java.util.List;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import jakarta.persistence.Basic;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+
+import org.hibernate.Session;
+
+import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
+import org.junit.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class ConcurrentQueriesByIdsTest extends BaseCoreFunctionalTestCase {
+
+	public static final String QUERY_STRING = "select e from simple e where e.id in (:ids)";
+
+	@Override
+	protected Class<?>[] getAnnotatedClasses() {
+		return new Class[] { SimpleEntity.class };
+	}
+
+	@Test
+	public void run() throws InterruptedException {
+		inTransaction( session -> {
+			for ( int i = 0; i < 100; i++ ) {
+				SimpleEntity entity = new SimpleEntity();
+				entity.setId( i );
+				entity.setName( "Name: " + i );
+				session.persist( entity );
+			}
+		} );
+
+		ExecutorService executorService = Executors.newFixedThreadPool( 3 );
+		CompletableFuture<List<SimpleEntity>>[] results = new CompletableFuture[10];
+
+		for ( int i = 0; i < 10; i++ ) {
+			int index = i;
+			results[i] = CompletableFuture.supplyAsync( () -> executeQuery( index ), executorService );
+		}
+		for ( int i = 0; i < 10; i++ ) {
+			assertThat( results[i].join() ).hasSize( 10 );
+		}
+
+		executorService.shutdown();
+	}
+
+	private List<SimpleEntity> executeQuery(int index) {
+		try (Session session = sessionFactory().openSession()) {
+			return executeQuery( session, index );
+		}
+	}
+
+	private List<SimpleEntity> executeQuery(Session session, int index) {
+		int base = index * 10;
+
+		return session.createQuery( QUERY_STRING, SimpleEntity.class )
+				.setParameter(
+						"ids",
+						Arrays.asList( base + 0, base + 1, base + 2, base + 3, base + 4, base + 5,
+									   base + 6, base + 7, base + 8, base + 9
+						)
+				)
+				.list();
+	}
+
+	@Entity(name = "simple")
+	public static class SimpleEntity {
+
+		@Id
+		private Integer id;
+
+		@Basic
+		private String name;
+
+		public Integer getId() {
+			return id;
+		}
+
+		public void setId(Integer id) {
+			this.id = id;
+		}
+
+		public String getName() {
+			return name;
+		}
+
+		public void setName(String name) {
+			this.name = name;
+		}
+	}
+}