mirror of https://github.com/portainer/portainer
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package cron
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/portainer/portainer"
|
|
)
|
|
|
|
// SnapshotTaskContext represents the context required for the execution
|
|
// of a SnapshotTask.
|
|
type SnapshotTaskContext struct {
|
|
EndpointService portainer.EndpointService
|
|
Snapshotter portainer.Snapshotter
|
|
}
|
|
|
|
// SnapshotTask represents a task used to create endpoint snapshots.
|
|
// It can be scheduled.
|
|
type SnapshotTask struct {
|
|
context *SnapshotTaskContext
|
|
}
|
|
|
|
// NewSnapshotTask creates a new ScriptTask using the specified context.
|
|
func NewSnapshotTask(context *SnapshotTaskContext) SnapshotTask {
|
|
return SnapshotTask{
|
|
context: context,
|
|
}
|
|
}
|
|
|
|
// Run triggers the execution of the task.
|
|
// It will iterate through all the endpoints available in the database to
|
|
// create a snapshot of each one of them.
|
|
func (task SnapshotTask) Run() {
|
|
endpoints, err := task.context.EndpointService.Endpoints()
|
|
if err != nil {
|
|
log.Printf("background task error (endpoint snapshot). Unable to retrieve endpoint list (err=%s)\n", err)
|
|
return
|
|
}
|
|
|
|
for _, endpoint := range endpoints {
|
|
if endpoint.Type == portainer.AzureEnvironment {
|
|
continue
|
|
}
|
|
|
|
snapshot, err := task.context.Snapshotter.CreateSnapshot(&endpoint)
|
|
endpoint.Status = portainer.EndpointStatusUp
|
|
if err != nil {
|
|
log.Printf("background task error (endpoint snapshot). Unable to create snapshot (endpoint=%s, URL=%s) (err=%s)\n", endpoint.Name, endpoint.URL, err)
|
|
endpoint.Status = portainer.EndpointStatusDown
|
|
}
|
|
|
|
if snapshot != nil {
|
|
endpoint.Snapshots = []portainer.Snapshot{*snapshot}
|
|
}
|
|
|
|
err = task.context.EndpointService.UpdateEndpoint(endpoint.ID, &endpoint)
|
|
if err != nil {
|
|
log.Printf("background task error (endpoint snapshot). Unable to update endpoint (endpoint=%s, URL=%s) (err=%s)\n", endpoint.Name, endpoint.URL, err)
|
|
return
|
|
}
|
|
}
|
|
}
|