sdis1819-t4g11

package peer.protocols.restore;

import java.io.File;

import java.io.IOException;

import java.util.concurrent.BlockingQueue;

import chunk.Chunk;

import disk.ChunkManagement;

import message.Message;

import peer.Peer;

import peer.channels.Sender;

import peer.channels.SenderUDP;

/**

 * Restore

 */

public class Restore implements Runnable {

        private Peer peer;

        private String filePath;

        private File file;

        public Restore(Peer peer, String filePath) {

                this.peer = peer;

                this.filePath = filePath;

                file = new File(filePath);

        }

        public void sendGetChunk(int chunkNo) {

                Message message = Message.parseGetChunkMessage(Chunk.generateFileId(file), chunkNo, peer);

                try {

                        peer.sendToMc(message);

                } catch (IOException e) {

                        e.printStackTrace();

                }

        }

        public void restore() {

                File originalFile = new File(filePath);

                if (!originalFile.exists()) {

                        System.out.println("file not found");

                        return;

                }

                int chunksNo = Chunk.getNumberOfFileChunks(originalFile);

                Chunk[] chunks = new Chunk[chunksNo];

                String fileId = Chunk.generateFileId(originalFile);

                for (int i = 0; i < chunksNo; i++) {

                        sendGetChunk(i + 1);

                        BlockingQueue<Chunk> queue = ChunkManagement.getInstance().getRestoreChunks();

                        try {

                                System.out.println("Waiting for chunk " + (i+1));

                                Chunk chunk = queue.take();

                                if (chunk.getFileID().equals(fileId)){

                                        chunks[i] = chunk;

                                        System.out.println("received chunk " + chunk.getChunkNo());

                                }

                                else {

                                        queue.put(chunk);

                                }

                        } catch (InterruptedException e) {

                                e.printStackTrace();

                        }

                }

                peer.getDisk().restoreFile(chunks,originalFile.getName());

        }

        @Override

        public void run() {

                restore();

        }

}