sdis1819-t7g02

package channels;

import java.io.IOException;

import protocols.SendSoredMessageProtocol;

import service.Cloud;

import service.Constants;

public class RestoreChannel extends Channel{

        private Message receivedMessage;

        public RestoreChannel(String addr, String p, Cloud f) throws IOException {

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

        }

        public void printLog(String msg) {

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

        }

        @Override 

        public void run() 

        {

                printLog(": UP");

                int messageSenderId = -1;

                String messageType;

                try {

                        joinChannelGroup();

                } catch (IOException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                do 

                {

                        try

                        {

                                receivedMessage = captureData();

                                if(receivedMessage == null) {

                                        printLog(" passing message.");

                                        continue;

                                }

                                messageType = receivedMessage.getHeader().getMessageType();

                                messageSenderId = Integer.parseInt(receivedMessage.getHeader().getSenderId());

                                if(messageType.equals(Constants.CHUNK) && messageSenderId != getServerId()) {                        

                                        if(father.isFileChunkRegistered(receivedMessage.getHeader().getFileId(), receivedMessage.getHeader().getChunkNumber())) {

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

                                                if(father.storeRestoreChunk(receivedMessage)) {

                                                        SendSoredMessageProtocol mp = new SendSoredMessageProtocol(buildMessage(), father);

                                                        mp.start();

                                                }

                                                else {

                                                        printLog(" ERROR: couldn't store chunk");

                                                }

                                        }

                                }

                                else {

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

                                }

//                                leaveGroupChannel();                

                        }

                        catch(IOException e) 

                        {

                                printLog(" ERROR: " + "IOException -> " + e.getMessage());

                        }

                }

                while(true);

        }

        public MessageHeader buildMessage() {

                MessageHeader response;

                response = buildResponseMessage();

                return response;

        }

        public MessageHeader buildResponseMessage() {

                MessageHeader response;

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

            headerElements[0] = Constants.CHUNK;

            headerElements[1] = father.getProtocolVersion();

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

            headerElements[3] = receivedMessage.getHeader().getFileId();

            headerElements[4] = receivedMessage.getHeader().getChunkNumber();

            response = new MessageHeader(headerElements);

                return response;

        }

}