Basic workers stats

isolate/conn_handler.go

@@ -220,7 +220,8 @@ func (r *responseStream) close(ctx context.Context) {
  }
 }
 
-func (r *responseStream) Write(ctx context.Context, num uint64, data []byte) error {
+// Writes messagepacked payload `as is` as a packet of Cocaine
+func (r *responseStream) WriteMessage(ctx context.Context, num uint64, packedPayload []byte) error {
  r.Lock()
  defer r.Unlock()
 
@@ -238,7 +239,7 @@ func (r *responseStream) Write(ctx context.Context, num uint64, data []byte) err
  p = msgp.AppendUint64(p, num)
 
  p = msgp.AppendArrayHeader(p, 1)
- p = msgp.AppendStringFromBytes(p, data)
+ p = append(p, packedPayload...)
 
  if _, err := r.wr.Write(p); err != nil {
  log.G(r.ctx).WithError(err).Error("responseStream.Write")
@@ -247,6 +248,15 @@ func (r *responseStream) Write(ctx context.Context, num uint64, data []byte) err
  return nil
 }
 
+// IMHO: should be named WriteString, really.
+func (r *responseStream) Write(ctx context.Context, num uint64, str []byte) error {
+ p := msgpackBytePool.Get().([]byte)[:0]
+ defer msgpackBytePool.Put(p)
+
+ p = msgp.AppendStringFromBytes(p, str)
+ return r.WriteMessage(ctx, num, p)
+}
+
 func (r *responseStream) Error(ctx context.Context, num uint64, code [2]int, msg string) error {
  r.Lock()
  defer r.Unlock()

isolate/container_metrics.go

@@ -0,0 +1,38 @@
+//go:generate msgp --tests=false
+//msgp:ignore isolate.MarkedWorkerMetrics
+package isolate
+
+type (
+ NetStat struct {
+ RxBytes uint64 `msg:"rx_bytes"`
+ TxBytes uint64 `msg:"tx_bytes"`
+ }
+
+ WorkerMetrics struct {
+ UptimeSec uint64 `msg:"uptime"`
+ CpuUsageSec uint64 `msg:"cpu_usage"`
+
+ CpuLoad float32 `msg:"cpu_load"`
+
+ Mem uint64 `msg:"mem"`
+
+ // iface -> net stat
+ Net map[string]NetStat `msg:"net"`
+ }
+
+ MetricsResponse map[string]*WorkerMetrics
+
+ MarkedWorkerMetrics struct {
+ uuid string
+ m *WorkerMetrics
+ }
+)
+
+func NewWorkerMetrics() (c WorkerMetrics) {
+ c.Net = make(map[string]NetStat)
+ return
+}
+
+func NewMarkedMetrics(uuid string, cm *WorkerMetrics) MarkedWorkerMetrics {
+ return MarkedWorkerMetrics{uuid: uuid, m: cm}
+}