forked from aws/amazon-ecs-agent
/
engine.go
674 lines (572 loc) · 23.6 KB
/
engine.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
// Copyright 2014-2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may
// not use this file except in compliance with the License. A copy of the
// License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file is distributed
// on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
// express or implied. See the License for the specific language governing
// permissions and limitations under the License.
package stats
//go:generate go run ../../scripts/generate/mockgen.go github.com/aws/amazon-ecs-agent/agent/stats Engine mock/$GOFILE
import (
"context"
"fmt"
"sync"
"time"
"github.com/cihub/seelog"
docker "github.com/fsouza/go-dockerclient"
"github.com/pborman/uuid"
"github.com/pkg/errors"
"github.com/aws/amazon-ecs-agent/agent/api"
"github.com/aws/amazon-ecs-agent/agent/config"
"github.com/aws/amazon-ecs-agent/agent/dockerclient"
"github.com/aws/amazon-ecs-agent/agent/dockerclient/dockerapi"
ecsengine "github.com/aws/amazon-ecs-agent/agent/engine"
"github.com/aws/amazon-ecs-agent/agent/eventstream"
"github.com/aws/amazon-ecs-agent/agent/stats/resolver"
"github.com/aws/amazon-ecs-agent/agent/tcs/model/ecstcs"
"github.com/aws/aws-sdk-go/aws"
)
const (
containerChangeHandler = "DockerStatsEngineDockerEventsHandler"
listContainersTimeout = 10 * time.Minute
queueResetThreshold = 2 * dockerapi.StatsInactivityTimeout
)
var (
// EmptyMetricsError indicates an error for a task when there are no container
// metrics to report
EmptyMetricsError = errors.New("stats engine: no task metrics to report")
// EmptyHealthMetricsError indicates an error for a task when there are no container
// health metrics to report
EmptyHealthMetricsError = errors.New("stats engine: no task health metrics to report")
)
// DockerContainerMetadataResolver implements ContainerMetadataResolver for
// DockerTaskEngine.
type DockerContainerMetadataResolver struct {
dockerTaskEngine *ecsengine.DockerTaskEngine
}
// Engine defines methods to be implemented by the engine struct. It is
// defined to make testing easier.
type Engine interface {
GetInstanceMetrics() (*ecstcs.MetricsMetadata, []*ecstcs.TaskMetric, error)
ContainerDockerStats(taskARN string, containerID string) (*docker.Stats, error)
GetTaskHealthMetrics() (*ecstcs.HealthMetadata, []*ecstcs.TaskHealth, error)
}
// DockerStatsEngine is used to monitor docker container events and to report
// utlization metrics of the same.
type DockerStatsEngine struct {
ctx context.Context
stopEngine context.CancelFunc
client dockerapi.DockerClient
cluster string
containerInstanceArn string
lock sync.RWMutex
disableMetrics bool
containerChangeEventStream *eventstream.EventStream
resolver resolver.ContainerMetadataResolver
// tasksToContainers maps task arns to a map of container ids to StatsContainer objects.
tasksToContainers map[string]map[string]*StatsContainer
// tasksToHealthCheckContainers map task arns to the containers that has health check enabled
tasksToHealthCheckContainers map[string]map[string]*StatsContainer
// tasksToDefinitions maps task arns to task definition name and family metadata objects.
tasksToDefinitions map[string]*taskDefinition
}
// ResolveTask resolves the api task object, given container id.
func (resolver *DockerContainerMetadataResolver) ResolveTask(dockerID string) (*api.Task, error) {
if resolver.dockerTaskEngine == nil {
return nil, fmt.Errorf("Docker task engine uninitialized")
}
task, found := resolver.dockerTaskEngine.State().TaskByID(dockerID)
if !found {
return nil, fmt.Errorf("Could not map docker id to task: %s", dockerID)
}
return task, nil
}
// ResolveContainer resolves the api container object, given container id.
func (resolver *DockerContainerMetadataResolver) ResolveContainer(dockerID string) (*api.DockerContainer, error) {
if resolver.dockerTaskEngine == nil {
return nil, fmt.Errorf("Docker task engine uninitialized")
}
container, found := resolver.dockerTaskEngine.State().ContainerByID(dockerID)
if !found {
return nil, fmt.Errorf("Could not map docker id to container: %s", dockerID)
}
return container, nil
}
// NewDockerStatsEngine creates a new instance of the DockerStatsEngine object.
// MustInit() must be called to initialize the fields of the new event listener.
func NewDockerStatsEngine(cfg *config.Config, client dockerapi.DockerClient, containerChangeEventStream *eventstream.EventStream) *DockerStatsEngine {
return &DockerStatsEngine{
client: client,
resolver: nil,
disableMetrics: cfg.DisableMetrics,
tasksToContainers: make(map[string]map[string]*StatsContainer),
tasksToHealthCheckContainers: make(map[string]map[string]*StatsContainer),
tasksToDefinitions: make(map[string]*taskDefinition),
containerChangeEventStream: containerChangeEventStream,
}
}
// synchronizeState goes through all the containers on the instance to synchronize the state on agent start
func (engine *DockerStatsEngine) synchronizeState() error {
listContainersResponse := engine.client.ListContainers(engine.ctx, false, dockerclient.ListContainersTimeout)
if listContainersResponse.Error != nil {
return listContainersResponse.Error
}
for _, containerID := range listContainersResponse.DockerIDs {
engine.addAndStartStatsContainer(containerID)
}
return nil
}
// addAndStartStatsContainer add the container into stats engine and start collecting the container stats
func (engine *DockerStatsEngine) addAndStartStatsContainer(containerID string) {
engine.lock.Lock()
defer engine.lock.Unlock()
statsContainer, err := engine.addContainerUnsafe(containerID)
if err != nil {
seelog.Debugf("Adding container to stats watch list failed, container: %s, err: %v", containerID, err)
return
}
if engine.disableMetrics || statsContainer == nil {
return
}
statsContainer.StartStatsCollection()
}
// MustInit initializes fields of the DockerStatsEngine object.
func (engine *DockerStatsEngine) MustInit(ctx context.Context, taskEngine ecsengine.TaskEngine, cluster string, containerInstanceArn string) error {
derivedCtx, cancel := context.WithCancel(ctx)
engine.stopEngine = cancel
engine.ctx = derivedCtx
// TODO ensure that this is done only once per engine object
seelog.Info("Initializing stats engine")
engine.cluster = cluster
engine.containerInstanceArn = containerInstanceArn
var err error
engine.resolver, err = newDockerContainerMetadataResolver(taskEngine)
if err != nil {
return err
}
// Subscribe to the container change event stream
err = engine.containerChangeEventStream.Subscribe(containerChangeHandler, engine.handleDockerEvents)
if err != nil {
return fmt.Errorf("Failed to subscribe to container change event stream, err %v", err)
}
err = engine.synchronizeState()
if err != nil {
seelog.Warnf("Synchronize the container state failed, err: %v", err)
}
go engine.waitToStop()
return nil
}
// Shutdown cleans up the resources after the statas engine.
func (engine *DockerStatsEngine) Shutdown() {
engine.stopEngine()
engine.Disable()
}
// Disable prevents this engine from managing any additional tasks.
func (engine *DockerStatsEngine) Disable() {
engine.lock.Lock()
}
// waitToStop waits for the container change event stream close ans stop collection metrics
func (engine *DockerStatsEngine) waitToStop() {
// Waiting for the event stream to close
ctx := engine.containerChangeEventStream.Context()
select {
case <-ctx.Done():
seelog.Debug("Event stream closed, stop listening to the event stream")
engine.containerChangeEventStream.Unsubscribe(containerChangeHandler)
engine.removeAll()
}
}
// removeAll stops the periodic usage data collection for all containers
func (engine *DockerStatsEngine) removeAll() {
engine.lock.Lock()
defer engine.lock.Unlock()
for task, containers := range engine.tasksToContainers {
for _, statsContainer := range containers {
statsContainer.StopStatsCollection()
}
delete(engine.tasksToContainers, task)
}
for task := range engine.tasksToHealthCheckContainers {
delete(engine.tasksToContainers, task)
}
}
// addContainerUnsafe adds a container to the map of containers being watched.
func (engine *DockerStatsEngine) addContainerUnsafe(dockerID string) (*StatsContainer, error) {
// Make sure that this container belongs to a task and that the task
// is not terminal.
task, err := engine.resolver.ResolveTask(dockerID)
if err != nil {
return nil, errors.Wrapf(err, "could not map container to task, ignoring container: %s", dockerID)
}
if len(task.Arn) == 0 || len(task.Family) == 0 {
return nil, errors.Errorf("stats add container: invalid task fields, arn: %s, familiy: %s", task.Arn, task.Family)
}
if task.GetKnownStatus().Terminal() {
return nil, errors.Errorf("stats add container: task is terminal, ignoring container: %s, task: %s", dockerID, task.Arn)
}
seelog.Debugf("Adding container to stats watch list, id: %s, task: %s", dockerID, task.Arn)
statsContainer := newStatsContainer(dockerID, engine.client, engine.resolver)
engine.tasksToDefinitions[task.Arn] = &taskDefinition{family: task.Family, version: task.Version}
watchStatsContainer := false
if !engine.disableMetrics {
// Adding container to the map for collecting stats
watchStatsContainer = engine.addToStatsContainerMapUnsafe(task.Arn, dockerID, statsContainer, engine.containerMetricsMapUnsafe)
}
if dockerContainer, err := engine.resolver.ResolveContainer(dockerID); err != nil {
seelog.Debugf("Could not map container ID to container, container: %s, err: %s", dockerID, err)
} else if dockerContainer.Container.HealthStatusShouldBeReported() {
// Track the container health status
engine.addToStatsContainerMapUnsafe(task.Arn, dockerID, statsContainer, engine.healthCheckContainerMapUnsafe)
seelog.Debugf("Adding container to stats health check watch list, id: %s, task: %s", dockerID, task.Arn)
}
if !watchStatsContainer {
return nil, nil
}
return statsContainer, nil
}
func (engine *DockerStatsEngine) containerMetricsMapUnsafe() map[string]map[string]*StatsContainer {
return engine.tasksToContainers
}
func (engine *DockerStatsEngine) healthCheckContainerMapUnsafe() map[string]map[string]*StatsContainer {
return engine.tasksToHealthCheckContainers
}
// addToStatsContainerMapUnsafe adds the statscontainer into stats for tracking and returns a boolean indicates
// whether this container should be tracked for collecting metrics
func (engine *DockerStatsEngine) addToStatsContainerMapUnsafe(
taskARN, containerID string,
statsContainer *StatsContainer,
statsMapToUpdate func() map[string]map[string]*StatsContainer) bool {
taskToContainerMap := statsMapToUpdate()
// Check if this container is already being watched.
_, taskExists := taskToContainerMap[taskARN]
if taskExists {
// task arn exists in map.
_, containerExists := taskToContainerMap[taskARN][containerID]
if containerExists {
// container arn exists in map.
seelog.Debugf("Container already being watched, ignoring, id: %s", containerID)
return false
}
} else {
// Create a map for the task arn if it doesn't exist yet.
taskToContainerMap[taskARN] = make(map[string]*StatsContainer)
}
taskToContainerMap[taskARN][containerID] = statsContainer
return true
}
// GetInstanceMetrics gets all task metrics and instance metadata from stats engine.
func (engine *DockerStatsEngine) GetInstanceMetrics() (*ecstcs.MetricsMetadata, []*ecstcs.TaskMetric, error) {
var taskMetrics []*ecstcs.TaskMetric
idle := engine.isIdle()
metricsMetadata := &ecstcs.MetricsMetadata{
Cluster: aws.String(engine.cluster),
ContainerInstance: aws.String(engine.containerInstanceArn),
Idle: aws.Bool(idle),
MessageId: aws.String(uuid.NewRandom().String()),
}
if idle {
seelog.Debug("Instance is idle. No task metrics to report")
fin := true
metricsMetadata.Fin = &fin
return metricsMetadata, taskMetrics, nil
}
engine.lock.Lock()
defer engine.lock.Unlock()
for taskArn := range engine.tasksToContainers {
containerMetrics, err := engine.taskContainerMetricsUnsafe(taskArn)
if err != nil {
seelog.Debugf("Error getting container metrics for task: %s, err: %v", taskArn, err)
continue
}
if len(containerMetrics) == 0 {
seelog.Debugf("Empty containerMetrics for task, ignoring, task: %s", taskArn)
continue
}
taskDef, exists := engine.tasksToDefinitions[taskArn]
if !exists {
seelog.Debugf("Could not map task to definition, task: %s", taskArn)
continue
}
metricTaskArn := taskArn
taskMetric := &ecstcs.TaskMetric{
TaskArn: &metricTaskArn,
TaskDefinitionFamily: &taskDef.family,
TaskDefinitionVersion: &taskDef.version,
ContainerMetrics: containerMetrics,
}
taskMetrics = append(taskMetrics, taskMetric)
}
if len(taskMetrics) == 0 {
// Not idle. Expect taskMetrics to be there.
return nil, nil, EmptyMetricsError
}
// Reset current stats. Retaining older stats results in incorrect utilization stats
// until they are removed from the queue.
engine.resetStatsUnsafe()
return metricsMetadata, taskMetrics, nil
}
// GetTaskHealthMetrics returns the container health metrics
func (engine *DockerStatsEngine) GetTaskHealthMetrics() (*ecstcs.HealthMetadata, []*ecstcs.TaskHealth, error) {
var taskHealths []*ecstcs.TaskHealth
metadata := &ecstcs.HealthMetadata{
Cluster: aws.String(engine.cluster),
ContainerInstance: aws.String(engine.containerInstanceArn),
MessageId: aws.String(uuid.NewRandom().String()),
}
if !engine.containerHealthsToMonitor() {
return metadata, taskHealths, nil
}
engine.lock.RLock()
defer engine.lock.RUnlock()
for taskARN := range engine.tasksToHealthCheckContainers {
taskHealth := engine.getTaskHealthUnsafe(taskARN)
if taskHealth == nil {
continue
}
taskHealths = append(taskHealths, taskHealth)
}
if len(taskHealths) == 0 {
return nil, nil, EmptyHealthMetricsError
}
return metadata, taskHealths, nil
}
func (engine *DockerStatsEngine) isIdle() bool {
engine.lock.RLock()
defer engine.lock.RUnlock()
return len(engine.tasksToContainers) == 0
}
func (engine *DockerStatsEngine) containerHealthsToMonitor() bool {
engine.lock.RLock()
defer engine.lock.RUnlock()
return len(engine.tasksToHealthCheckContainers) != 0
}
// stopTrackingContainerUnsafe removes the StatsContainer from stats engine and
// returns true if the container is stopped or no longer tracked in agent. Otherwise
// it does nothing and return false
func (engine *DockerStatsEngine) stopTrackingContainerUnsafe(container *StatsContainer, taskARN string) bool {
terminal, err := container.terminal()
if err != nil {
// Error determining if the container is terminal. This means that the container
// id could not be resolved to a container that is being tracked by the
// docker task engine. If the docker task engine has already removed
// the container from its state, there's no point in stats engine tracking the
// container. So, clean-up anyway.
seelog.Warnf("Error determining if the container %s is terminal, removing from stats, err: %v", container.containerMetadata.DockerID, err)
engine.doRemoveContainerUnsafe(container, taskARN)
return true
}
if terminal {
// Container is in known terminal state. Stop collection metrics.
seelog.Infof("Container %s is terminal, removing from stats", container.containerMetadata.DockerID)
engine.doRemoveContainerUnsafe(container, taskARN)
return true
}
return false
}
func (engine *DockerStatsEngine) getTaskHealthUnsafe(taskARN string) *ecstcs.TaskHealth {
// Acquire the task definition information
taskDefinition, ok := engine.tasksToDefinitions[taskARN]
if !ok {
seelog.Debugf("Could not map task to definitions, task: %s", taskARN)
return nil
}
// Check all the stats container for the task
containers, ok := engine.tasksToHealthCheckContainers[taskARN]
if !ok {
seelog.Debugf("Could not map task to health containers, task: %s", taskARN)
return nil
}
// Aggregate container health information for all the containers in the task
var containerHealths []*ecstcs.ContainerHealth
for _, container := range containers {
// check if the container is stopped/untracked, and remove it from stats
//engine if needed
if engine.stopTrackingContainerUnsafe(container, taskARN) {
continue
}
dockerContainer, err := engine.resolver.ResolveContainer(container.containerMetadata.DockerID)
if err != nil {
seelog.Debugf("Could not resolve the Docker ID in agent state: %s", container.containerMetadata.DockerID)
continue
}
// Check if the container has health check enabled
if !dockerContainer.Container.HealthStatusShouldBeReported() {
continue
}
healthInfo := dockerContainer.Container.GetHealthStatus()
if healthInfo.Since == nil {
// container was started but the health status isn't ready
healthInfo.Since = aws.Time(time.Now())
}
containerHealth := &ecstcs.ContainerHealth{
ContainerName: aws.String(dockerContainer.Container.Name),
HealthStatus: aws.String(healthInfo.Status.BackendStatus()),
StatusSince: aws.Time(healthInfo.Since.UTC()),
}
containerHealths = append(containerHealths, containerHealth)
}
if len(containerHealths) == 0 {
return nil
}
taskHealth := &ecstcs.TaskHealth{
Containers: containerHealths,
TaskArn: aws.String(taskARN),
TaskDefinitionFamily: aws.String(taskDefinition.family),
TaskDefinitionVersion: aws.String(taskDefinition.version),
}
return taskHealth
}
// handleDockerEvents must be called after openEventstream; it processes each
// event that it reads from the docker event stream.
func (engine *DockerStatsEngine) handleDockerEvents(events ...interface{}) error {
for _, event := range events {
dockerContainerChangeEvent, ok := event.(dockerapi.DockerContainerChangeEvent)
if !ok {
return fmt.Errorf("Unexpected event received, expected docker container change event")
}
switch dockerContainerChangeEvent.Status {
case api.ContainerRunning:
engine.addAndStartStatsContainer(dockerContainerChangeEvent.DockerID)
case api.ContainerStopped:
engine.removeContainer(dockerContainerChangeEvent.DockerID)
default:
seelog.Debugf("Ignoring event for container, id: %s, status: %d", dockerContainerChangeEvent.DockerID, dockerContainerChangeEvent.Status)
}
}
return nil
}
// removeContainer deletes the container from the map of containers being watched.
// It also stops the periodic usage data collection for the container.
func (engine *DockerStatsEngine) removeContainer(dockerID string) {
engine.lock.Lock()
defer engine.lock.Unlock()
// Make sure that this container belongs to a task.
task, err := engine.resolver.ResolveTask(dockerID)
if err != nil {
seelog.Debugf("Could not map container to task, ignoring, err: %v, id: %s", err, dockerID)
return
}
_, taskExists := engine.tasksToContainers[task.Arn]
if !taskExists {
seelog.Debugf("Container not being watched, id: %s", dockerID)
return
}
// task arn exists in map.
container, containerExists := engine.tasksToContainers[task.Arn][dockerID]
if !containerExists {
// container arn does not exist in map.
seelog.Debugf("Container not being watched, id: %s", dockerID)
return
}
engine.doRemoveContainerUnsafe(container, task.Arn)
}
// newDockerContainerMetadataResolver returns a new instance of DockerContainerMetadataResolver.
func newDockerContainerMetadataResolver(taskEngine ecsengine.TaskEngine) (*DockerContainerMetadataResolver, error) {
dockerTaskEngine, ok := taskEngine.(*ecsengine.DockerTaskEngine)
if !ok {
// Error type casting docker task engine.
return nil, fmt.Errorf("Could not load docker task engine")
}
resolver := &DockerContainerMetadataResolver{
dockerTaskEngine: dockerTaskEngine,
}
return resolver, nil
}
// taskContainerMetricsUnsafe gets all container metrics for a task arn.
func (engine *DockerStatsEngine) taskContainerMetricsUnsafe(taskArn string) ([]*ecstcs.ContainerMetric, error) {
containerMap, taskExists := engine.tasksToContainers[taskArn]
if !taskExists {
return nil, fmt.Errorf("Task not found")
}
var containerMetrics []*ecstcs.ContainerMetric
for _, container := range containerMap {
dockerID := container.containerMetadata.DockerID
// Check if the container is terminal. If it is, make sure that it is
// cleaned up properly. We might sometimes miss events from docker task
// engine and this helps in reconciling the state. The tcs client's
// GetInstanceMetrics probe is used as the trigger for this.
if engine.stopTrackingContainerUnsafe(container, taskArn) {
continue
}
if !container.statsQueue.enoughDatapointsInBuffer() &&
!container.statsQueue.resetThresholdElapsed(queueResetThreshold) {
seelog.Debugf("Stats not ready for container %s", dockerID)
continue
}
// Container is not terminal. Get CPU stats set.
cpuStatsSet, err := container.statsQueue.GetCPUStatsSet()
if err != nil {
seelog.Warnf("Error getting cpu stats, err: %v, container: %v", err, dockerID)
continue
}
// Get memory stats set.
memoryStatsSet, err := container.statsQueue.GetMemoryStatsSet()
if err != nil {
seelog.Warnf("Error getting memory stats, err: %v, container: %v", err, dockerID)
continue
}
containerMetrics = append(containerMetrics, &ecstcs.ContainerMetric{
CpuStatsSet: cpuStatsSet,
MemoryStatsSet: memoryStatsSet,
})
}
return containerMetrics, nil
}
func (engine *DockerStatsEngine) doRemoveContainerUnsafe(container *StatsContainer, taskArn string) {
container.StopStatsCollection()
dockerID := container.containerMetadata.DockerID
delete(engine.tasksToContainers[taskArn], dockerID)
seelog.Debugf("Deleted container from tasks, id: %s", dockerID)
if len(engine.tasksToContainers[taskArn]) == 0 {
// No containers in task, delete task arn from map.
delete(engine.tasksToContainers, taskArn)
// No need to verify if the key exists in tasksToDefinitions.
// Delete will do nothing if the specified key doesn't exist.
delete(engine.tasksToDefinitions, taskArn)
seelog.Debugf("Deleted task from tasks, arn: %s", taskArn)
}
// Remove the container from health container watch list
if _, ok := engine.tasksToHealthCheckContainers[taskArn][dockerID]; !ok {
return
}
delete(engine.tasksToHealthCheckContainers[taskArn], dockerID)
if len(engine.tasksToHealthCheckContainers[taskArn]) == 0 {
delete(engine.tasksToHealthCheckContainers, taskArn)
seelog.Debugf("Deleted task from container health watch list, arn: %s", taskArn)
}
}
// resetStatsUnsafe resets stats for all watched containers.
func (engine *DockerStatsEngine) resetStatsUnsafe() {
for _, containerMap := range engine.tasksToContainers {
for _, container := range containerMap {
container.statsQueue.Reset()
}
}
}
// ContainerDockerStats returns the last stored raw docker stats object for a container
func (engine *DockerStatsEngine) ContainerDockerStats(taskARN string, containerID string) (*docker.Stats, error) {
engine.lock.RLock()
defer engine.lock.RUnlock()
containerIDToStatsContainer, ok := engine.tasksToContainers[taskARN]
if !ok {
return nil, errors.Errorf("stats engine: task '%s' for container '%s' not found",
taskARN, containerID)
}
container, ok := containerIDToStatsContainer[containerID]
if !ok {
return nil, errors.Errorf("stats engine: container not found: %s", containerID)
}
return container.statsQueue.GetLastStat(), nil
}
// newMetricsMetadata creates the singleton metadata object.
func newMetricsMetadata(cluster *string, containerInstance *string) *ecstcs.MetricsMetadata {
return &ecstcs.MetricsMetadata{
Cluster: cluster,
ContainerInstance: containerInstance,
}
}