sdis1819-t2g03

package chunk;

import java.util.HashSet;

import java.util.Set;

import java.util.concurrent.ConcurrentHashMap;

import java.util.concurrent.ConcurrentMap;

import java.util.concurrent.atomic.AtomicIntegerArray;

public class Chunk {

        private ConcurrentMap<String, AtomicIntegerArray> chunkReplication;

        private String fileID;

        private int chunkNo;

        private int replication;

        private byte[] data;

        private int size;

        private Set< Integer > chunks;

        public Chunk(String fileID, int chunkNo, byte[] data) {

                this.fileID = fileID;

                this.chunkNo = chunkNo;

                this.data = data;

        }

        public Chunk(String fileID, int chunkNo, int replication, byte[] data) {

                this(fileID, chunkNo, data);

                this.replication = replication;

        }

        public Chunk(Chunk chunkData, byte[] data) {

                this(chunkData.getFileID(), chunkData.getChunkNo(), chunkData.getReplication(), data);

        }

        public Chunk(int chunkNo, int replication) {

                this.chunkNo = chunkNo;

                this.replication = replication;

                this.chunks = new HashSet<>();

        }

        public Chunk(String fileID, int chunkNo, int replication, int size) {

                this(chunkNo, replication);

                this.fileID = fileID;

                this.size = size;

        }

        // Peer information

        public Chunk() {

                chunkReplication = new ConcurrentHashMap<>();

        }

        public void resetChunkReplication(String fileID) {

                chunkReplication.remove(fileID);

        }

        public void startChunkReplication(String fileID, int numChunks) {

                chunkReplication.putIfAbsent(fileID, new AtomicIntegerArray(numChunks));

        }

        public Integer addChunkReplication(String fileID, int chunkNo) {

                if (!chunkReplication.containsKey(fileID))

                        return null;

                int replication = chunkReplication.get(fileID).addAndGet(chunkNo, 1);

                System.out.println("Incrementing replication of " + fileID + "/" + chunkNo + " to " + replication);

                return replication;

        }

        public int getChunkReplication(String fileID, int chunkNo) {

                return chunkReplication.get(fileID).get(chunkNo);

        }

        public AtomicIntegerArray getChunkReplication(String fileID) {

                return chunkReplication.get(fileID);

        }

        // Getters and setters 

        public String getFileID() { return fileID; }

        public int getChunkNo() { return chunkNo; }

        public int getReplication() { return replication; }

        public byte[] getData() { return data; }

        public boolean removeChunk(Integer peerID) { return chunks.remove(peerID); }

        public boolean addChunk(Integer peerID) { return chunks.add(peerID); }

        public int getNumChunks() { return chunks.size(); }

        public int getSize() { return size; }

        public Set<Integer> getChunks() { return chunks; }

        public void setFileID(String fileID) { this.fileID = fileID; }

        public void setChunkNo(int chunkNo) { this.chunkNo = chunkNo; }

        public void setReplication(int replication) { this.replication = replication; }

        public void setData(byte[] data) { this.data = data; }

        public void setSize(int size) { this.size = size; }

        public void setChunks(Set<Integer> mirrors) { this.chunks = mirrors; }

}