com.wordnik.util.RotatingFileWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongo-admin-utils_2.10 Show documentation

mongo-admin-utils

The newest version!

// Copyright (C) 2010  Wordnik, Inc.
//
// This program is free software: you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or (at your 
// option) any later version.  This program is distributed in the hope that it
// will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty 
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser 
// General Public License for more details.  You should have received a copy 
// of the GNU Lesser General Public License along with this program.  If not,
// see .

package com.wordnik.util;

import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;

public class RotatingFileWriter extends AbstractFileWriter {
	Writer writer;

	public RotatingFileWriter(String filePrefix, String destinationDirectory, String extension, long maxFileSizeInBytes, boolean compressOnRotate){
		super(filePrefix, destinationDirectory, extension, maxFileSizeInBytes, compressOnRotate);
	}

	public void write(String stringToWrite) throws IOException {
		getWriter().write(stringToWrite + "\n");
		int sz = stringToWrite.length();
		currentFileSize += sz;
		totalBytesWritten += sz;
		if(currentFileSize > maxFileSizeInBytes){
			rotateFile();
		}
	}
	
	public void close() throws IOException {
		rotateFile();
	}

	Writer getWriter() throws IOException {
		if(!hasWriter()){
			openFile(filePrefix);
		}
		return writer;
	}
	
	boolean hasWriter(){
		if(writer == null){
			return false;
		}
		return true;
	}
	
	public void closeWriter() throws IOException {
		if(hasWriter()){
			writer.close();
		}
	}

	Writer openFile(String prefix) throws IOException {
		writer = new OutputStreamWriter(getOutputStream(prefix), "UTF-8");
		currentFileSize = 0;
		return writer;
	}
	
	void resetWriter(){
		if(hasWriter()){
			writer = null;
		}
	}
}