A draft of connector to kubernetse

connector/collector/kubernetes.go

@@ -0,0 +1,125 @@
+package collector
+
+import (
+	"github.com/bcicen/ctop/config"
+	"github.com/bcicen/ctop/models"
+	"k8s.io/client-go/kubernetes"
+)
+
+// Kubernetes collector
+type Kubernetes struct {
+	models.Metrics
+	name       string
+	client     *kubernetes.Clientset
+	running    bool
+	stream     chan models.Metrics
+	done       chan bool
+	lastCpu    float64
+	lastSysCpu float64
+	scaleCpu   bool
+}
+
+func NewKubernetes(client *kubernetes.Clientset, name string) *Kubernetes {
+	return &Kubernetes{
+		Metrics:  models.Metrics{},
+		name:     name,
+		client:   client,
+		scaleCpu: config.GetSwitchVal("scaleCpu"),
+	}
+}
+
+func (c *Kubernetes) Start() {
+	//c.done = make(chan bool)
+	//c.stream = make(chan models.Metrics)
+	//stats := make(chan *api.Stats)
+
+	//go func() {
+	//	opts := api.StatsOptions{
+	//		ID:     c.id,
+	//		Stats:  stats,
+	//		Stream: true,
+	//		Done:   c.done,
+	//	}
+	//	c.client.Stats(opts)
+	//	c.running = false
+	//}()
+
+	//go func() {
+	//	defer close(c.stream)
+	//	for s := range stats {
+	//		c.ReadCPU(s)
+	//		c.ReadMem(s)
+	//		c.ReadNet(s)
+	//		c.ReadIO(s)
+	//		c.stream <- c.Metrics
+	//	}
+	//	log.Infof("collector stopped for container: %s", c.id)
+	//}()
+
+	//c.running = true
+	//log.Infof("collector started for container: %s", c.id)
+}
+
+func (c *Kubernetes) Running() bool {
+	return c.running
+}
+
+func (c *Kubernetes) Stream() chan models.Metrics {
+	return c.stream
+}
+
+func (c *Kubernetes) Logs() LogCollector {
+	return NewKubernetesLogs(c.name, c.client)
+}
+
+// Stop collector
+func (c *Kubernetes) Stop() {
+	c.done <- true
+}
+
+//
+//func (c *Kubernetes) ReadCPU(stats *api.Stats) {
+//	ncpus := float64(len(stats.CPUStats.CPUUsage.PercpuUsage))
+//	total := float64(stats.CPUStats.CPUUsage.TotalUsage)
+//	system := float64(stats.CPUStats.SystemCPUUsage)
+//
+//	cpudiff := total - c.lastCpu
+//	syscpudiff := system - c.lastSysCpu
+//
+//	if c.scaleCpu {
+//		c.CPUUtil = round((cpudiff / syscpudiff * 100))
+//	} else {
+//		c.CPUUtil = round((cpudiff / syscpudiff * 100) * ncpus)
+//	}
+//	c.lastCpu = total
+//	c.lastSysCpu = system
+//	c.Pids = int(stats.PidsStats.Current)
+//}
+
+//func (c *Kubernetes) ReadMem(stats *api.Stats) {
+//	c.MemUsage = int64(stats.MemoryStats.Usage - stats.MemoryStats.Stats.Cache)
+//	c.MemLimit = int64(stats.MemoryStats.Limit)
+//	c.MemPercent = percent(float64(c.MemUsage), float64(c.MemLimit))
+//}
+
+//func (c *Kubernetes) ReadNet(stats *api.Stats) {
+//	var rx, tx int64
+//	for _, network := range stats.Networks {
+//		rx += int64(network.RxBytes)
+//		tx += int64(network.TxBytes)
+//	}
+//	c.NetRx, c.NetTx = rx, tx
+//}
+//
+//func (c *Kubernetes) ReadIO(stats *api.Stats) {
+//	var read, write int64
+//	for _, blk := range stats.BlkioStats.IOServiceBytesRecursive {
+//		if blk.Op == "Read" {
+//			read = int64(blk.Value)
+//		}
+//		if blk.Op == "Write" {
+//			write = int64(blk.Value)
+//		}
+//	}
+//	c.IOBytesRead, c.IOBytesWrite = read, write
+//}