sdis1819-t2g03

package protocols;

import server.Peer;

import sockets.Socket;

import utils.Utils;

import java.io.IOException;

import java.util.Random;

import java.util.concurrent.TimeUnit;

import chunk.Chunk;

import handlers.Message;

import static chunk.FolderManager.makeDir;

import static chunk.FolderManager.backupFile;

public class Backup implements Runnable {

    private Peer peer;

    private Message req;

    private byte[] data;

    private int replication;

    private String fileID;

    private int chunkNo;

    private String version;

    private int senderID;

    private static int delayLimit;

    public Backup(Peer peer, Message req) {

        this.peer = peer;

        this.req = req;

        System.out.println("BACKUP begins!");

    }

    @Override

    public void run() {

        version = req.getVersion();

        senderID = req.getSenderID();

        fileID = req.getFileID();

        chunkNo = req.getChunkNo();

        replication = req.getReplicationDegree();

        delayLimit = 300;

        if (senderID == peer.getID()) {

                System.out.println("Ignoring backup of own files");

            return;

        }

        data = req.getBody();

        String chunkPathname = peer.getPath("chunks") + "/" + fileID;

        makeDir(peer.getPath("chunks") + "/" + fileID);

        boolean success = false;

        try {

            success = backupFile(Integer.toString(chunkNo), chunkPathname, data);

            //save to database

            peer.addChunkToFileManager(new Chunk(fileID, chunkNo, replication, data.length));

        } catch (IOException e) {

            e.printStackTrace();

        }

        if (! success) {

                System.out.println("Memory overflow");

        } else {

            sendMsgStored();

        }

        System.out.println("Finished backup!");

    }

    private void sendMsgStored() {

        String[] args = {version, Integer.toString(peer.getID()),fileID, Integer.toString(chunkNo)};

        Message msg = new Message(Utils.MessageType.STORED, args);

        Random random = new Random();

        peer.sendLateMsg(Socket.SocketType.MC, msg, random.nextInt(delayLimit + 1), TimeUnit.MILLISECONDS);

    }

}