github/docker-labs
1
0
Fork 0
mirror of https://github.com/bingohuang/docker-labs.git synced 2025-10-05 18:03:21 +08:00
Code Issues Releases Wiki Activity

Periodic tasks refactor (#62)

Browse Source 
* Once every second the session run a list of periodic tasks on every
instance concurrently. We use these tasks to do things like:
- Collect mem and cpu stats
- Check if instance is part of a swarm cluster
- Broadcast information to connected clients
This commit is contained in:
Jonathan Leibiusky
2016-12-01 15:57:30 -03:00
committed by GitHub
parent 07fee4c1bf
commit afa47c0bfc
7 changed files with 178 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

95
services/instance.go
View File

@@ -2,8 +2,6 @@ package services
import (
"context"
"encoding/json"
"fmt"
"io"
"log"
"os"
@@ -12,20 +10,23 @@ import (
"golang.org/x/text/encoding"
"github.com/docker/docker/api/types"
"github.com/docker/docker/api/types/swarm"
units "github.com/docker/go-units"
"github.com/docker/docker/client"
)
var rw sync.Mutex
type Instance struct {
session *Session `json:"-"`
Name string `json:"name"`
Hostname string `json:"hostname"`
IP string `json:"ip"`
conn *types.HijackedResponse `json:"-"`
ctx context.Context `json:"-"`
statsReader io.ReadCloser `json:"-"`
session *Session `json:"-"`
Name string `json:"name"`
Hostname string `json:"hostname"`
IP string `json:"ip"`
conn *types.HijackedResponse `json:"-"`
ctx context.Context `json:"-"`
statsReader io.ReadCloser `json:"-"`
dockerClient *client.Client `json:"-"`
IsManager *bool `json:"is_manager"`
Mem string `json:"mem"`
Cpu string `json:"cpu"`
}
func (i *Instance) IsConnected() bool {
@@ -75,9 +76,6 @@ func NewInstance(session *Session) (*Instance, error) {
wsServer.BroadcastTo(session.Id, "new instance", instance.Name, instance.IP, instance.Hostname)
// Start collecting stats
go instance.CollectStats()
return instance, nil
}
@@ -90,75 +88,6 @@ func (s *sessionWriter) Write(p []byte) (n int, err error) {
return len(p), nil
}
func (o *Instance) CollectStats() {
reader, err := GetContainerStats(o.Name)
if err != nil {
log.Println("Error while trying to collect instance stats", err)
return
}
o.statsReader = reader
dec := json.NewDecoder(o.statsReader)
var (
mem = 0.0
memLimit = 0.0
memPercent = 0.0
v *types.StatsJSON
memFormatted = ""
cpuPercent = 0.0
previousCPU uint64
previousSystem uint64
cpuFormatted = ""
)
for {
e := dec.Decode(&v)
if e != nil {
break
}
var isManager *bool
if info, err := GetDaemonInfo(fmt.Sprintf("http://%s:2375", o.IP)); err == nil {
if info.Swarm.LocalNodeState != swarm.LocalNodeStateInactive && info.Swarm.LocalNodeState != swarm.LocalNodeStateLocked {
isManager = &info.Swarm.ControlAvailable
}
}
// Memory
if v.MemoryStats.Limit != 0 {
memPercent = float64(v.MemoryStats.Usage) / float64(v.MemoryStats.Limit) * 100.0
}
mem = float64(v.MemoryStats.Usage)
memLimit = float64(v.MemoryStats.Limit)
memFormatted = fmt.Sprintf("%.2f%% (%s / %s)", memPercent, units.BytesSize(mem), units.BytesSize(memLimit))
// cpu
previousCPU = v.PreCPUStats.CPUUsage.TotalUsage
previousSystem = v.PreCPUStats.SystemUsage
cpuPercent = calculateCPUPercentUnix(previousCPU, previousSystem, v)
cpuFormatted = fmt.Sprintf("%.2f%%", cpuPercent)
wsServer.BroadcastTo(o.session.Id, "instance stats", o.Name, memFormatted, cpuFormatted, isManager)
}
}
func calculateCPUPercentUnix(previousCPU, previousSystem uint64, v *types.StatsJSON) float64 {
var (
cpuPercent = 0.0
// calculate the change for the cpu usage of the container in between readings
cpuDelta = float64(v.CPUStats.CPUUsage.TotalUsage) - float64(previousCPU)
// calculate the change for the entire system between readings
systemDelta = float64(v.CPUStats.SystemUsage) - float64(previousSystem)
)
if systemDelta > 0.0 && cpuDelta > 0.0 {
cpuPercent = (cpuDelta / systemDelta) * float64(len(v.CPUStats.CPUUsage.PercpuUsage)) * 100.0
}
return cpuPercent
}
func (i *Instance) ResizeTerminal(cols, rows uint) error {
return ResizeConnection(i.Name, cols, rows)
}