io.jboot.config.server.ConfigFileScanner Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of jboot Show documentation
Jboot is a similar SpringCloud project base on JFinal, Dubbo and Undertow.
There is a newer version: 4.1.9
/**
 * Copyright (c) 2015-2018, Michael Yang 杨福海 ([email protected]).
 * 
 * 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.
 */
package io.jboot.config.server;

import com.jfinal.kit.LogKit;
import com.jfinal.kit.StrKit;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Timer;
import java.util.TimerTask;

/**
 * 定时扫描本地配置文件，提供给远程读取
 *
 * @author Michael Yang 杨福海 （[email protected]）
 * @version V1.0
 * @Package io.jboot.config
 */
public abstract class ConfigFileScanner {

    public static final String ACTION_ADD = "add";
    public static final String ACTION_DELETE = "delete";
    public static final String ACTION_UPDATE = "update";

    private Timer timer;
    private TimerTask task;
    private String rootDir;
    private int interval;
    private boolean running = false;

    private final Map preScan = new HashMap();
    private final Map curScan = new HashMap();

    public ConfigFileScanner(String rootDir, int interval) {
        if (StrKit.isBlank(rootDir))
            throw new IllegalArgumentException("The parameter rootDir can not be blank.");
        this.rootDir = rootDir;
        if (interval <= 0)
            throw new IllegalArgumentException("The parameter interval must more than zero.");
        this.interval = interval;
    }

    public abstract void onChange(String action, String file);

    private void working() {
        if (!rootDir.contains(";")) {
            scan(new File(rootDir));
        } else {
            String[] paths = rootDir.split(";");
            for (String path : paths) {
                scan(new File(path));
            }
        }

        compare();

        preScan.clear();
        preScan.putAll(curScan);
        curScan.clear();
    }

    private void scan(File file) {
        if (file == null || !file.exists())
            return;

        if (file.isFile() && file.getName().toLowerCase().endsWith(".properties")) {
            try {
                curScan.put(file.getCanonicalPath(), new TimeSize(file));
            } catch (IOException e) {
                LogKit.error(e.getMessage(), e);
            }
        } else if (file.isDirectory()) {
            File[] fs = file.listFiles();
            if (fs != null && fs.length > 0) {
                for (File f : fs) {
                    scan(f);
                }
            }
        }
    }

    private void compare() {

        for (Map.Entry entry : curScan.entrySet()) {
            if (preScan.get(entry.getKey()) == null)
                onChange(ACTION_ADD, entry.getKey());
        }

        for (Map.Entry entry : preScan.entrySet()) {
            if (curScan.get(entry.getKey()) == null)
                onChange(ACTION_DELETE, entry.getKey());
        }


        for (Map.Entry entry : curScan.entrySet()) {
            TimeSize pre = preScan.get(entry.getKey());
            if (pre != null && !pre.equals(entry.getValue()))
                onChange(ACTION_UPDATE, entry.getKey());
        }

    }

    public void start() {
        if (!running) {
            timer = new Timer("Jboot-Config-Scanner", true);
            task = new TimerTask() {
                public void run() {
                    working();
                }
            };
            timer.schedule(task, 0, 1010L * interval);
            running = true;
        }
    }

    public void stop() {
        if (running) {
            timer.cancel();
            task.cancel();
            running = false;
        }
    }
}