我正在嘗試在 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？
1 回答
泛舟湖上清波郎朗
TA貢獻1818條經驗 獲得超3個贊
將我的最終代碼放在這里
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
? ? ? ? ? ? }
? ? ? ? }
? ? }
- 1 回答
- 0 關注
- 128 瀏覽
添加回答
舉報
0/150
提交
取消