From 57a7a125c024c90a47467dbc49338e2127cadc8b Mon Sep 17 00:00:00 2001
From: Harald Musum <musum@yahooinc.com>
Date: Wed, 6 Jul 2022 15:02:38 +0200
Subject: Temporarily add env varioable for forcing download of file references

This can be used for testing file distributuion on one node.
E.g. config proxy takes no config or feature flag, so need to use an
environment variable.
---
 .../main/java/com/yahoo/vespa/filedistribution/FileDownloader.java   | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'filedistribution')

diff --git a/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/FileDownloader.java b/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/FileDownloader.java
index dc87ae2a0b4..63ae8faacfe 100644
--- a/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/FileDownloader.java
+++ b/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/FileDownloader.java
@@ -31,6 +31,7 @@ public class FileDownloader implements AutoCloseable {
     private static final Logger log = Logger.getLogger(FileDownloader.class.getName());
     private static final Duration defaultSleepBetweenRetries = Duration.ofSeconds(5);
     public static final File defaultDownloadDirectory = new File(Defaults.getDefaults().underVespaHome("var/db/vespa/filedistribution"));
+    private static final boolean forceDownload = Boolean.parseBoolean(System.getenv("VESPA_CONFIG_PROXY_FORCE_DOWNLOAD_OF_FILE_REFERENCES"));
 
     private final ConnectionPool connectionPool;
     private final Supervisor supervisor;
@@ -65,6 +66,8 @@ public class FileDownloader implements AutoCloseable {
                                                                    sleepBetweenRetries,
                                                                    downloadDirectory,
                                                                    acceptedCompressionTypes);
+        if (forceDownload)
+            log.log(Level.INFO, "Force download of file references (download even if file reference exists on disk)");
     }
 
     public Optional<File> getFile(FileReferenceDownload fileReferenceDownload) {
@@ -103,6 +106,8 @@ public class FileDownloader implements AutoCloseable {
     }
 
     private static Optional<File> getFileFromFileSystem(FileReference fileReference, File downloadDirectory) {
+        if (forceDownload) return Optional.empty();
+
         File[] files = new File(downloadDirectory, fileReference.value()).listFiles();
         if (files == null) return Optional.empty();
         if (files.length == 0) return Optional.empty();
-- 
cgit v1.2.3