mirror of
https://github.com/bcicen/ctop.git
synced 2025-12-06 15:16:41 +08:00
129 lines
2.8 KiB
Go
129 lines
2.8 KiB
Go
package collector
|
|
|
|
import (
|
|
"time"
|
|
|
|
"k8s.io/metrics/pkg/apis/metrics/v1alpha1"
|
|
"k8s.io/metrics/pkg/client/clientset_generated/clientset"
|
|
|
|
"github.com/bcicen/ctop/config"
|
|
"github.com/bcicen/ctop/models"
|
|
"k8s.io/api/core/v1"
|
|
|
|
"k8s.io/client-go/kubernetes"
|
|
)
|
|
|
|
// Kubernetes collector
|
|
type Kubernetes struct {
|
|
models.Metrics
|
|
name string
|
|
client clientset.Interface
|
|
clientset *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: clientset.New(client.RESTClient()),
|
|
clientset: client,
|
|
scaleCpu: config.GetSwitchVal("scaleCpu"),
|
|
}
|
|
}
|
|
|
|
func (k *Kubernetes) Start() {
|
|
k.done = make(chan bool)
|
|
k.stream = make(chan models.Metrics)
|
|
|
|
go func() {
|
|
k.running = false
|
|
for {
|
|
|
|
result := &v1alpha1.PodMetrics{}
|
|
err := k.clientset.RESTClient().Get().AbsPath("/api/v1/namespaces/kube-system/services/http:heapster:/proxy/apis/metrics/v1alpha1/namespaces/" + config.GetVal("namespace") + "/pods/" + k.name).Do().Into(result)
|
|
|
|
if err != nil {
|
|
log.Errorf("has error %s here %s", k.name, err.Error())
|
|
time.Sleep(1 * time.Second)
|
|
continue
|
|
}
|
|
k.ReadCPU(result)
|
|
k.ReadMem(result)
|
|
k.stream <- k.Metrics
|
|
}
|
|
}()
|
|
|
|
k.running = true
|
|
log.Infof("collector started for container: %s", k.name)
|
|
}
|
|
|
|
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.clientset)
|
|
}
|
|
|
|
// Stop collector
|
|
func (c *Kubernetes) Stop() {
|
|
c.done <- true
|
|
}
|
|
|
|
func (k *Kubernetes) ReadCPU(metrics *v1alpha1.PodMetrics) {
|
|
all := int64(0)
|
|
for _, c := range metrics.Containers {
|
|
v := c.Usage[v1.ResourceCPU]
|
|
all += v.Value()
|
|
}
|
|
if all != 0 {
|
|
k.CPUUtil = round(float64(all))
|
|
}
|
|
}
|
|
|
|
func (k *Kubernetes) ReadMem(metrics *v1alpha1.PodMetrics) {
|
|
all := int64(0)
|
|
for _, c := range metrics.Containers {
|
|
v := c.Usage[v1.ResourceMemory]
|
|
a, ok := v.AsInt64()
|
|
if ok {
|
|
all += a
|
|
}
|
|
}
|
|
k.MemUsage = all
|
|
k.MemLimit = int64(0)
|
|
//k.MemPercent = percent(float64(k.MemUsage), float64(k.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
|
|
//}
|