summaryrefslogtreecommitdiffstats
path: root/artifactbroker/plugins/reception-plugins/src/main/java/org/onap/policy/distribution/reception/handling/sdc/FileSystemReceptionHandler.java
blob: 941cdd610d46ba9f45414ea6ccd1c8c3b30e1e6f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/*-
 * ============LICENSE_START=======================================================
 *  Copyright (C) 2018 Intel Corp. All rights reserved.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 * ============LICENSE_END=========================================================
 */

package org.onap.policy.distribution.reception.handling.sdc;

import static java.nio.file.StandardWatchEventKinds.ENTRY_CREATE;

import java.io.File;
import java.io.IOException;
import java.nio.file.FileSystems;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.WatchEvent;
import java.nio.file.WatchKey;
import java.nio.file.WatchService;

import org.onap.policy.common.logging.flexlogger.FlexLogger;
import org.onap.policy.common.logging.flexlogger.Logger;

import org.onap.policy.common.parameters.ParameterService;
import org.onap.policy.distribution.model.Csar;
import org.onap.policy.distribution.reception.decoding.PolicyDecodingException;
import org.onap.policy.distribution.reception.handling.AbstractReceptionHandler;

/**
 * Handles reception of inputs from File System which can be used to decode policies.
 */
public class FileSystemReceptionHandler extends AbstractReceptionHandler {
    private boolean running = false;
    private static final Logger LOGGER = FlexLogger.getLogger(FileSystemReceptionHandler.class);

    @Override
    protected void initializeReception(final String parameterGroupName) {
        LOGGER.debug("FileSystemReceptionHandler init...");
        try {
            final FileSystemReceptionHandlerConfigurationParameterGroup handlerParameters =
                                    ParameterService.get(parameterGroupName);
            main(handlerParameters.getWatchPath());
        } catch (final Exception ex) {
            LOGGER.error(ex);
        }
        running = false;
        LOGGER.debug("FileSystemReceptionHandler main loop exited...");
    }

    @Override
    public void destroy() {
        // Tear down subscription etc
        running = false;
    }

    public boolean isRunning() {
        return running;
    }

    /**
     * Main entry point.
     * 
     * @param watchPath Path to watch
     */
    public void main(String watchPath) throws IOException {
        try (final WatchService watcher = FileSystems.getDefault().newWatchService()) {
            final Path dir = Paths.get(watchPath);
            dir.register(watcher, ENTRY_CREATE);
            LOGGER.debug("Watch Service registered for dir: " + dir.getFileName());
            startMainLoop(watcher, dir);
        } catch (final InterruptedException ex) {
            LOGGER.debug(ex);
            Thread.currentThread().interrupt();
        }
    }

    @SuppressWarnings("unchecked")
    protected void startMainLoop(WatchService watcher, Path dir) throws InterruptedException {
        WatchKey key;
        running = true;
        while (running) {
            key = watcher.take();

            for (final WatchEvent<?> event : key.pollEvents()) {
                final WatchEvent<Path> ev = (WatchEvent<Path>) event;
                final Path fileName = ev.context();
                LOGGER.debug("new CSAR found: " + fileName);
                createPolicyInputAndCallHandler(dir.toString() + File.separator + fileName.toString());
                LOGGER.debug("CSAR complete: " + fileName);
            }
            final boolean valid = key.reset();
            if (!valid) {
                LOGGER.error("Watch key no longer valid!");
                break;
            }
        }
    }

    protected void createPolicyInputAndCallHandler(final String fileName) {
        try {
            final Csar csarObject = new Csar(fileName);
            inputReceived(csarObject);
        } catch (final PolicyDecodingException ex) {
            LOGGER.error(ex);
        }
    }
}