sdis1819-t2g03

package handlers;

import server.Peer;

import sockets.Socket;

import utils.Utils;

import java.io.IOException;

import java.util.concurrent.atomic.AtomicIntegerArray;

import chunk.Chunk;

import protocols.BackupChunk;

public class BackupChunkHandler implements Runnable {

    private final String protocolVersion;

    private Peer parentPeer;

    private Chunk chunk;

    private AtomicIntegerArray chunkReplication;

    private int retries = 5;

    public BackupChunkHandler(BackupChunk backupInitiator, Chunk chunk) {

        this.chunk = chunk;

        this.parentPeer = backupInitiator.getPeer();

        this.protocolVersion = backupInitiator.getVersion();

        this.chunkReplication = parentPeer.getPeerInformation().getChunkReplication(chunk.getFileID());

    }

    BackupChunkHandler(Peer parentPeer, Chunk chunk) {

        this.chunk = chunk;

        this.parentPeer = parentPeer;

        this.protocolVersion = Utils.VERSION;

        this.chunkReplication = null;

    }

    @Override

    public void run() {

        int waitTime = 1000;

        Message msg = generatePutChunkMsg(chunk, protocolVersion);

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

            if (isDesiredReplicationDegree()) {

                    System.out.println("Achieved desired replication at i=" + i);

                break;

            }

            try {

                parentPeer.sendMsg(Socket.SocketType.MDB, msg);

            } catch (IOException e) {

                    System.out.println(e.getMessage());

            }

            sleep(waitTime);

            waitTime *= 2;

        }

    }

    protected boolean isDesiredReplicationDegree() {

        return chunkReplication != null && chunkReplication.get(chunk.getChunkNo()) >= chunk.getReplication();

    }

    private void sleep(int waitTime) {

        try {

            Thread.sleep(waitTime);

        } catch (InterruptedException e) {

            e.printStackTrace();

        }

    }

    private Message generatePutChunkMsg(Chunk chunk, String protocolVersion) {

        String[] args = { protocolVersion, Integer.toString(parentPeer.getID()), chunk.getFileID(), Integer.toString(chunk.getChunkNo()), Integer.toString(chunk.getReplication())

        };

        return new Message(Utils.MessageType.PUTCHUNK, args, chunk.getData());

    }

}