feat: add builder cache prune

docker-host/main.go

@@ -28,9 +28,11 @@ var logLevel = machineryvars.GetEnv("LOG_LEVEL", "info")
 var REGISTRY_MIRROR = machineryvars.GetEnv("REGISTRY_MIRROR", "")
 var pruneImagesSchedule = machineryvars.GetEnv("PRUNE_SCHEDULE", "22 1 * * *")
 var pruneDanglingSchedule = machineryvars.GetEnv("PRUNE_DANGLING_SCHEDULE", "22 1 * * *")
+var pruneBuilderCacheSchedule = machineryvars.GetEnv("PRUNE_BUILDER_CACHE_SCHEDULE", "22 1 * * *")
 var removeExitedSchedule = machineryvars.GetEnv("REMOVE_EXITED_SCHEDULE", "22 */4 * * *")
 var updateImagesSchedule = machineryvars.GetEnv("UPDATE_IMAGES_SCHEDULE", "*/15 * * * *")
 var pruneImagesUntil = machineryvars.GetEnv("PRUNE_IMAGES_UNTIL", "168h")
+var pruneBuilderCacheUntil = machineryvars.GetEnv("PRUNE_BUILDER_CACHE_UNTIL", "168h")
 
 func main() {
 	cli, err := client.NewClientWithOpts(
@@ -57,6 +59,7 @@ func main() {
 	c := cron.New()
 	pruneImages(cli, c)
 	pruneDanglingImages(cli, c)
+	pruneBuilderCache(cli, c)
 	removeExited(cli, c)
 	updateImages(cli, c)
 	c.Start()
@@ -113,6 +116,29 @@ func pruneDanglingImages(client *client.Client, c *cron.Cron) {
 	}
 }
 
+func pruneBuilderCache(client *client.Client, c *cron.Cron) {
+	_, err := c.AddFunc(pruneBuilderCacheSchedule, func() {
+		log.Println("Starting builder cache prune")
+		ageFilter := filters.NewArgs()
+		ageFilter.Add("until", pruneBuilderCacheUntil)
+		builderCacheOpts := types.BuildCachePruneOptions{
+			Filters: ageFilter,
+		}
+
+		// Cleans up dangling images
+		pruneReport, pruneErr := client.BuildCachePrune(context.Background(), builderCacheOpts)
+		if pruneErr != nil {
+			log.Println(pruneErr)
+		}
+		log.Printf("Builder Cache prune complete: %d deleted, %d bytes reclaimed\n",
+			len(pruneReport.CachesDeleted), pruneReport.SpaceReclaimed)
+	})
+
+	if err != nil {
+		log.Printf("Error initiating pruneBuilderCache cron: %v\n", err)
+	}
+}
+
 func removeExited(client *client.Client, c *cron.Cron) {
 	_, err := c.AddFunc(removeExitedSchedule, func() {
 		log.Println("Starting removeExited")
@@ -197,7 +223,7 @@ func updateImages(client *client.Client, c *cron.Cron) {
 		for _, img := range postUpdateImages {
 			for _, updatedImg := range updatedImages {
 				if img.ID == updatedImg {
-					log.Println(fmt.Sprintf("Updated image %s", img.RepoTags))
+					log.Printf("Updated image %s", img.RepoTags)
 				}
 			}
 		}
@@ -208,7 +234,7 @@ func updateImages(client *client.Client, c *cron.Cron) {
 		} else {
 			imgPluralize = "images"
 		}
-		log.Println(fmt.Sprintf("Update images complete | %d %s updated", len(updatedImages), imgPluralize))
+		log.Printf("Update images complete | %d %s updated", len(updatedImages), imgPluralize)
 	})
 
 	if err != nil {

go.mod

@@ -0,0 +1,3 @@
+module github.com/uselagoon/lagoon-service-images
+
+go 1.22