filter number of logs

1 files changed, 29 insertions, 3 deletions

diff --git a/container-core/src/main/java/com/yahoo/container/handler/LogReader.java b/container-core/src/main/java/com/yahoo/container/handler/LogReader.java
index eb00446dd0e..9ac40158e07 100644
--- a/container-core/src/main/java/com/yahoo/container/handler/LogReader.java
+++ b/container-core/src/main/java/com/yahoo/container/handler/LogReader.java
@@ -7,21 +7,42 @@ import javax.xml.bind.DatatypeConverter;
 import java.io.File;
 import java.io.IOException;
 import java.nio.file.Files;
+import java.util.Arrays;
+import java.util.Comparator;
 
 public class LogReader {
 
-    protected static JSONObject readLogs(String logDirectory) throws IOException, JSONException {
+    int numberOfLogs;
+
+    public LogReader() {
+        this.numberOfLogs = -1;
+    }
+
+    public LogReader(int numberOfLogs) {
+        this.numberOfLogs = numberOfLogs;
+    }
+
+    protected JSONObject readLogs(String logDirectory) throws IOException, JSONException {
         JSONObject json = new JSONObject();
         File root = new File(logDirectory);
         traverse_folder(root, json);
         return json;
     }
 
-    private static void traverse_folder(File root, JSONObject json) throws IOException, JSONException {
-        for(File child : root.listFiles()) {
+    private void traverse_folder(File root, JSONObject json) throws IOException, JSONException {
+        File[] files = root.listFiles();
+        Arrays.sort(files,new Comparator<File>(){
+            public int compare(File f1, File f2)
+            {
+                return Long.valueOf(f1.lastModified()).compareTo(f2.lastModified());
+            } });
+        Arrays.sort(files, Comparator.reverseOrder());
+        for(File child : files) {
+            if (numberOfLogs == 0) return;
             JSONObject childJson = new JSONObject();
             if(child.isFile()) {
                 json.put(child.getName(), DatatypeConverter.printBase64Binary(Files.readAllBytes(child.toPath())));
+                decrementLogNumber();
             }
             else {
                 json.put(child.getName(), childJson);
@@ -29,4 +50,9 @@ public class LogReader {
             }
         }
     }
+
+    private void decrementLogNumber() {
+        numberOfLogs -= 1;
+    }
+
 }