LInux/C++

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

2020-11-28  本文已影响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.在读到文件末尾时,关闭流,整个程序运行完毕。

上一篇下一篇

猜你喜欢

热点阅读