mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-04 10:41:14 +00:00
71f9d8874f
Comment before exported functions. When `if` block ends with a return statement, drop the else and outdent its block.
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// State is the container internal state
|
|
type State int
|
|
|
|
const (
|
|
// Created state
|
|
Created State = iota
|
|
// Running state
|
|
Running
|
|
// Stopped state
|
|
Stopped
|
|
// Destroyed state
|
|
Destroyed
|
|
)
|
|
|
|
// Container data structure
|
|
type Container struct {
|
|
State State
|
|
PID int
|
|
}
|
|
|
|
// Store data structure
|
|
type Store struct {
|
|
lock sync.Mutex
|
|
containers map[string]Container
|
|
}
|
|
|
|
// NewStore instantiates a new Store
|
|
func NewStore() *Store {
|
|
return &Store{
|
|
containers: map[string]Container{},
|
|
}
|
|
}
|
|
|
|
// Container returns a container form its ID
|
|
func (s *Store) Container(containerID string) (Container, bool) {
|
|
s.lock.Lock()
|
|
defer s.lock.Unlock()
|
|
container, found := s.containers[containerID]
|
|
return container, found
|
|
}
|
|
|
|
// SetContainer sets a container into the store
|
|
func (s *Store) SetContainer(containerID string, container Container) {
|
|
s.lock.Lock()
|
|
defer s.lock.Unlock()
|
|
s.containers[containerID] = container
|
|
}
|
|
|
|
// DeleteContainer deletes a container from the store
|
|
func (s *Store) DeleteContainer(containerID string) {
|
|
s.lock.Lock()
|
|
defer s.lock.Unlock()
|
|
delete(s.containers, containerID)
|
|
}
|
|
|
|
// ForEach execute a function on each container in the store
|
|
func (s *Store) ForEach(callback func(ID string, c Container)) {
|
|
s.lock.Lock()
|
|
defer s.lock.Unlock()
|
|
for containerID, container := range s.containers {
|
|
callback(containerID, container)
|
|
}
|
|
}
|