mr_protocol.h

#ifndef mr_protocol_h_
#define mr_protocol_h_

#include <string>
#include <vector>

#include "rpc.h"

using namespace std;

#define REDUCER_COUNT 4

enum mr_tasktype {
	NONE = 0, // this flag means no task needs to be performed at this point
	MAP,
	REDUCE
};

class mr_protocol {
public:
	typedef int status;
	enum xxstatus { OK, RPCERR, NOENT, IOERR };
	enum rpc_numbers {
		asktask = 0xa001,
		submittask,
	};

	struct AskTaskResponse {
		// Lab4: Your definition here.
        int tasktype;
        int index;
        int nfiles;
        string filename;
	};

    friend unmarshall &operator>>(unmarshall &u, AskTaskResponse &res){
        return u >> res.tasktype >> res.filename >> res.index >> res.nfiles;
    }
    friend marshall &operator<<(marshall &m, const AskTaskResponse &res){
        return m << res.tasktype << res.filename << res.index << res.nfiles;
    }

	struct AskTaskRequest {
		// Lab4: Your definition here.
        int index;
	};

	struct SubmitTaskResponse {
		// Lab4: Your definition here.
	};

	struct SubmitTaskRequest {
		// Lab4: Your definition here.
	};

};

#endif