mirror of https://github.com/bcicen/ctop.git
77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/bcicen/ctop/cwidgets"
|
|
"github.com/bcicen/ctop/cwidgets/compact"
|
|
"github.com/bcicen/ctop/metrics"
|
|
)
|
|
|
|
// Metrics and metadata representing a container
|
|
type Container struct {
|
|
metrics.Metrics
|
|
Id string
|
|
Meta map[string]string
|
|
Widgets *compact.Compact
|
|
updater cwidgets.WidgetUpdater
|
|
collector metrics.Collector
|
|
display bool // display this container in compact view
|
|
}
|
|
|
|
func NewContainer(id string, collector metrics.Collector) *Container {
|
|
widgets := compact.NewCompact(id)
|
|
return &Container{
|
|
Metrics: metrics.NewMetrics(),
|
|
Id: id,
|
|
Meta: make(map[string]string),
|
|
Widgets: widgets,
|
|
updater: widgets,
|
|
collector: collector,
|
|
}
|
|
}
|
|
|
|
func (c *Container) SetUpdater(u cwidgets.WidgetUpdater) {
|
|
c.updater = u
|
|
for k, v := range c.Meta {
|
|
c.updater.SetMeta(k, v)
|
|
}
|
|
}
|
|
|
|
func (c *Container) SetMeta(k, v string) {
|
|
c.Meta[k] = v
|
|
c.updater.SetMeta(k, v)
|
|
}
|
|
|
|
func (c *Container) GetMeta(k string) string {
|
|
if v, ok := c.Meta[k]; ok {
|
|
return v
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (c *Container) SetState(s string) {
|
|
c.SetMeta("state", s)
|
|
// start collector, if needed
|
|
if s == "running" && !c.collector.Running() {
|
|
c.collector.Start()
|
|
c.Read(c.collector.Stream())
|
|
}
|
|
// stop collector, if needed
|
|
if s != "running" && c.collector.Running() {
|
|
c.collector.Stop()
|
|
}
|
|
}
|
|
|
|
// Read metric stream, updating widgets
|
|
func (c *Container) Read(stream chan metrics.Metrics) {
|
|
go func() {
|
|
for metrics := range stream {
|
|
c.Metrics = metrics
|
|
c.updater.SetMetrics(metrics)
|
|
}
|
|
log.Infof("reader stopped for container: %s", c.Id)
|
|
c.Metrics = metrics.NewMetrics()
|
|
c.Widgets.Reset()
|
|
}()
|
|
log.Infof("reader started for container: %s", c.Id)
|
|
}
|