Fix using multi query hints

Fix using multi query hints #3109
OpenFeign · Mar 27, 2024 · de86680 · de86680
1 parent da5b440
commit de86680
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/querydsl-jpa/src/main/java/com/querydsl/jpa/impl/AbstractJPAQuery.java b/querydsl-jpa/src/main/java/com/querydsl/jpa/impl/AbstractJPAQuery.java
@@ -14,9 +14,12 @@
 package com.querydsl.jpa.impl;
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.LinkedHashMap;
+import java.util.LinkedHashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 import java.util.stream.Stream;
@@ -46,9 +49,9 @@
  */
 public abstract class AbstractJPAQuery<T, Q extends AbstractJPAQuery<T, Q>> extends JPAQueryBase<T, Q> {
 
-    private static final Logger logger = Logger.getLogger(JPAQuery.class.getName());
+    private static final Logger logger = Logger.getLogger(AbstractJPAQuery.class.getName());
 
-    protected final Map<String, Object> hints = new LinkedHashMap<>();
+    protected final Map<String, Collection<Object>> hints = new LinkedHashMap<>();
 
     protected final EntityManager entityManager;
 
@@ -148,8 +151,8 @@ protected Query createQuery(@Nullable QueryModifiers modifiers, boolean forCount
             query.setFlushMode(flushMode);
         }
 
-        for (Map.Entry<String, Object> entry : hints.entrySet()) {
-            query.setHint(entry.getKey(), entry.getValue());
+        for (Map.Entry<String, Collection<Object>> entry : hints.entrySet()) {
+            entry.getValue().forEach(value -> query.setHint(entry.getKey(), value));
         }
 
         // set transformer, if necessary and possible
@@ -348,7 +351,8 @@ public Q setFlushMode(FlushModeType flushMode) {
 
     @SuppressWarnings("unchecked")
     public Q setHint(String name, Object value) {
-        hints.put(name, value);
+        hints.computeIfAbsent(name,key -> new LinkedHashSet<>());
+        hints.get(name).add(value);
         return (Q) this;
     }