libhv/event/nlog.c

#include "nlog.h"

#include "list.h"
#include "hdef.h"
#include "hbase.h"
#include "hsocket.h"
#include "hmutex.h"

typedef struct network_logger_s {
    hloop_t*            loop;
    hio_t*              listenio;
    struct list_head    clients;
} network_logger_t;

typedef struct nlog_client {
    hio_t*              io;
    struct list_node    node;
} nlog_client;

static network_logger_t s_logger = {0};
static hmutex_t         s_mutex;

static void on_close(hio_t* io) {
    printd("on_close fd=%d error=%d\n", hio_fd(io), hio_error(io));

    nlog_client* client = (nlog_client*)hevent_userdata(io);
    if (client) {
        hevent_set_userdata(io, NULL);

        hmutex_lock(&s_mutex);
        list_del(&client->node);
        hmutex_unlock(&s_mutex);

        HV_FREE(client);
    }
}

static void on_read(hio_t* io, void* buf, int readbytes) {
    printd("on_read fd=%d readbytes=%d\n", hio_fd(io), readbytes);
    printd("< %s\n", (char*)buf);
    // nothing to do
}

static void on_accept(hio_t* io) {
    /*
    printd("on_accept connfd=%d\n", hio_fd(io));
    char localaddrstr[SOCKADDR_STRLEN] = {0};
    char peeraddrstr[SOCKADDR_STRLEN] = {0};
    printd("accept connfd=%d [%s] <= [%s]\n", hio_fd(io),
            SOCKADDR_STR(hio_localaddr(io), localaddrstr),
            SOCKADDR_STR(hio_peeraddr(io), peeraddrstr));
    */

    hio_setcb_read(io, on_read);
    hio_setcb_close(io, on_close);
    hio_read(io);

    // free on_close
    nlog_client* client;
    HV_ALLOC_SIZEOF(client);
    client->io = io;
    hevent_set_userdata(io, client);

    hmutex_lock(&s_mutex);
    list_add(&client->node, &s_logger.clients);
    hmutex_unlock(&s_mutex);
}

void network_logger(int loglevel, const char* buf, int len) {
    struct list_node* node;
    nlog_client* client;

    hmutex_lock(&s_mutex);
    list_for_each (node, &s_logger.clients) {
        client = list_entry(node, nlog_client, node);
        hio_write(client->io, buf, len);
    }
    hmutex_unlock(&s_mutex);
}

hio_t* nlog_listen(hloop_t* loop, int port) {
    s_logger.loop = loop;
    s_logger.listenio = hloop_create_tcp_server(loop, "0.0.0.0", port, on_accept);
    list_init(&s_logger.clients);
    hmutex_init(&s_mutex);
    return s_logger.listenio;
}