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.HashMap;
32 import java.util.Optional;
33 import java.util.concurrent.TimeUnit;
34 import org.apache.commons.lang.time.StopWatch;
35 import org.junit.Assert;
37 public class LogReader {
39 private Map<String, Path> cachedLogMap = new HashMap<>();
40 private Map<String, BufferedReader> readersMap = new HashMap<>();
41 private BufferedReader cachedReader;
43 public LogReader(String logDirectory, String logFilePrefix) throws IOException {
44 cachedReader = getReader(logDirectory, logFilePrefix);
47 private BufferedReader getReader(String logDirectory, String logFilePrefix) throws IOException {
48 BufferedReader reader = readersMap.get(logFilePrefix);
50 reader = new BufferedReader(new FileReader(getLogFile(logDirectory, logFilePrefix)));
51 while (reader.readLine() != null) {
54 readersMap.put(logFilePrefix, reader);
61 * @return the most recently created log file.
64 public File getLogFile(String logDirectory, String filenamePrefix) throws IOException {
65 Path cachedLog = cachedLogMap.get(filenamePrefix);
67 if (cachedLog == null) {
68 Optional<Path> latestFilePath = Files.list(Paths.get(logDirectory))
69 .filter(f -> Files.isDirectory(f) == false && f.getFileName().toString().startsWith(filenamePrefix))
70 .max((f1, f2) -> (int) (f1.toFile().lastModified() - f2.toFile().lastModified()));
71 if (latestFilePath.isPresent()) {
72 cachedLog = latestFilePath.get();
74 throw new IOException("No validation log files were found!");
77 return cachedLog.toFile();
81 * @return new lines appended to the log file
84 public String getNewLines() throws IOException {
85 StopWatch stopwatch = new StopWatch();
88 while (!cachedReader.ready()) {
89 if (stopwatch.getTime() > TimeUnit.SECONDS.toMillis(30)) {
90 Assert.fail("Test took too long");
95 StringBuilder lines = new StringBuilder();
97 while ((line = cachedReader.readLine()) != null) {
98 lines.append(line).append(System.lineSeparator());
100 return lines.toString();