通过websocket将kubernetes日志暴露到浏览器

一分耕耘，一分收获！既然打开了这篇文章《通过websocket将kubernetes日志暴露到浏览器》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我正在尝试在 kubernetes 中使用 sidecar 模式来创建日志 sidecar 来公开特定的容器日志。我正在使用 kubernetes 客户端从 kubernetes api 获取日志并通过 websocket 发送出去。代码如下：

func serveWs(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Access-Control-Allow-Origin", "*")
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        if _, ok := err.(websocket.HandshakeError); !ok {
            log.Println(err)
        }
        return
    }

    defer conn.Close()

    logsClient, err := InitKubeLogsClient(config.InCluster)
    if err != nil {
        log.Fatalln(err)
    }

    stream, err := logsClient.GetLogs(config.Namespace, config.PodName, config.ContainerName)
    if err != nil {
        log.Fatalln(err)
    }
    defer stream.Close()
    reader := bufio.NewReader(stream)

    for {
        line, err := reader.ReadString('\n')
        if err != nil {
            log.Fatalln(err)
        }
        conn.WriteMessage(websocket.TextMessage, []byte(line))
    }
}

我使用 https://github.com/gorilla/websocket 作为 websocket 库。在浏览器上

这是做我想做的事的最佳方式吗？有没有更好的方法将日志 api 从 k8s 公开到 websocket？

解决方案

将我的最终代码放在这里，感谢@peter 的提示：

func serveWs(w http.ResponseWriter, r *http.Request) {
    w.Header().Set("Access-Control-Allow-Origin", "*")
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        if _, ok := err.(websocket.HandshakeError); !ok {
            log.Println(err)
        }
        return
    }

    log.Println("create new connection")

    defer func() {
        conn.Close()
        log.Println("connection close")
    }()

    logsClient, err := InitKubeLogsClient(config.InCluster)
    if err != nil {
        log.Println(err)
        return
    }

    stream, err := logsClient.GetLogs(config.Namespace, config.PodName, config.ContainerName)
    if err != nil {
        log.Println(err)
        return
    }
    defer stream.Close()
    reader := bufio.NewReaderSize(stream, 16)
    lastLine := ""
    for {
        data, isPrefix, err := reader.ReadLine()
        if err != nil {
            log.Println(err)
            return
        }

        lines := strings.Split(string(data), "\r")

        length := len(lines)

        if len(lastLine) > 0 {
            lines[0] = lastLine + lines[0]
            lastLine = ""
        }

        if isPrefix {
            lastLine = lines[length-1]
            lines = lines[:(length - 1)]
        }

        for _, line := range lines {
            if err := conn.WriteMessage(websocket.TextMessage, []byte(line)); err != nil {
                log.Println(err)
                return
            }
        }
    }
}

理论要掌握，实操不能落！以上关于《通过websocket将kubernetes日志暴露到浏览器》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！