sdis1819-t2g03

package handlers;

import protocols.*;

import server.Peer;

import java.util.Random;

import java.util.concurrent.*;

import chunk.Chunk;

import chunk.FileManager;

public class Handler implements Runnable {

        private Peer parentPeer;

        private Chunk peerInformation;

        private BlockingQueue<Message> msgQueue;

        private ScheduledExecutorService executor;

        private int delayLimit;

        private Random random;

        public Handler(Peer parentPeer) {

                this.parentPeer = parentPeer;

                this.peerInformation = parentPeer.getPeerInformation();

                msgQueue = new LinkedBlockingQueue<>();

                executor = Executors.newScheduledThreadPool(5);

                delayLimit = 300;

                this.random = new Random();

        }

        @Override

        public void run() {

                Message msg;

                while (true) {

                        try {

                                msg = msgQueue.take();

                                if (msg == null) {

                                        System.out.println("Null Message Received");

                                        return;

                                }

                                System.out.println("R: " + msg.toString());

                                switch (msg.getType()) {

                                case PUTCHUNK:

                                        Backup backup = new Backup(parentPeer, msg);

                                        executor.execute(backup);

                                        break;

                                case STORED:

                                        peerInformation.addChunkReplication(msg.getFileID(), msg.getChunkNo());

                                        break;

                                case GETCHUNK:

                                        break;

                                case CHUNK:

                                        break;

                                case REMOVED:

                                        FileManager database = parentPeer.getDatabase();

                                        String fileID = msg.getFileID();

                                        int chunkNo = msg.getChunkNo();

                                        Chunk chunkInfo = database.getChunkInformation(fileID, chunkNo);

                                        int perceivedReplication = database.getRecognizedReplication(fileID, chunkNo);

                                        int targetReplication = chunkInfo.getReplication();

                                        if (perceivedReplication < targetReplication) {

                                                byte[] chunkData = parentPeer.loadChunk(fileID, chunkNo);

                                                executor.schedule(

                                                                new RemovedChunkHandler(parentPeer, chunkInfo, chunkData),

                                                                this.random.nextInt(delayLimit + 1),

                                                                TimeUnit.MILLISECONDS

                                                                );

                                        }

                                        break;

                                case DELETE:

                                        Delete delete = new Delete(parentPeer, msg);

                                        executor.execute(delete);

                                        break;

                                default:

                                        return;

                                }

                        } catch (InterruptedException e) {

                                e.printStackTrace();

                        }

                }

        }

        public void pushMessage(byte[] data, int length) {

                Message msgParsed = new Message(data, length);

                msgQueue.add(msgParsed);

        }

}