Revert "Delete file reference through file directory [run-systemtest]"

1 files changed, 36 insertions, 28 deletions

diff --git a/configserver/src/main/java/com/yahoo/vespa/config/server/session/SessionRepository.java b/configserver/src/main/java/com/yahoo/vespa/config/server/session/SessionRepository.java
index 2a863e58a21..0c7fc217e40 100644
--- a/configserver/src/main/java/com/yahoo/vespa/config/server/session/SessionRepository.java
+++ b/configserver/src/main/java/com/yahoo/vespa/config/server/session/SessionRepository.java
@@ -29,7 +29,6 @@ import com.yahoo.vespa.config.server.application.PermanentApplicationPackage;
 import com.yahoo.vespa.config.server.application.TenantApplications;
 import com.yahoo.vespa.config.server.configchange.ConfigChangeActions;
 import com.yahoo.vespa.config.server.deploy.TenantFileSystemDirs;
-import com.yahoo.vespa.config.server.filedistribution.FileDirectory;
 import com.yahoo.vespa.config.server.filedistribution.FileDistributionFactory;
 import com.yahoo.vespa.config.server.http.InvalidApplicationException;
 import com.yahoo.vespa.config.server.http.UnknownVespaVersionException;
@@ -579,7 +578,13 @@ public class SessionRepository {
         zkWatcherExecutor.execute(() -> {
             log.log(Level.FINE, () -> "Got child event: " + event);
             switch (event.getType()) {
-                case CHILD_ADDED, CHILD_REMOVED, CONNECTION_RECONNECTED -> sessionsChanged();
+                case CHILD_ADDED:
+                case CHILD_REMOVED:
+                case CONNECTION_RECONNECTED:
+                    sessionsChanged();
+                    break;
+                default:
+                    break;
             }
         });
     }
@@ -630,7 +635,9 @@ public class SessionRepository {
             }
 
             sessionIdsToDelete.forEach(this::deleteLocalSession);
-        } catch (Throwable e) { // Make sure to catch here, to avoid executor just dying in case of issues ...
+
+            // Make sure to catch here, to avoid executor just dying in case of issues ...
+        } catch (Throwable e) {
             log.log(Level.WARNING, "Error when purging old sessions ", e);
         }
         log.log(Level.FINE, () -> "Done purging old sessions");
@@ -776,12 +783,13 @@ public class SessionRepository {
     }
 
     private Optional<ApplicationSet> getApplicationSet(long sessionId) {
+        Optional<ApplicationSet> applicationSet = Optional.empty();
         try {
-            return Optional.ofNullable(getRemoteSession(sessionId)).map(this::ensureApplicationLoaded);
+            applicationSet = Optional.ofNullable(getRemoteSession(sessionId)).map(this::ensureApplicationLoaded);
         } catch (IllegalArgumentException e) {
             // Do nothing if we have no currently active session
-            return Optional.empty();
         }
+        return applicationSet;
     }
 
     private void copyApp(File sourceDir, File destinationDir) throws IOException {
@@ -852,9 +860,9 @@ public class SessionRepository {
     }
 
     /**
-     * Create a new local session for the given session id if it does not already exist and
-     * will add the session to the local session cache. If there is no remote session matching
-     * the session id the remote session will also be created.
+     * Create a new local session for the given session id if it does not already exist.
+     * Will also add the session to the local session cache if necessary. If there is no
+     * remote session matching the session it will also be created.
      */
     public void createLocalSessionFromDistributedApplicationPackage(long sessionId) {
         if (applicationRepo.sessionExistsInFileSystem(sessionId)) {
@@ -866,26 +874,26 @@ public class SessionRepository {
         SessionZooKeeperClient sessionZKClient = createSessionZooKeeperClient(sessionId);
         FileReference fileReference = sessionZKClient.readApplicationPackageReference();
         log.log(Level.FINE, () -> "File reference for session id " + sessionId + ": " + fileReference);
-        if (fileReference == null) throw new RuntimeException("file reference for application package is null");
-
-        File sessionDir;
-        FileDirectory fileDirectory = fileDistributionFactory.fileDirectory();
-        try {
-            sessionDir = fileDirectory.getFile(fileReference);
-        } catch (IllegalArgumentException e) {
-            // We cannot be guaranteed that the file reference exists (it could be that it has not
-            // been downloaded yet), and e.g. when bootstrapping we cannot throw an exception in that case
-            log.log(Level.FINE, () -> "File reference for session id " + sessionId + ": " + fileReference + " not found");
-            return;
-        }
-        ApplicationId applicationId = sessionZKClient.readApplicationId()
-                                                     .orElseThrow(() -> new RuntimeException("Could not find application id for session " + sessionId));
-        log.log(Level.FINE, () -> "Creating local session for tenant '" + tenantName + "' with session id " + sessionId);
-        try {
-            createLocalSession(sessionDir, applicationId, sessionZKClient.readTags(), sessionId);
-        } finally {
-            log.log(Level.FINE, "Deleting file distribution reference " + fileReference + " for app package with session id " + sessionId);
-            fileDirectory.delete(fileReference, (reference) -> true); // Delete downloaded file reference, not needed anymore
+        if (fileReference != null) {
+            File sessionDir;
+            try {
+                sessionDir = fileDistributionFactory.fileDirectory().getFile(fileReference);
+            } catch (IllegalArgumentException e) {
+                // We cannot be guaranteed that the file reference exists (it could be that it has not
+                // been downloaded yet), and e.g. when bootstrapping we cannot throw an exception in that case
+                log.log(Level.FINE, () -> "File reference for session id " + sessionId + ": " + fileReference + " not found");
+                return;
+            }
+            ApplicationId applicationId = sessionZKClient.readApplicationId()
+                    .orElseThrow(() -> new RuntimeException("Could not find application id for session " + sessionId));
+            log.log(Level.FINE, () -> "Creating local session for tenant '" + tenantName + "' with session id " + sessionId);
+            try {
+                createLocalSession(sessionDir, applicationId, sessionZKClient.readTags(), sessionId);
+            } finally {
+                // Delete downloaded file reference, not needed anymore
+                log.log(Level.FINE, "Deleting file distribution reference for app package with session id " + sessionDir);
+                IOUtils.recursiveDeleteDir(sessionDir);
+            }
         }
     }