124 lines
3.5 KiB
C
124 lines
3.5 KiB
C
/*
|
|
* json rpc server
|
|
*
|
|
* @build make jsonrpc
|
|
* @server bin/jsonrpc_server 1234
|
|
* @client bin/jsonrpc_client 127.0.0.1 1234 add 1 2
|
|
*
|
|
*/
|
|
|
|
#include "hloop.h"
|
|
#include "hbase.h"
|
|
#include "hsocket.h"
|
|
|
|
#include "cJSON.h"
|
|
#include "router.h"
|
|
#include "handler.h"
|
|
|
|
// hloop_create_tcp_server -> on_accept -> hio_read -> on_recv -> hio_write
|
|
|
|
static int verbose = 0;
|
|
static unpack_setting_t jsonrpc_unpack_setting;
|
|
|
|
jsonrpc_router router[] = {
|
|
{"add", calc_add},
|
|
{"sub", calc_sub},
|
|
{"mul", calc_mul},
|
|
{"div", calc_div},
|
|
};
|
|
#define JSONRPC_ROUTER_NUM (sizeof(router)/sizeof(router[0]))
|
|
|
|
static void on_close(hio_t* io) {
|
|
printf("on_close fd=%d error=%d\n", hio_fd(io), hio_error(io));
|
|
}
|
|
|
|
static void on_recv(hio_t* io, void* readbuf, int readbytes) {
|
|
// printf("on_recv fd=%d readbytes=%d\n", hio_fd(io), readbytes);
|
|
if (verbose) {
|
|
char localaddrstr[SOCKADDR_STRLEN] = {0};
|
|
char peeraddrstr[SOCKADDR_STRLEN] = {0};
|
|
printf("[%s] <=> [%s]\n",
|
|
SOCKADDR_STR(hio_localaddr(io), localaddrstr),
|
|
SOCKADDR_STR(hio_peeraddr(io), peeraddrstr));
|
|
}
|
|
|
|
// cJSON_Parse -> router -> cJSON_Print -> hio_write
|
|
char* req_str = (char*)readbuf;
|
|
printf("> %s\n", req_str);
|
|
cJSON* jreq = cJSON_Parse(req_str);
|
|
cJSON* jres = cJSON_CreateObject();
|
|
cJSON* jid = cJSON_GetObjectItem(jreq, "id");
|
|
cJSON* jmethod = cJSON_GetObjectItem(jreq, "method");
|
|
if (cJSON_IsNumber(jid)) {
|
|
long id = cJSON_GetNumberValue(jid);
|
|
cJSON_AddItemToObject(jres, "id", cJSON_CreateNumber(id));
|
|
}
|
|
if (cJSON_IsString(jmethod)) {
|
|
// router
|
|
char* method = cJSON_GetStringValue(jmethod);
|
|
bool found = false;
|
|
for (int i = 0; i < JSONRPC_ROUTER_NUM; ++i) {
|
|
if (strcmp(method, router[i].method) == 0) {
|
|
found = true;
|
|
router[i].handler(jreq, jres);
|
|
break;
|
|
}
|
|
}
|
|
if (!found) {
|
|
not_found(jreq, jres);
|
|
}
|
|
} else {
|
|
bad_request(jreq, jres);
|
|
}
|
|
|
|
char* resp_str = cJSON_PrintUnformatted(jres);
|
|
printf("< %s\n", resp_str);
|
|
// NOTE: +1 for \0
|
|
hio_write(io, resp_str, strlen(resp_str) + 1);
|
|
|
|
cJSON_Delete(jreq);
|
|
cJSON_Delete(jres);
|
|
cJSON_free(resp_str);
|
|
}
|
|
|
|
static void on_accept(hio_t* io) {
|
|
printf("on_accept connfd=%d\n", hio_fd(io));
|
|
if (verbose) {
|
|
char localaddrstr[SOCKADDR_STRLEN] = {0};
|
|
char peeraddrstr[SOCKADDR_STRLEN] = {0};
|
|
printf("accept connfd=%d [%s] <= [%s]\n", hio_fd(io),
|
|
SOCKADDR_STR(hio_localaddr(io), localaddrstr),
|
|
SOCKADDR_STR(hio_peeraddr(io), peeraddrstr));
|
|
}
|
|
|
|
hio_setcb_close(io, on_close);
|
|
hio_setcb_read(io, on_recv);
|
|
hio_set_unpack(io, &jsonrpc_unpack_setting);
|
|
hio_read(io);
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
if (argc < 2) {
|
|
printf("Usage: %s port\n", argv[0]);
|
|
return -10;
|
|
}
|
|
int port = atoi(argv[1]);
|
|
|
|
// init jsonrpc_unpack_setting
|
|
memset(&jsonrpc_unpack_setting, 0, sizeof(unpack_setting_t));
|
|
jsonrpc_unpack_setting.mode = UNPACK_BY_DELIMITER;
|
|
jsonrpc_unpack_setting.package_max_length = DEFAULT_PACKAGE_MAX_LENGTH;
|
|
jsonrpc_unpack_setting.delimiter[0] = '\0';
|
|
jsonrpc_unpack_setting.delimiter_bytes = 1;
|
|
|
|
hloop_t* loop = hloop_new(0);
|
|
hio_t* listenio = hloop_create_tcp_server(loop, "0.0.0.0", port, on_accept);
|
|
if (listenio == NULL) {
|
|
return -20;
|
|
}
|
|
printf("listenfd=%d\n", hio_fd(listenio));
|
|
hloop_run(loop);
|
|
hloop_free(&loop);
|
|
return 0;
|
|
}
|