com.yhzdys.myosotis.processor.SnapshotProcessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of myosotis-client Show documentation
There is a newer version: 1.3.4
package com.yhzdys.myosotis.processor;

import com.yhzdys.myosotis.constant.SysConst;
import com.yhzdys.myosotis.entity.MyosotisConfig;
import com.yhzdys.myosotis.entity.MyosotisEvent;
import com.yhzdys.myosotis.misc.JsonUtil;
import com.yhzdys.myosotis.misc.LoggerFactory;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.Collections;
import java.util.List;

public final class SnapshotProcessor implements Processor {

    public final String sn_dir = SysConst.myosotis_dir + SysConst.separator + "snapshot" + SysConst.separator + "%s";
    public final String sn_file = sn_dir + SysConst.separator + "%s" + ".snapshot";

    private final boolean enable;

    public SnapshotProcessor(boolean enable) {
        this.enable = enable;
    }

    @Override
    public void init(String namespace) {
        if (!enable) {
            return;
        }
        File dir = new File(String.format(sn_dir, namespace));
        boolean b = dir.mkdirs();
    }

    @Override
    public List fetchEvents() {
        return Collections.emptyList();
    }

    @Override
    public MyosotisConfig getConfig(String namespace, String configKey) {
        if (!enable) {
            return null;
        }
        File file = new File(String.format(sn_file, namespace, configKey));
        if (!file.exists()) {
            return null;
        }
        try {
            return JsonUtil.toObject(this.readFile(file), MyosotisConfig.class);
        } catch (Exception e) {
            LoggerFactory.getLogger().error("Read snapshot config failed", e);
            return null;
        }
    }

    @Override
    public List getConfigs(String namespace) {
        return Collections.emptyList();
    }

    @Override
    public void save(MyosotisConfig config) {
        if (!enable) {
            return;
        }
        this.saveFile(JsonUtil.toString(config), String.format(sn_file, config.getNamespace(), config.getConfigKey()));
    }

    private String readFile(File file) {
        try (FileInputStream fis = new FileInputStream(file); ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
            byte[] buffer = new byte[128];
            int length;
            while ((length = fis.read(buffer)) != -1) {
                bos.write(buffer, 0, length);
            }
            return bos.toString("UTF-8").trim();
        } catch (Exception e) {
            LoggerFactory.getLogger().error("Read snapshot file failed", e);
            return null;
        }
    }

    private void saveFile(String data, String path) {
        File file = new File(path);
        Writer writer = null;
        try {
            if (!file.exists() && !file.createNewFile()) {
                LoggerFactory.getLogger().warn("Create snapshot file failed");
                return;
            }
            writer = new OutputStreamWriter(Files.newOutputStream(file.toPath()), StandardCharsets.UTF_8);
            writer.write(data);
        } catch (Exception e) {
            LoggerFactory.getLogger().error("Save file failed", e);
        } finally {
            if (writer != null) {
                try {
                    writer.close();
                } catch (Exception ignored) {
                }
            }
        }
    }
}