Add privileges for removing files (#45)

Signed-off-by: Sruti Parthiban <partsrut@amazon.com>
opensearch-project · Aug 17, 2021 · 2cbc697 · 2cbc697
1 parent 63cd019
commit 2cbc697
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/...ain/java/org/opensearch/performanceanalyzer/reader_writer_shared/EventLogFileHandler.java b/...ain/java/org/opensearch/performanceanalyzer/reader_writer_shared/EventLogFileHandler.java
@@ -182,7 +182,11 @@ private void readInternal(Path pathToFile, int bufferSize, EventDispatcher proce
     }
 
     public void deleteAllFiles() {
-        LOG.debug("Cleaning up any leftover files.");
+        Util.invokePrivileged(this::deleteAllFilesWithPrivilege);
+    }
+
+    public void deleteAllFilesWithPrivilege() {
+        LOG.debug("Cleaning up any leftover files in [{}]", metricsLocation);
         File root = new File(metricsLocation);
         String[] filesToDelete = root.list();
         if (filesToDelete == null) {
@@ -202,7 +206,7 @@ public void deleteFiles(List<String> filesToDelete) {
         File root = new File(metricsLocation);
         for (String fileToDelete : filesToDelete) {
             File file = new File(root, fileToDelete);
-            PerformanceAnalyzerMetrics.removeMetrics(file);
+            removeFilesWithPrivilege(file);
             filesDeletedCount += 1;
         }
         long duration = System.currentTimeMillis() - startTime;
@@ -212,4 +216,8 @@ public void deleteFiles(List<String> filesToDelete) {
                 WriterMetrics.EVENT_LOG_FILES_DELETED, "", filesDeletedCount);
         LOG.debug("'{}' Old writer files cleaned up.", filesDeletedCount);
     }
+
+    public void removeFilesWithPrivilege(File file) {
+        Util.invokePrivileged(() -> PerformanceAnalyzerMetrics.removeMetrics(file));
+    }
 }