BackupQueue.java

package com.dedupeer.backup;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.LinkedBlockingQueue;

import com.dedupeer.navigation.DFile;
import com.dedupeer.utils.FileUtils;


/**
 * @author Paulo Fernando (pf@paulofernando.net.br)
 */
public class BackupQueue extends Thread {
	
	private static BackupQueue instance;
	private FileUtils fileUtils = new FileUtils();
	
	/**
	 * Map to store the file name to use to deduplicate the other file. 
	 */
	private Map<String, String> deduplicateMap = new HashMap<String, String>();
	
	/**
	 * Map with a file path as key and the backup as value
	 */
	private LinkedBlockingQueue<DFile> backupQueue = new LinkedBlockingQueue<DFile>();
	
	public static BackupQueue getInstance() {
		if(instance == null) {
			instance = new BackupQueue();			
			instance.start();
		}
		return instance;
	}
	
	/**
	 * Adds a DFile in the queue
	 * @param dFile The DFile to add in the queue
	 */
	public void addBackup(DFile dFile) {
		backupQueue.add(dFile);
	}
	
	/**
	 * Adds in the backup queue and informs the filename to compare the chunks
	 */
	public void addBackup(DFile dFile, String deduplicateWith) {
		deduplicateMap.put(dFile.getName(), deduplicateWith);
		addBackup(dFile);		
	}

	@SuppressWarnings("static-access")
	@Override
	public void run() {
		for(;;) {
			try {
				
				/* Blocks the Thread until the queue has a DFile */
				DFile currentBackup = backupQueue.take();
				
				if(deduplicateMap.containsKey(currentBackup.getName())) {
					currentBackup.deduplicateABigFileByThrift(deduplicateMap.get(currentBackup.getName()), 
							Integer.parseInt(fileUtils.getPropertiesLoader().getProperties().getProperty("default.chunk.size")) * 
							Integer.parseInt(fileUtils.getPropertiesLoader().getProperties().getProperty("chunks.to.load")));
					deduplicateMap.remove(currentBackup.getName());
				} else {
					currentBackup.store();
				}				
			} catch (InterruptedException e) {
				e.printStackTrace();
			}
		}
	}
}