mirror of
https://github.com/openfaas/faasd.git
synced 2025-06-11 01:16:47 +00:00
969fc566e1
Fixes a bug when attempting to access a non-existant IP from GetIPfromPID called via the list API. Renames the provider from faas-containerd Updates function deletion grace period to 30s to prevent any errors in the REST API during a long-running deletion. Tested on Linux with the figlet function which by default takes around 5s to delete due to its write_timeout value, the deletion now blocks rather than throwing an error. Signed-off-by: Alex Ellis (OpenFaaS Ltd) <alexellis2@gmail.com>
122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/containerd/containerd"
|
|
"github.com/containerd/containerd/errdefs"
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
// Remove removes a container
|
|
func Remove(ctx context.Context, client *containerd.Client, name string) error {
|
|
|
|
container, containerErr := client.LoadContainer(ctx, name)
|
|
|
|
if containerErr == nil {
|
|
found := true
|
|
t, err := container.Task(ctx, nil)
|
|
if err != nil {
|
|
if errdefs.IsNotFound(err) {
|
|
found = false
|
|
} else {
|
|
return fmt.Errorf("unable to get task %s: ", err)
|
|
}
|
|
}
|
|
|
|
if found {
|
|
status, _ := t.Status(ctx)
|
|
fmt.Printf("Status of %s is: %s\n", name, status.Status)
|
|
|
|
log.Printf("Need to kill %s\n", name)
|
|
err := killTask(ctx, t)
|
|
if err != nil {
|
|
return fmt.Errorf("error killing task %s, %s, %s", container.ID(), name, err)
|
|
}
|
|
}
|
|
|
|
err = container.Delete(ctx, containerd.WithSnapshotCleanup)
|
|
if err != nil {
|
|
return fmt.Errorf("error deleting container %s, %s, %s", container.ID(), name, err)
|
|
}
|
|
} else {
|
|
service := client.SnapshotService("")
|
|
key := name + "snapshot"
|
|
if _, err := client.SnapshotService("").Stat(ctx, key); err == nil {
|
|
service.Remove(ctx, key)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Adapted from Stellar - https://github.com/stellar
|
|
func killTask(ctx context.Context, task containerd.Task) error {
|
|
|
|
killTimeout := 30 * time.Second
|
|
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
var err error
|
|
go func() {
|
|
defer wg.Done()
|
|
if task != nil {
|
|
wait, err := task.Wait(ctx)
|
|
if err != nil {
|
|
err = fmt.Errorf("error waiting on task: %s", err)
|
|
return
|
|
}
|
|
if err := task.Kill(ctx, unix.SIGTERM, containerd.WithKillAll); err != nil {
|
|
log.Printf("error killing container task: %s", err)
|
|
}
|
|
|
|
select {
|
|
case <-wait:
|
|
task.Delete(ctx)
|
|
return
|
|
case <-time.After(killTimeout):
|
|
if err := task.Kill(ctx, unix.SIGKILL, containerd.WithKillAll); err != nil {
|
|
log.Printf("error force killing container task: %s", err)
|
|
}
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
wg.Wait()
|
|
|
|
return err
|
|
}
|
|
|
|
func PrepareImage(ctx context.Context, client *containerd.Client, imageName, snapshotter string) (containerd.Image, error) {
|
|
|
|
var empty containerd.Image
|
|
image, err := client.GetImage(ctx, imageName)
|
|
if err != nil {
|
|
if !errdefs.IsNotFound(err) {
|
|
return empty, err
|
|
}
|
|
|
|
img, err := client.Pull(ctx, imageName, containerd.WithPullUnpack)
|
|
if err != nil {
|
|
return empty, fmt.Errorf("cannot pull: %s", err)
|
|
}
|
|
image = img
|
|
}
|
|
|
|
unpacked, err := image.IsUnpacked(ctx, snapshotter)
|
|
if err != nil {
|
|
return empty, fmt.Errorf("cannot check if unpacked: %s", err)
|
|
}
|
|
|
|
if !unpacked {
|
|
if err := image.Unpack(ctx, snapshotter); err != nil {
|
|
return empty, fmt.Errorf("cannot unpack: %s", err)
|
|
}
|
|
}
|
|
|
|
return image, nil
|
|
}
|