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=========================================================
21 package org.onap.aai.babel.logging;
23 import java.io.BufferedReader;
25 import java.io.FileReader;
26 import java.io.IOException;
27 import java.nio.file.Files;
28 import java.nio.file.Path;
29 import java.nio.file.Paths;
30 import java.util.Comparator;
31 import java.util.HashMap;
33 import java.util.Optional;
34 import java.util.concurrent.TimeUnit;
35 import org.apache.commons.lang3.time.StopWatch;
36 import org.junit.Assert;
38 public class LogReader {
40 private Map<String, Path> cachedLogMap = new HashMap<>();
41 private Map<String, BufferedReader> readersMap = new HashMap<>();
42 private BufferedReader cachedReader;
44 public LogReader(String logDirectory, String logFilePrefix) throws IOException {
45 cachedReader = getReader(logDirectory, logFilePrefix);
48 private BufferedReader getReader(String logDirectory, String logFilePrefix) throws IOException {
49 BufferedReader reader = readersMap.get(logFilePrefix);
51 reader = new BufferedReader(new FileReader(getLogFile(logDirectory, logFilePrefix)));
52 while (reader.readLine() != null) {
55 readersMap.put(logFilePrefix, reader);
62 * @return the most recently created log file.
65 public File getLogFile(String logDirectory, String filenamePrefix) throws IOException {
66 Path cachedLog = cachedLogMap.get(filenamePrefix);
68 if (cachedLog == null) {
69 Optional<Path> latestFilePath = Files.list(Paths.get(logDirectory))
70 .filter(f -> Files.isDirectory(f) == false && f.getFileName().toString().startsWith(filenamePrefix))
71 .max(Comparator.comparingLong(f -> f.toFile().lastModified()));
72 if (latestFilePath.isPresent()) {
73 cachedLog = latestFilePath.get();
75 throw new IOException("No validation log files were found!");
78 return cachedLog.toFile();
82 * @return new lines appended to the log file
85 public String getNewLines() throws IOException {
86 StopWatch stopwatch = new StopWatch();
89 while (!cachedReader.ready()) {
90 if (stopwatch.getTime() > TimeUnit.SECONDS.toMillis(30)) {
91 Assert.fail("Test took too long");
96 StringBuilder lines = new StringBuilder();
98 while ((line = cachedReader.readLine()) != null) {
99 lines.append(line).append(System.lineSeparator());
101 return lines.toString();