gr.iti.mklab.sfc.storages.FlatFileStorage Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mklab-stream-manager Show documentation

Monitors a set of social streams (e.g. Twitter status updates) and collects the incoming content.

The newest version!

package gr.iti.mklab.sfc.storages;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;

import gr.iti.mklab.framework.common.domain.config.Configuration;
import gr.iti.mklab.framework.common.domain.Item;

/**
 * Class for storing items to a flat file
 * @author manosetro - [email protected]
 *
 */
public class FlatFileStorage implements Storage {	
	
	private static String NAME = "name";
	private static String STORAGE_FILE = "file";
	
	private String storageName = "FlatFile";
	private String fileToStore;
	
	private File storageDirectory;
	private PrintWriter out = null;
	
	long items = 0;
	
	public FlatFileStorage(Configuration config) {
		this.fileToStore = config.getParameter(FlatFileStorage.NAME);
		
		String storage = config.getParameter(FlatFileStorage.STORAGE_FILE);
		this.storageDirectory = new File(storage);
	}
	
	
	public FlatFileStorage(String storageDirectory) {
		this.storageDirectory = new File(storageDirectory);
		
		this.fileToStore = "/items.";
	}
	
	@Override
	public void store(Item item){
		items++;
		if (out != null) {
			out.println(item.toString());
			out.flush();
			
		}
		if(items%1000==0) {
			open();	
		}
	
	}

	
	@Override
	public boolean delete(String id) throws IOException {
		// cannot delete rows from flat files
		return false;
	}
	

	@Override
	public boolean open(){
		File storeFile = new File(storageDirectory, fileToStore+System.currentTimeMillis());
		try {
			out = new PrintWriter(new BufferedWriter(new FileWriter(storeFile, false)));
			return true;
		} catch (IOException e) {
			return false;
		}
	}

	@Override
	public void close() {
		out.close();
	}

	@Override
	public boolean checkStatus() {
		return true;
	}
	
	@Override
	public String getStorageName(){
		return this.storageName;
	}
	
}