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 Peer peer;

        private Chunk chunk;

        private final String protocolVersion;

        private AtomicIntegerArray replication;

        private int retries = 5;

        public BackupChunkHandler(BackupChunk backupChunk, Chunk chunk) {

                this.chunk = chunk;

                this.peer = backupChunk.getPeer();

                this.protocolVersion = backupChunk.getVersion();

                this.replication = peer.getPeerInformation().getChunkReplication(chunk.getFileID());

        }

        BackupChunkHandler(Peer peer, Chunk chunk) {

                this.chunk = chunk;

                this.peer = peer;

                this.protocolVersion = Utils.VERSION;

                this.replication = null;

        }

        @Override

        public void run() {

                int waitTime = 1000;

                Message msg = generatePutChunkMsg(chunk, protocolVersion);

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

                        if (isReplicationtheWantedOne() == true) {

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

                                break;

                        }

                        try {

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

                        } catch (IOException e1) {

                                e1.printStackTrace();

                        }

                        try {

                                sleep(waitTime);

                        } catch (InterruptedException e) {

                                e.printStackTrace();

                        }

                        waitTime *= 2;

                }

        }

        protected boolean isReplicationtheWantedOne() {

                if(replication != null && replication.get(chunk.getChunkNo()) >= chunk.getReplication())

                        return true;

                else return false;

        }

        private void sleep(int waitTime) throws InterruptedException {

                Thread.sleep(waitTime);

        }

        private Message generatePutChunkMsg(Chunk chunk, String protocolVersion) {

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

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

        }

        // Getters and Setters

        public Peer getPeer() {

                return peer;

        }

        public Chunk getChunk() {

                return chunk;

        }

        public String getProtocolVersion() {

                return protocolVersion;

        }

        public AtomicIntegerArray getReplication() {

                return replication;

        }

        public int getRetries() {

                return retries;

        }

        public void setPeer(Peer peer) {

                this.peer = peer;

        }

        public void setChunk(Chunk chunk) {

                this.chunk = chunk;

        }

        public void setReplication(AtomicIntegerArray replication) {

                this.replication = replication;

        }

        public void setRetries(int retries) {

                this.retries = retries;

        }

}