Libuv 1.34.2 源碼詳解 ---- 以uvCat爲例講解

這些天病毒肆虐，心裏很難過。一直對libuv高併發處理的能力好奇，只好呆家裏看了一下libuv的源碼。

libuva其實代碼量很小，很精悍。但理解起來並不容易。網絡上有不少解析的文章，不過大多見樹葉而不見森林，我前幾天剛在網上找資料時，看了一天都還沒摸着頭腦。我想對於很多剛開始學習libuv的朋友，可能不少和我類似，最重要的應該是知道誰，起的作用是什麼？整個框架大概是什麼樣子的？這樣，當碰到具體的細節時，可以大大加速對源碼的理解。

我用的版本是最新版v1.34.2，源碼是網上找的uvCat，就是讀取一個文件，再把文件內容顯示出來，命令類似uvCat d:\test.txt。

其實大家平常使用最多的UDP/TCP，其原理都類似的。全部源碼如下，

#include <assert.h>
#include <stdio.h>
#include <fcntl.h>
#include <io.h> //#include <unistd.h>

#include "../libuv/include/uv.h"

#pragma comment(lib, "ws2_32.lib")
#pragma comment(lib, "Iphlpapi.lib")
#pragma comment(lib, "Userenv.lib")
#pragma comment(lib, "Psapi.lib")

// gcc -Wall main.c -o uvcat -luv
// ./uvcat ./1.txt

void on_read(uv_fs_t* req);

uv_fs_t open_req;
uv_fs_t read_req;
uv_fs_t write_req;

static char buffer[1024];

static uv_buf_t iov;

void on_write(uv_fs_t* req) {
    if (req->result < 0) {
        fprintf(stderr, "Write error: %s\n", uv_strerror((int)req->result));
    }
    else {
        uv_fs_read(uv_default_loop(), &read_req, open_req.result, &iov, 1, -1, on_read);
    }
}

void on_read(uv_fs_t* req) {
    if (req->result < 0) {
        fprintf(stderr, "Read error: %s\n", uv_strerror(req->result));
    }
    else if (req->result == 0) {
        uv_fs_t close_req;
        // synchronous
        uv_fs_close(uv_default_loop(), &close_req, open_req.result, NULL);
        printf("******Read file done!!!");
    }
    else if (req->result > 0) {
        iov.len = req->result;  // 讀取數據的長度(數據讀寫完畢才觸發回調)
        // 因爲cat功能，所以將讀取的數據寫入控制檯
        uv_fs_write(uv_default_loop(), &write_req, 1, &iov, 1, -1, on_write);
    }
}

void on_open(uv_fs_t* req) {
    // The request passed to the callback is the same as the one the call setup
    // function was passed.
    assert(req == &open_req);
    if (req->result >= 0) {
        iov = uv_buf_init(buffer, sizeof(buffer));
        uv_fs_read(uv_default_loop(), &read_req, req->result,
            &iov, 1, -1, on_read);
    }
    else {
        fprintf(stderr, "error opening file: %s\n", uv_strerror((int)req->result));
    }
}

int main(int argc, char** argv) {
    uv_fs_open(uv_default_loop(), &open_req, argv[1], O_RDONLY, 0, on_open);
    uv_run(uv_default_loop(), UV_RUN_DEFAULT);

    uv_fs_req_cleanup(&open_req);
    uv_fs_req_cleanup(&read_req);
    uv_fs_req_cleanup(&write_req);
    return 0;
}

整體上來說，剛開始看到這種不斷callback(算是循環遞交作業的一種程序架構吧)的風格，讓人印象深刻，