From ba6483c0142054f277fd1fc795d8396dcde9bb8a Mon Sep 17 00:00:00 2001 From: Mojtaba Date: Thu, 4 Jul 2024 15:50:32 +0200 Subject: [PATCH] feat: cleanup func for preloader --- pkg/preloader/preloader.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pkg/preloader/preloader.go b/pkg/preloader/preloader.go index 7e49453..5b9a55f 100644 --- a/pkg/preloader/preloader.go +++ b/pkg/preloader/preloader.go @@ -125,3 +125,7 @@ func (p *Preloader) preloadImages(ctx context.Context) error { _, err = p.K8sClient.CreateDaemonSet(ctx, p.K8sName, labels, initContainers, containers) return err } + +func (p *Preloader) Cleanup(ctx context.Context) error { + return p.K8sClient.DeleteDaemonSet(ctx, p.K8sName) +}