com.yoncabt.ebr.logger.fs.FileSystemReportLogger.java Source code

Introduction

Here is the source code for com.yoncabt.ebr.logger.fs.FileSystemReportLogger.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.yoncabt.ebr.logger.fs;

import com.yoncabt.abys.core.util.EBRConf;
import com.yoncabt.abys.core.util.EBRParams;
import com.yoncabt.ebr.ReportOutputFormat;
import com.yoncabt.ebr.ReportRequest;
import com.yoncabt.ebr.logger.ReportLogger;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.util.Map;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;
import org.apache.commons.io.IOUtils;
import org.json.JSONObject;

/**
 *
 * @author myururdurmaz
 */
public class FileSystemReportLogger implements ReportLogger {

    @Override
    public void logReport(ReportRequest request, ReportOutputFormat outputFormat, InputStream reportData)
            throws IOException {
        String uuid = request.getUuid();
        Map<String, Object> reportParams = request.getReportParams();
        File saveDir = new File(EBRConf.INSTANCE.getValue(EBRParams.REPORT_LOGGER_FSLOGGER_PATH, "/tmp"));
        saveDir.mkdirs();
        boolean compress = EBRConf.INSTANCE.getValue(EBRParams.REPORT_LOGGER_FSLOGGER_COMPRESS, true);
        OutputStream osReport;
        OutputStream osParams;
        if (compress) {
            osReport = new GZIPOutputStream(new FileOutputStream(new File(saveDir, uuid + ".gz")));
            osParams = new GZIPOutputStream(new FileOutputStream(new File(saveDir, uuid + ".json.gz")));
        } else {
            osReport = new FileOutputStream(new File(saveDir, uuid));
            osParams = new FileOutputStream(new File(saveDir, uuid + ".json"));
        }
        IOUtils.copy(reportData, osReport);
        JSONObject jo = new JSONObject(reportParams);
        try (OutputStreamWriter osw = new OutputStreamWriter(osParams, "utf-8")) {
            jo.write(osw);
        }
        osReport.close();
        osParams.close();
    }

    @Override
    public byte[] getReportData(String uuid) throws IOException {
        //burada sktrma zellii akken kpatlm olabilir diye kontrol yapyorum
        File saveDir = new File(EBRConf.INSTANCE.getValue(EBRParams.REPORT_LOGGER_FSLOGGER_PATH, "/tmp"));
        File reportFile = new File(saveDir, uuid + ".gz");
        if (reportFile.exists()) {
            try (FileInputStream fis = new FileInputStream(reportFile);
                    GZIPInputStream gzis = new GZIPInputStream(fis);) {
                return IOUtils.toByteArray(gzis);
            }
        } else {
            reportFile = new File(saveDir, uuid);
            try (FileInputStream fis = new FileInputStream(reportFile)) {
                return IOUtils.toByteArray(fis);
            }
        }
    }

}