257c356c11062e16052a582d4b205ab671659174
[dcaegen2/collectors/datafile.git] /
1 /*-
2  * ============LICENSE_START======================================================================
3  * Copyright (C) 2019 Nordix Foundation. All rights reserved.
4  * ===============================================================================================
5  * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
6  * in compliance with the License. You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software distributed under the License
11  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
12  * or implied. See the License for the specific language governing permissions and limitations under
13  * the License.
14  * ============LICENSE_END========================================================================
15  */
16
17 package org.onap.dcaegen2.collectors.datafile.service;
18
19 import java.nio.file.Path;
20 import java.time.Instant;
21 import java.util.Collections;
22 import java.util.HashMap;
23 import java.util.Iterator;
24 import java.util.Map;
25
26 /**
27  * A cache of all files that already has been published. Key is the local file path and the value is a time stamp, when
28  * the key was last used.
29  */
30 public class PublishedFileCache {
31     private final Map<Path, Instant> publishedFiles = Collections.synchronizedMap(new HashMap<Path, Instant>());
32
33     /**
34      * Adds a file to the cache.
35      *
36      * @param path the name of the file to add.
37      * @return <code>null</code> if the file is not already in the cache.
38      */
39     public Instant put(Path path) {
40         return publishedFiles.put(path, Instant.now());
41     }
42
43     /**
44      * Removes a file from the cache.
45      *
46      * @param localFileName name of the file to remove.
47      */
48     public void remove(Path localFileName) {
49         publishedFiles.remove(localFileName);
50     }
51
52     /**
53      * Removes files 24 hours older than the given instant.
54      *
55      * @param now the instant will determine which files that will be purged.
56      */
57     public void purge(Instant now) {
58         for (Iterator<Map.Entry<Path, Instant>> it = publishedFiles.entrySet().iterator(); it.hasNext();) {
59             Map.Entry<Path, Instant> pair = it.next();
60             if (isCachedPublishedFileOutdated(now, pair.getValue())) {
61                 it.remove();
62             }
63         }
64     }
65
66     int size() {
67         return publishedFiles.size();
68     }
69
70     private boolean isCachedPublishedFileOutdated(Instant now, Instant then) {
71         final int timeToKeepInfoInSeconds = 60 * 60 * 24;
72         return now.getEpochSecond() - then.getEpochSecond() > timeToKeepInfoInSeconds;
73     }
74 }