Change executeWithRetry signature to accept ODatabaseSession

orientechnologies · Aug 1, 2017 · 616b42d · 616b42d
1 parent f584f6b
commit 616b42d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/main/java/com/orientechnologies/orient/core/db/ODatabase.java b/core/src/main/java/com/orientechnologies/orient/core/db/ODatabase.java
@@ -1141,7 +1141,7 @@ default OResultSet execute(String language, String script, Map<String, ?> args)
    * @throws IllegalArgumentException if nRetries is <= 0
    * @throws UnsupportedOperationException if this type of database does not support automatic commit/retry
    */
-  default <T> T executeWithRetry(int nRetries, Function<ODatabase, T> function)
+  default <T> T executeWithRetry(int nRetries, Function<ODatabaseSession, T> function)
       throws IllegalStateException, IllegalArgumentException, ONeedRetryException, UnsupportedOperationException {
     if (nRetries < 1) {
       throw new IllegalArgumentException("invalid number of retries: " + nRetries);
@@ -1162,7 +1162,7 @@ default <T> T executeWithRetry(int nRetries, Function<ODatabase, T> function)
 
     for (int i = 0; i < nRetries; i++) {
       try {
-        result = function.apply(this);
+        result = function.apply((ODatabaseSession) this);
         commit();
         break;
       } catch (ONeedRetryException e) {