aboutsummaryrefslogtreecommitdiffstats
path: root/P1/src/mapreduce.c
blob: 0c1510e9b2aa10235e014a5a91f560151c5bc861 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include "mapreduce.h"

int main(int argc, char *argv[]) {
	
	if(argc < 4) {
		printf("Less number of arguments.\n");
		printf("./mapreduce #mappers #reducers inputFile\n");
		exit(0);
	}

	// ###### DO NOT REMOVE ######
	int nMappers 	= strtol(argv[1], NULL, 10);
	int nReducers 	= strtol(argv[2], NULL, 10);
	if (nMappers < nReducers) {
		printf("Number of mappers must be greater than or equal to the number of reducers.\n");
		exit(2);
	}
	char *inputFile = argv[3];

	// ###### DO NOT REMOVE ######
	bookeepingCode();

	// ###### DO NOT REMOVE ######
	pid_t pid = fork();
	if(pid == 0){
		//send chunks of data to the mappers in RR fashion
		sendChunkData(inputFile, nMappers);
		exit(0);
	}
	sleep(1);


	pid_t mapperPid;
	for (int i = 1; i <= nMappers; i++) {
		mapperPid = fork();
		if (mapperPid == 0) {
			//TODO: exec here
			char numMap[10];
			sprintf(numMap, "%d", i);
			execl("./mapper", "./mapper", &numMap, NULL);
			printf("Mapper execution failed! Make sure \"mapper\" exists in the directory.\n");
			return 4;
		} else if (mapperPid == -1) {
			printf("Fork failed!\n");
			return 1;
		}
	}
	
    for (int i = 0; i <= nMappers; i++) {
		//Hopefully this works
		int waitError = wait(NULL);
		if (waitError == -1)
		{
			printf("Wait failed!\n");
			return 5;
		}

	// ###### DO NOT REMOVE ######
    // shuffle sends the word.txt files generated by mapper 
    // to reducer based on a hash function
	pid = fork();
	if(pid == 0){
		shuffle(nMappers, nReducers);
		exit(0);
	}
	sleep(1);


	pid_t reducerPid;
	for (int i = 1; i <= nReducers; i++) {
		reducerPid = fork();
		if (reducerPid == 0) {
			char numRed[10];
			sprintf(numRed, "%d", i);
			execl("./reducer", "./reducer", &numRed, NULL);
			printf("Reducer execution failed! Make sure \"reducer\" exists in the directory.\n");
			return 3;
		} else if (reducerPid == -1) {
			printf("Fork failed!\n");
			return 2;
		}
	}

	for (int i = 0; i <= nReducers; i++) {
		//Maybe this works too?
		int waitError = wait(NULL);
		if (waitError == -1) {
			printf("Wait failed!\n");
			return 5;
		}
	}

	return 0;
}