feat: get realtime pod logs in a new tab

Author: Yann-Gael GAUTHERON

cmd/ktop.go

@@ -132,8 +132,9 @@ func (k *ktopCmd) run(cmd *cobra.Command, args []string) error {
 			termui.NewCol(1./2, logo),
 			termui.NewCol(1./2, hint),
 		),
-		termui.NewRow(6./12, monitor.GetPodTable()),
-		termui.NewRow(4./12,
+		termui.NewRow(3./12, monitor.GetPodTable()),
+		termui.NewRow(5./12, monitor.GetLogs()),
+		termui.NewRow(2./12,
 			termui.NewCol(1./2, monitor.GetCPUGraph()),
 			termui.NewCol(1./2, monitor.GetMemGraph()),
 		),

pkg/ktop/ktop.go

@@ -40,6 +40,7 @@ const (
 type Monitor struct {
 	*kube.KubeClients
 
+	logs            *ui.TextField
 	table           *ui.Table
 	tableTypeCircle *ring.Ring
 
@@ -67,6 +68,11 @@ func NewMonitor(kubeclients *kube.KubeClients, podQuery, containerQuery, nodeQue
 	table.BorderStyle = termui.NewStyle(borderColor)
 	table.CursorColor = selectedTableColor
 
+	// logs of pod
+	logs := ui.NewTextField()
+	logs.Text = `TEST TextField`
+	logs.TextStyle = termui.NewStyle(termui.Color(244), termui.ColorClear)
+
 	// graph for cpu
 	cpu := ui.NewGraph()
 	cpu.Title = "⎈ CPU Usage ⎈"
@@ -86,6 +92,7 @@ func NewMonitor(kubeclients *kube.KubeClients, podQuery, containerQuery, nodeQue
 	mem.LimitColor = graphLimitColor
 
 	monitor.table = table
+	monitor.logs = logs
 	monitor.cpuGraph = cpu
 	monitor.memGraph = mem
 	return monitor
@@ -149,6 +156,10 @@ func (m *Monitor) GetPodTable() *ui.Table {
 	return m.table
 }
 
+func (m *Monitor) GetLogs() *ui.TextField {
+	return m.logs
+}
+
 func (m *Monitor) Update() error {
 	nodeList, err := m.GetNodeList(labels.Everything())
 	if err != nil {
@@ -277,6 +288,12 @@ func (m *Monitor) fetchPodResources() ([]*resource.Resource, []*resource.Summari
 	// filtered
 	for _, podMetrics := range FilterPodMetrics(m.podQuery, podMetricsList.Items) {
 		podName := podMetrics.Name
+		podLogs, err := m.GetPodLogs(*m.Flags.Namespace, podName)
+		// fmt.Print("podLogs")
+		// fmt.Print(podLogs)
+		if err != nil {
+			fmt.Print(err)
+		}
 		pod := FindPod(podName, podList.Items)
 		if pod == nil {
 			continue
@@ -297,7 +314,7 @@ func (m *Monitor) fetchPodResources() ([]*resource.Resource, []*resource.Summari
 			corev1.ResourceList{
 				corev1.ResourceCPU:    cpu,
 				corev1.ResourceMemory: mem,
-			})
+			}, podLogs)
 		summarizedResources = append(summarizedResources, summarizedResource)
 	}
 	return resources, summarizedResources, nil
@@ -330,6 +347,8 @@ func (m *Monitor) updateSummarizedGraph(nodeList *corev1.NodeList, summarized *r
 	limitMemory := GetResourceValue(node.Status.Allocatable, corev1.ResourceMemory)
 	limitMemoryStr := GetResourceValueString(node.Status.Allocatable, corev1.ResourceMemory)
 
+	m.logs.Text = summarized.GetLogs()
+
 	m.cpuGraph.LabelHeader = fmt.Sprintf("Name: %v", summarized.GetPodName())
 	m.cpuGraph.Data = append(m.cpuGraph.Data, cpuUsage)
 	m.cpuGraph.LabelData = fmt.Sprintf("Usage: %v", cpuUsageStr)

pkg/kube/clients.go

@@ -1,6 +1,8 @@
 package kube
 
 import (
+	"io"
+
 	"github.com/pkg/errors"
 
 	corev1 "k8s.io/api/core/v1"
@@ -53,6 +55,40 @@ func (k *KubeClients) GetPodList(namespace string, labelSelector labels.Selector
 	return k.clientset.CoreV1().Pods(namespace).List(metav1.ListOptions{LabelSelector: labelSelector.String()})
 }
 
+func (k *KubeClients) GetPodLogs(namespace string, podName string) (string, error) {
+	count := int64(100)
+	follow := false
+	message := ""
+	podLogOptions := corev1.PodLogOptions{
+		//Container: containerName,
+		Follow:    follow,
+		TailLines: &count,
+	}
+	podLogRequest := k.clientset.CoreV1().
+		Pods(namespace).
+		GetLogs(podName, &podLogOptions)
+	stream, err := podLogRequest.Stream()
+	if err != nil {
+		return "failure1", err
+	}
+	defer stream.Close()
+	for {
+		buf := make([]byte, 2000)
+		numBytes, err := stream.Read(buf)
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			return "failure2", err
+		}
+		if numBytes == 0 {
+			continue
+		}
+		message += string(buf[:numBytes])
+	}
+	return message, nil
+}
+
 func (k *KubeClients) GetPodMetricsList(namespace string, labelSelector labels.Selector) (*metrics.PodMetricsList, error) {
 	return k.metricsClient.getPodMetricsList(namespace, labelSelector)
 }

pkg/resource/summarized.go

@@ -9,13 +9,15 @@ import (
 type SummarizedResource struct {
 	podName  string
 	nodeName string
+	logs     string
 	usage    corev1.ResourceList
 }
 
-func NewSummarizedResource(p corev1.Pod, sumUsage corev1.ResourceList) *SummarizedResource {
+func NewSummarizedResource(p corev1.Pod, sumUsage corev1.ResourceList, logs string) *SummarizedResource {
 	return &SummarizedResource{
 		podName:  p.Name,
 		nodeName: p.Spec.NodeName,
+		logs:     logs,
 		usage:    sumUsage,
 	}
 }
@@ -33,6 +35,10 @@ func (s *SummarizedResource) GetCpuUsage() (float64, string) {
 		GetResourceValueString(s.usage, corev1.ResourceCPU)
 }
 
+func (s *SummarizedResource) GetLogs() string {
+	return s.logs
+}
+
 func (s *SummarizedResource) GetMemoryUsage() (float64, string) {
 	return GetResourceValue(s.usage, corev1.ResourceMemory),
 		GetResourceValueString(s.usage, corev1.ResourceMemory)