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

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

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,324评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,356评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,328评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,147评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,160评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,115评论 1 296
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,025评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,867评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,307评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,528评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,688评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,409评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,001评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,657评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,811评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,685评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,573评论 2 353

推荐阅读更多精彩内容