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

Question

我正在尝试在 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？

Answer 1

将我的最终代码放在这里

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

            }

        }

    }