2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6 * Copyright © 2017-2018 European Software Marketing Ltd.
7 * ================================================================================
8 * Licensed under the Apache License, Version 2.0 (the "License");
9 * you may not use this file except in compliance with the License.
10 * You may obtain a copy of the License at
12 * http://www.apache.org/licenses/LICENSE-2.0
14 * Unless required by applicable law or agreed to in writing, software
15 * distributed under the License is distributed on an "AS IS" BASIS,
16 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17 * See the License for the specific language governing permissions and
18 * limitations under the License.
19 * ============LICENSE_END=========================================================
22 package org.onap.aai.babel.logging;
24 import java.io.BufferedReader;
26 import java.io.FileReader;
27 import java.io.IOException;
28 import java.nio.file.Files;
29 import java.nio.file.Path;
30 import java.nio.file.Paths;
31 import java.util.Comparator;
32 import java.util.HashMap;
34 import java.util.Optional;
35 import java.util.concurrent.TimeUnit;
36 import org.apache.commons.lang3.time.StopWatch;
37 import org.junit.Assert;
39 public class LogReader {
41 private Map<String, Path> cachedLogMap = new HashMap<>();
42 private Map<String, BufferedReader> readersMap = new HashMap<>();
43 private BufferedReader cachedReader;
45 public LogReader(String logDirectory, String logFilePrefix) throws IOException {
46 cachedReader = getReader(logDirectory, logFilePrefix);
49 private BufferedReader getReader(String logDirectory, String logFilePrefix) throws IOException {
50 BufferedReader reader = readersMap.get(logFilePrefix);
52 reader = new BufferedReader(new FileReader(getLogFile(logDirectory, logFilePrefix)));
53 while (reader.readLine() != null) {
56 readersMap.put(logFilePrefix, reader);
62 * Find the most recently modified log file with a name starting with the specified prefix.
65 * the root folder storing log files
66 * @param filenamePrefix
67 * the log file name prefix
68 * @return the most recently created log file
70 * if an I/O error occurs
72 public File getLogFile(String logDirectory, String filenamePrefix) throws IOException {
73 Path cachedLog = cachedLogMap.get(filenamePrefix);
75 if (cachedLog == null) {
76 Optional<Path> latestFilePath = Files.list(Paths.get(logDirectory))
77 .filter(f -> Files.isDirectory(f) == false && f.getFileName().toString().startsWith(filenamePrefix))
78 .max(Comparator.comparingLong(f -> f.toFile().lastModified()));
79 if (latestFilePath.isPresent()) {
80 cachedLog = latestFilePath.get();
82 throw new IOException("No validation log files were found!");
85 return cachedLog.toFile();
89 * Wait for and read new log entries.
91 * @return new lines appended to the log file
93 * If an I/O error occurs
95 public String getNewLines() throws IOException {
96 StopWatch stopwatch = new StopWatch();
99 while (!cachedReader.ready()) {
100 if (stopwatch.getTime() > TimeUnit.SECONDS.toMillis(30)) {
101 Assert.fail("Test took too long");
106 StringBuilder lines = new StringBuilder();
108 while ((line = cachedReader.readLine()) != null) {
109 lines.append(line).append(System.lineSeparator());
111 return lines.toString();