sdis1819-t7g05

import java.io.IOException;

import java.net.DatagramPacket;

import java.net.DatagramSocket;

import java.net.InetAddress;

import java.util.concurrent.ConcurrentHashMap;

/**

 * ReclaimTask

 */

public class ReclaimTask implements Runnable {

    private static final String CRLF = Integer.toHexString(0xD) + Integer.toHexString(0xA);

    private InetAddress MCaddress;

    private int MCport;

    private Peer peer;

    private String version;

    private int diskspace;

    ReclaimTask(int diskspace, Peer peer) {

        this.MCaddress = peer.getMCaddress();

        this.MCport = peer.getMCport();

        this.diskspace = diskspace;

        this.peer = peer;

    }

    @Override

    public void run() {

        try {

            long storageSize = this.peer.getStorageUsed();

            diskspace = diskspace*1000; //change from KByte to Byte

            if (diskspace > storageSize)

                System.out.println("This amount of KB hasn't been reached yet.");

            else if (diskspace == storageSize)

                System.out.println("This amount of KB is the current amount in the peer.");

            else {

                DatagramSocket socket = new DatagramSocket();

                while(diskspace < this.peer.getStorageUsed()){

                    for(String i : this.peer.chunksMap.keySet()){

                        if(diskspace >= this.peer.getStorageUsed())

                                break;

                        BackupFile fileToCheck = this.peer.chunksMap.get(i);

                        ConcurrentHashMap<Integer, Chunk> chunkMapToCheck = fileToCheck.getChunksMap();

                        for(Integer j : chunkMapToCheck.keySet()) {

                            if(diskspace >= this.peer.getStorageUsed())

                                break;

                            if (this.peer.repDegMap.get(i).get(j) > fileToCheck.getDesiredReplicationDegree()){

                                BackupFile newBackup = this.peer.chunksMap.get(fileToCheck.getFileId());

                                newBackup.removeChunk(j, this.peer);

                                this.peer.repDegMap.get(fileToCheck.getFileId()).put(j, this.peer.repDegMap.get(fileToCheck.getFileId()).get(j)-1);

                                this.peer.chunksMap.put(fileToCheck.getFileId(), newBackup); //apaga chunk do peer atual

                                //REMOVED <Version> <SenderId> <FileId> <ChunkNo> <CRLF><CRLF>

                                String message = "REMOVED " + this.version + " " + peer.getSenderId() + " " + fileToCheck.getFileId() + " " + j + " " + CRLF + CRLF;

                                DatagramPacket packet = new DatagramPacket(message.getBytes(), message.length(), this.MCaddress, this.MCport);

                                socket.send(packet);       

                            }

                        }   

                    }

                }

                socket.close(); 

            }

                this.peer.setStorageCapacity(diskspace);

            this.peer.print("RECLAIM protocol finished");

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

}