golang HTTP 服务器 处理 日志/Stream流的操作

(编辑:jimmy 日期: 2024/12/24 浏览:2)

目前,我开发 HTTP 服务, 用的是 beego框架, 方便了很多。

但是, 有时候,还是会遇到一些 特殊的场景。

比如: 过滤日志。

这应该是一种典型的stream,同时数据量也适中, 不会有人,为了这个, 就用一些很重的框架。

可以这样直观的描述这个 逻辑

其他组件 产生 log
||
\ /
我的组件,业务处理
||
\ /
用户, http client

这种情景下, 有几个特殊点:

1. 难以用 string,或者 byte 数组 收集数据

2. 数据Source 端,不断的有数据产生

3. 数据缓冲,如果占有的 内存太多, 可能导致 服务崩溃

通常情况下,我们准备好数据, 然后调用Beego框架的方法,将数据发送到客户端,就不管了。

而如果,我们需要根据处理的情况,多次写数据到客户端,该怎么办呢?

首先,对于 这种简单的 流数据, golang 提供了一个 结构。

pipeReader, pipeWriter := io.Pipe()

这个方法的原型是这样的

func Pipe() (*PipeReader, *PipeWriter)

它返回紧密相连的一对 Reader 和 Writer。 他们的“生命周期”相同。

任何 写到 Writer中的数据, 直接流到了Reader中。这个 和 Linux 命令行中 “管道 |” 很像。

我们先开个goroutine 接收 日志数据

 go func () {
 for {
 var log []byte
 //log = 
 pipeWriter.Write(log)
 
 //break;
 }
 pipeWriter.CloseWithError(io.EOF)
 }

主逻辑中, 处理日志

 defer pipeReader.Close()
 rr := bufio.NewReader(io.Reader(pipeReader))
 for {

 line, err := rr.ReadBytes('\n')
 if io.EOF == err {
 break
 }
........
 }

最后, 输出到客户端

var out []byte
ctl.Ctx.ResponseWriter.Write(out)
ctl.Ctx.ResponseWriter.Flush()

总结:

iopipe 直接 对接了 日志输出, 缓冲很小,

处理后的结果, 直接输出到 http 客户端。

尤其是第二点,很重要,我在处理这个逻辑的时候, 发现服务器,有几次意外崩溃,后来,才意识到,beego的controller 如果缓冲 处理后的数据,有可能仍然占有大量内存。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。如有错误或未考虑完全的地方,望不吝赐教。

一句话新闻

一文看懂荣耀MagicBook Pro 16
荣耀猎人回归!七大亮点看懂不只是轻薄本,更是游戏本的MagicBook Pro 16.
人们对于笔记本电脑有一个固有印象:要么轻薄但性能一般,要么性能强劲但笨重臃肿。然而,今年荣耀新推出的MagicBook Pro 16刷新了人们的认知——发布会上,荣耀宣布猎人游戏本正式回归,称其继承了荣耀 HUNTER 基因,并自信地为其打出“轻薄本,更是游戏本”的口号。
众所周知,寻求轻薄本的用户普遍更看重便携性、外观造型、静谧性和打字办公等用机体验,而寻求游戏本的用户则普遍更看重硬件配置、性能释放等硬核指标。把两个看似难以相干的产品融合到一起,我们不禁对它产生了强烈的好奇:作为代表荣耀猎人游戏本的跨界新物种,它究竟做了哪些平衡以兼顾不同人群的各类需求呢?