re-enable sync function

shinebayar-g · Nov 16, 2022 · da9c35a · da9c35a
1 parent 384430b
commit da9c35a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/main.go b/main.go
@@ -61,7 +61,7 @@ func main() {
 	go ufwhandler.CreateUfwRule(createChannel, trackedContainers)
 	go ufwhandler.DeleteUfwRule(deleteChannel, trackedContainers)
 	// go ufwhandler.Cleanup(client, ctx)
-	// go ufwhandler.Sync(createChannel, client, ctx)
+	go ufwhandler.Sync(createChannel, client, ctx)
 
 	messages, errors := streamEvents(ctx, client)
 	for {

diff --git a/ufwhandler/sync.go b/ufwhandler/sync.go
@@ -5,19 +5,23 @@ import (
 	"log"
 
 	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/api/types/filters"
 	"github.com/docker/docker/client"
 )
 
 func Sync(createChannel chan *types.ContainerJSON, client *client.Client, ctx *context.Context) {
-	containers, err := client.ContainerList(*ctx, types.ContainerListOptions{})
+	// Returns only running containers
+	filter := filters.NewArgs()
+	filter.Add("label", "UFW_MANAGED=TRUE")
+	containers, err := client.ContainerList(*ctx, types.ContainerListOptions{Filters: filter})
 	if err != nil {
-		log.Println(err)
+		log.Println("ufw-docker-automated: Couldn't retrieve existing containers:", err)
 	}
 
 	for _, c := range containers {
 		container, err := client.ContainerInspect(*ctx, c.ID)
 		if err != nil {
-			log.Println("ufw-docker-automated: Couldn't inspect container:", err)
+			log.Println("ufw-docker-automated: Couldn't inspect existing container:", err)
 			continue
 		}
 		createChannel <- &container