You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package session
|
|
|
|
import (
|
|
"github.com/containrrr/watchtower/pkg/types"
|
|
)
|
|
|
|
// Progress contains the current session container status
|
|
type Progress map[types.ContainerID]*ContainerStatus
|
|
|
|
// UpdateFromContainer sets various status fields from their corresponding container equivalents
|
|
func UpdateFromContainer(cont types.Container, newImage types.ImageID, state State) *ContainerStatus {
|
|
return &ContainerStatus{
|
|
ID: cont.ID(),
|
|
Name: cont.Name(),
|
|
ImageName: cont.ImageName(),
|
|
OldImageID: cont.SafeImageID(),
|
|
NewImageID: newImage,
|
|
State: state,
|
|
}
|
|
}
|
|
|
|
// AddSkipped adds a container to the Progress with the state set as skipped
|
|
func (m Progress) AddSkipped(cont types.Container, err error) {
|
|
update := UpdateFromContainer(cont, cont.SafeImageID(), SkippedState)
|
|
update.Error = err
|
|
m.Add(update)
|
|
}
|
|
|
|
// AddScanned adds a container to the Progress with the state set as scanned
|
|
func (m Progress) AddScanned(cont types.Container, newImage types.ImageID) {
|
|
m.Add(UpdateFromContainer(cont, newImage, ScannedState))
|
|
}
|
|
|
|
// UpdateFailed updates the containers passed, setting their state as failed with the supplied error
|
|
func (m Progress) UpdateFailed(failures map[types.ContainerID]error) {
|
|
for id, err := range failures {
|
|
update := m[id]
|
|
update.Error = err
|
|
update.State = FailedState
|
|
}
|
|
}
|
|
|
|
// Add a container to the map using container ID as the key
|
|
func (m Progress) Add(update *ContainerStatus) {
|
|
m[update.ID] = update
|
|
}
|
|
|
|
// MarkForUpdate marks the container identified by containerID for update
|
|
func (m Progress) MarkForUpdate(containerID types.ContainerID) {
|
|
m[containerID].State = UpdatedState
|
|
}
|