sdis1819-t4g03

package threads;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.IOException;

import java.nio.charset.StandardCharsets;

import java.util.Random;

import java.util.concurrent.TimeUnit;

import chunk.*;

import server.*;

public class ReceiveGetChunk implements Runnable {

    private String fileId;

    private int chunkNum;

        public ReceiveGetChunk( String fileId, int chunkNr) {

        this.fileId = fileId;

        this.chunkNum = chunkNr;

    }

        @Override

        public void run() {

                for (int i = 0; i < Server.getStorage().getSavedChunks().size(); i++) {

            if (isSameChunk(Server.getStorage().getSavedChunks().get(i).getfileId(), Server.getStorage().getSavedChunks().get(i).getNum()) && !isAbortSend()) {

                String header = "CHUNK " + "1.0" + " " + Server.getServerId() + " " + this.fileId + " " + this.chunkNum + "\r\n\r\n";

                try {

                    byte[] asciiHeader = header.getBytes(StandardCharsets.US_ASCII);

                    String chunkPath = "database/" + Server.getServerId() + "/" + fileId + "." + chunkNum;

                    File file = new File(chunkPath);

                    byte[] body = new byte[(int) file.length()];

                    FileInputStream in = new FileInputStream(file);

                    in.read(body);

                    byte[] message = new byte[asciiHeader.length + body.length];

                    System.arraycopy(asciiHeader, 0, message, 0, asciiHeader.length);

                    System.arraycopy(body, 0, message, asciiHeader.length, body.length);

                    SendMessage sendThread = new SendMessage(message, "mdr");

                    System.out.println("Sent" + "CHUNK " + "1.0" + " " + Server.getServerId() + " " + this.fileId + " " + this.chunkNum);

                    Random random = new Random();

                    Server.getThreadLauncher().schedule(sendThread, random.nextInt(401), TimeUnit.MILLISECONDS);

                } catch (IOException e) {

                    e.printStackTrace();

                }

            }

    }

}

    private boolean isSameChunk(String fileId, int chunkNr) {

        return fileId.equals(this.fileId) && chunkNr == this.chunkNum;

    }

    private boolean isAbortSend() {

        for (int i = 0; i < Server.getStorage().getReceivedChunks().size(); i++) {

            if (isSameChunk(Server.getStorage().getReceivedChunks().get(i).getfileId(), Server.getStorage().getReceivedChunks().get(i).getNum()))

                return true;

        }

        return false;

}

}