sdis1819-t7g02

/**

 * 

 */

package channels;

import java.io.IOException;

import protocols.recoverChunkRepDegreeProtocol;

import service.Chunk;

import service.Cloud;

import service.Constants;

/**

 * @author dnc18

 *

 */

public class ControlChannel extends Channel {

        private Message receivedMessage;

        public ControlChannel(String addr, String p, Cloud servicePeerRef) throws IOException {

                super(addr, p, servicePeerRef.getID(), servicePeerRef);

        }

        public void printLog(String msg) {

                System.out.println("MC " + serverId + msg);

        }

        @Override

        public void run() {

                recoverChunkRepDegreeProtocol rcrdp = null;

                Message m;

                printLog(": UP");

                try {

                        joinChannelGroup();

                } catch (IOException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                do {

                        try {

                                receivedMessage = captureData();

                                if(receivedMessage == null) {

                                        continue;

                                }

                                if(getServerId() != Integer.parseInt(receivedMessage.getHeader().getSenderId())) {

//                                        printLog(" received: " + receivedMessage.getHeader().toString());

                                        switch(receivedMessage.getHeader().getMessageType()) {

                                                case Constants.STORED:

                                                        father.intrepertStoredMessage(receivedMessage.getHeader());

                                                        break;

                                                case Constants.GETCHUNK:

                                                        father.intrepertGetChunkMessage(receivedMessage.getHeader());

                                                        break;

                                                case Constants.DELETE:

//                                                        printLog(" before delete: " + father.getAvailableSpace() + " avaliable bytes and " + father.getNumberOfStoredChunks() + " stored chunks.");

                                                        father.deleteFile(receivedMessage.getHeader());

//                                                        printLog(" adter delete: " + father.getAvailableSpace() + " avaliable bytes and " + father.getNumberOfStoredChunks() + " stored chunks.");

                                                        break;

                                                case Constants.REMOVED:

                                                        if(father.intrepertStoredMessage(receivedMessage.getHeader())) {

                                                                printLog(": chunk replication degree disrespected, sending again.");

                                                                m = buildReplicationRestoreMessage(receivedMessage.getHeader().getFileId(), receivedMessage.getHeader().getChunkNumber());

                                                                if(m != null) {

                                                                        rcrdp = new recoverChunkRepDegreeProtocol(father, m);

                                                                        System.out.println("Calling");

                                                                        rcrdp.run();

                                                                }

                                                                else {

                                                                        System.out.println("couldn t create backup protocol thread");

                                                                }

                                                        }

                                                        break;

                                                default: // MC, MDB, MDR

                                                        //printLog(" ERROR: Invalid message header received -> " + receivedMessage.getHeader().getMessageType());

                                                        continue;

                                        }

                                }

//                                leaveGroupChannel();

                        }

                        catch (Exception e) {

                                printLog(" ERROR: " + e.getMessage());

                                e.printStackTrace();

                                break;

                        }

                } while(true);

        }

        private Message buildReplicationRestoreMessage(String fileId, String chunkNo) {

                Message m = null;

                MessageHeader h = null;

                String[] headerElements = new String[Constants.PUTCHUNK_N_ARGS];

                Chunk toSave = father.getFileChunkRepDegree(fileId, chunkNo);

                if(toSave == null) {

                        return null;

                }

                try {

                        headerElements[0] = Constants.PUTCHUNK;

                    headerElements[1] = father.getProtocolVersion();

                    headerElements[2] = Integer.toString(father.getID());

                    headerElements[3] = fileId;

                             headerElements[4] = chunkNo;

                             headerElements[5] = Integer.toString(toSave.getReplicationDegree());

                             h = new MessageHeader(headerElements.clone());

                             m = new Message(h, toSave.getChunkContent());

                }

                   catch(IllegalArgumentException e) {

                           System.out.println("Error creating backup protocol -> " + e.getMessage());

                           return null;

                   }

                return m;

        }

}