raftMain.c

#include <stdio.h>
#include <unistd.h>
#include <pthread.h>
#include <string.h>
#include <assert.h>
#include <time.h>
#include <stdlib.h>
#include <getopt.h>
#include <raft.h>
#include <raft/uv.h>
#include "raftServer.h"
#include "common.h"

const char *exe_name;

void *startServerFunc(void *param) {
	SRaftServer *pServer = (SRaftServer*)param;
	int32_t r = raftServerStart(pServer);
	assert(r == 0);

	return NULL;
}

// Console ---------------------------------
void console(SRaftServer *pRaftServer) {
	while (1) {
		char cmd_buf[COMMAND_LEN];
		memset(cmd_buf, 0, sizeof(cmd_buf));
		char *ret = fgets(cmd_buf, COMMAND_LEN, stdin);
		if (!ret) {
			exit(-1);
		}

		int pos = strlen(cmd_buf);                  
		if(cmd_buf[pos - 1] == '\n') {
			cmd_buf[pos - 1] = '\0';
		}

		if (strncmp(cmd_buf, "", COMMAND_LEN) == 0) {
			continue;
		}

		printf("cmd_buf: [%s] \n", cmd_buf);
	}
}

void *startConsoleFunc(void *param) {
	SRaftServer *pServer = (SRaftServer*)param;
	console(pServer);
	return NULL;
}

// Config ---------------------------------
typedef struct SRaftServerConfig {
	char host[HOST_LEN];
	uint32_t port;
	char dir[DIR_LEN];
} SRaftServerConfig;

void parseConf(int argc, char **argv, SRaftServerConfig *pConf) {
	snprintf(pConf->host, sizeof(pConf->host), "%s", argv[1]);
	sscanf(argv[2], "%u", &pConf->port);
	snprintf(pConf->dir, sizeof(pConf->dir), "%s", argv[3]);
}

void printConf(SRaftServerConfig *pConf) {
	printf("conf: %s:%u %s \n", pConf->host, pConf->port, pConf->dir);
}

// -----------------------------------------
void usage() {
	printf("\n");
	printf("usage: %s host port dir \n", exe_name);
	printf("\n");
	printf("eg: \n");
	printf("%s 127.0.0.1 10000 ./data \n", exe_name);
	printf("\n");
}

int main(int argc, char **argv) { 
	srand(time(NULL));
	int32_t ret;

	exe_name = argv[0];
	if (argc != 4) {
		usage();
		exit(-1);
	}

	SRaftServerConfig conf;
	parseConf(argc, argv, &conf);
	printConf(&conf);	

	struct raft_fsm fsm;
	initFsm(&fsm);

	SRaftServer raftServer;
	ret = raftServerInit(&raftServer, conf.host, conf.port, conf.dir, &fsm);
	assert(ret == 0);

	pthread_t tidRaftServer;
	pthread_create(&tidRaftServer, NULL, startServerFunc, &raftServer);

	pthread_t tidConsole;
	pthread_create(&tidConsole, NULL, startConsoleFunc, &raftServer);

	while (1) {
		sleep(10);
	}

	return 0;
}