com.tqdev.metrics.influxdb.InfluxDbFileReporter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metrics-influxdb Show documentation
Support for publishing the metrics from instrumented components to InfluxDB.
There is a newer version: 0.9.6
/* Copyright (C) 2017 Maurits van der Schee
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package com.tqdev.metrics.influxdb;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FilenameFilter;
import java.io.IOException;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.zip.GZIPOutputStream;

import com.tqdev.metrics.core.MetricRegistry;

/**
 * The InfluxDbFileReporter class reports values in the metric registry to
 * InfluxDB readable files.
 */
public class InfluxDbFileReporter extends InfluxDbReporter {

	/** The metric path. */
	protected final String metricPath;

	/** The max file count. */
	protected final int maxFileCount;

	/** The date format. */
	protected final String dateFormat;

	/**
	 * Instantiates a new InfluxDB file reporter.
	 *
	 * @param registry
	 *            the registry
	 * @param instanceName
	 *            the instance name
	 * @param metricPath
	 *            the metric path
	 * @param dateFormat
	 *            the date format
	 * @param maxFileCount
	 *            the max file count
	 */
	public InfluxDbFileReporter(MetricRegistry registry, String instanceName, String metricPath, String dateFormat,
			int maxFileCount) {
		super(registry, instanceName);
		this.metricPath = metricPath;
		this.dateFormat = dateFormat;
		this.maxFileCount = maxFileCount;
	}

	/**
	 * Instantiates a new InfluxDB file reporter.
	 *
	 * @param registry
	 *            the registry
	 * @param instanceName
	 *            the instance name
	 * @param metricPath
	 *            the metric path
	 * @param dateFormat
	 *            the date format
	 * @param maxFileCount
	 *            the max file count
	 * @param intervalInSeconds
	 *            the interval in seconds
	 */
	public InfluxDbFileReporter(MetricRegistry registry, String instanceName, String metricPath, String dateFormat,
			int maxFileCount, int intervalInSeconds) {
		super(registry, instanceName);
		this.metricPath = metricPath;
		this.dateFormat = dateFormat;
		this.maxFileCount = maxFileCount;
		run(intervalInSeconds);
	}

	/**
	 * Report.
	 *
	 * @return true, if successful
	 */
	@Override
	public boolean report() {
		if (!registry.isEnabled()) {
			return true;
		}
		File dir = new File(metricPath);
		if (!dir.exists()) {
			if (!dir.mkdir()) {
				return false;
			}
		}
		DateFormat formatter = new SimpleDateFormat(dateFormat);
		String filename = metricPath + "/" + formatter.format(new Date(registry.getMillis())) + ".txt";
		try {
			FileOutputStream fos = new FileOutputStream(filename, true);
			BufferedOutputStream out = new BufferedOutputStream(fos, 8192);
			write(out);
			out.close();
			compress(filename);
			remove(maxFileCount);
		} catch (IOException e) {
			// TODO: log
			return false;
		}
		return true;
	}

	/**
	 * Compress.
	 *
	 * @param filename
	 *            the filename
	 * @throws IOException
	 *             Signals that an I/O exception has occurred.
	 */
	private void compress(String filename) throws IOException {
		File current = new File(filename);
		File dir = new File(metricPath);
		FilenameFilter textFileFilter = (f, s) -> s.endsWith(".txt");
		File[] directoryListing = dir.listFiles(textFileFilter);
		if (directoryListing != null) {
			for (File file : directoryListing) {
				if (file.getCanonicalPath() != current.getCanonicalPath()) {
					try (FileOutputStream fos = new FileOutputStream(file.getPath() + ".gz");
							GZIPOutputStream gzos = new GZIPOutputStream(fos)) {
						byte[] buffer = new byte[8192];
						int length;
						try (FileInputStream fis = new FileInputStream(file.getPath())) {
							while ((length = fis.read(buffer)) > 0) {
								gzos.write(buffer, 0, length);
							}
						}
					}
					file.delete();
				}
			}
		} else {
			throw new IOException("Directory not listable: " + metricPath);
		}
	}

	/**
	 * Removes the.
	 *
	 * @param maxFileCount
	 *            the max file count
	 * @throws IOException
	 *             Signals that an I/O exception has occurred.
	 */
	private void remove(int maxFileCount) throws IOException {
		File dir = new File(metricPath);
		FilenameFilter gzipFileFilter = (f, s) -> s.endsWith(".gz");
		File[] directoryListing = dir.listFiles(gzipFileFilter);
		if (directoryListing != null) {
			Arrays.sort(directoryListing);
			for (int i = 0; i < directoryListing.length - maxFileCount; i++) {
				directoryListing[i].delete();
			}
		} else {
			throw new IOException("Directory not listable: " + metricPath);
		}
	}

}