美文网首页LInux/C++
libuv学习笔记2------以stream的形式操作文件

libuv学习笔记2------以stream的形式操作文件

作者: _李恒 | 来源:发表于2020-11-28 21:04 被阅读0次

    在libuv中,流操作是很多操作的基础。比如读写管道、TCP/UDP套接字。在libuv的流操作中需要对输入持续流进行监控,一旦输入流有新的内容到来,便会进行空间申请、拿取数据、数据处理等一系列的操作。下面以读写管道为例开始介绍libuv的流操作:

    1.API函数介绍:

    1.1.开始输入流的监控

    int uv_read_start(uv_stream_t*,uv_alloc_cb alloc_cb,uv_read_cb read_cb);

    参数1:要监控的对象可以是uv_pipe_t、uv_udp_t、uv_tcp_t等。

    参数2:回调函数,用于申请存储读到的内容的空间;

    参数3:回调函数,用于消息处理。

    1.2.往输入流写内容

    int uv_write(uv_write_t* req,uv_stream_t* handle,const uv_buf_t bufs[],unsigned int nbufs,uv_write_cb cb);

    参数1:写文件流对象;

    参数2:要写的目标对象;

    参数3:已经初始化完成的指向uv_buf_t结构体的指针;

    参数4:一般为1;

    参数5:写数据完成后的回调函数。

    1.3.初始化管道

    int uv_pipe_init(uv_loop_t, uv_pipe_t handle, int ipc);

    参数1:event-loop;

    参数2:与管道关联的对象;

    参数3:一般写为0,具体作用在后面进行介绍;

    1.4.打开/绑定管道

    int uv_pipe_open(uv_pipe_t*, uv_file file);

    参数1:与管道关联的对象;

    参数2:文件描述符。

    1.5.关闭流

    void uv_close(uv_handle_t* handle, uv_close_cb close_cb);

    参数1:流对象,可以是uv_pipe_t、uv_udp_t、uv_tcp_t等。

    参数2:回调函数,一般为NULL(uv_close、uv_fs_close等函数都可认为是同步的)

    2.回调函数介绍:

    2.1.uv_read_start回调函数介绍

    2.1.1.alloc_cb回调函数介绍

    函数指针原型:

    void (uv_alloc_cb)(uv_handle_t handle,size_t suggested_size,uv_buf_t* buf);

    handle:上下文传递的对象

    suggested_size:libuv建议的应该申请的空间的大小

    buf:uv_buf_t结构体指针,在此回调函数里要进行的任务就是申请空间,并填充这个结构体。

    2.1.2.read_cb回调函数介绍

    函数指针原型:

    void (uv_read_cb)(uv_stream_t stream,ssize_t nread,const uv_buf_t* buf);

    nread:读到的字节数

    buf:对应alloc_cb回调函数里的buf,buf->base代表了存储读取到的数据的内容的首地址。

    此函数存在的目的便是对接收到的数据进行数据处理(需要注意的是数据处理时间不能太长)。

    2.2.uv_write回调函数介绍

    函数指针原型:

    void (uv_write_cb)(uv_write_t req, int status);

    其中req为传递的上下文对象,

    status表征是否成功写入。为0代表成功,其他值代表失败,可用uv_strerror函数打印失败原因。

    3.用libuv实现tee命令的代码

    #include <stdio.h>
    #include <fcntl.h>
    #include <unistd.h>
    #include <string.h>
    #include <stdlib.h>
     
    #include <uv.h>
     
    typedef struct {
        uv_write_t req;
        uv_buf_t buf;
    } write_req_t;
     
    uv_loop_t *loop;
    uv_pipe_t stdin_pipe;
    uv_pipe_t stdout_pipe;
    uv_pipe_t file_pipe;
     
    void alloc_buffer(uv_handle_t *handle, size_t suggested_size, uv_buf_t *buf) {
        *buf = uv_buf_init((char*) malloc(suggested_size), suggested_size);
    }
     
    void free_write_req(uv_write_t *req) {
        write_req_t *wr = (write_req_t*) req;
        free(wr->buf.base);
        free(wr);
    }
     
    void on_stdout_write(uv_write_t *req, int status) {
        free_write_req(req);
    }
     
    void on_file_write(uv_write_t *req, int status) {
        free_write_req(req);
    }
     
    void write_data(uv_stream_t *dest, size_t size, uv_buf_t buf, uv_write_cb cb) {
        write_req_t *req = (write_req_t*) malloc(sizeof(write_req_t));
        req->buf = uv_buf_init((char*) malloc(size), size);
        memcpy(req->buf.base, buf.base, size);
        uv_write((uv_write_t*) req, (uv_stream_t*)dest, &req->buf, 1, cb);
    }
     
    void read_stdin(uv_stream_t *stream, ssize_t nread, const uv_buf_t *buf) {
        if (nread < 0){
            if (nread == UV_EOF){
                // end of file
                uv_close((uv_handle_t *)&stdin_pipe, NULL);
                uv_close((uv_handle_t *)&stdout_pipe, NULL);
                uv_close((uv_handle_t *)&file_pipe, NULL);
            }
        } else if (nread > 0) {
            write_data((uv_stream_t *)&stdout_pipe, nread, *buf, on_stdout_write);
            write_data((uv_stream_t *)&file_pipe, nread, *buf, on_file_write);
        }
     
        // OK to free buffer as write_data copies it.
        if (buf->base)
            free(buf->base);
    }
     
    int main(int argc, char **argv) {
        loop = uv_default_loop();
     
        uv_pipe_init(loop, &stdin_pipe, 0);
        uv_pipe_open(&stdin_pipe, 0);
     
        uv_pipe_init(loop, &stdout_pipe, 0);
        uv_pipe_open(&stdout_pipe, 1);
        
        uv_fs_t file_req;
        int fd = uv_fs_open(loop, &file_req, argv[1], O_CREAT | O_RDWR, 0644, NULL);
        uv_pipe_init(loop, &file_pipe, 0);
        uv_pipe_open(&file_pipe, fd);
     
        uv_read_start((uv_stream_t*)&stdin_pipe, alloc_buffer, read_stdin);
     
        uv_run(loop, UV_RUN_DEFAULT);
        return 0;
    }
    

    代码的基本逻辑:

    1.初始化管道1,并将管道与stdin关联起来。

    2.初始化管道2,并将管道与stdout关联起来。

    3.初始化管道3,并将管道与打开的文件关联起来。

    4.开始监控stdin输入管道。

    5.在stdin有数据到来时,调用alloc_buffer回调函数,在堆上申请一定的空间。并将数据存入新申请的空间。

    6.调用read_stdin函数进行消息处理,在本例中的消息处理便是将stdin的内容分别写到标准输出和文件。

    7.在读到文件末尾时,关闭流,整个程序运行完毕。

    相关文章

      网友评论

        本文标题:libuv学习笔记2------以stream的形式操作文件

        本文链接:https://www.haomeiwen.com/subject/lkbcwktx.html