From 77d234831ed58c8851573e3e12e7c7dc136e41f8 Mon Sep 17 00:00:00 2001
From: Jay <admin@trj.tw>
Date: Thu, 7 Mar 2019 17:15:06 +0800
Subject: [PATCH] fix keep image number

---
 main.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 2fb816e..8176045 100644
--- a/main.go
+++ b/main.go
@@ -114,6 +114,7 @@ func main() {
 	}
 
 	iarr := ImageArr{}
+	keepNum := runFlag.KeepNum
 	for _, v := range imgs.ImageDetails {
 		excludeFlag := false
 		if len(runFlag.ExcludeTag) > 0 {
@@ -123,6 +124,7 @@ func main() {
 					if *v2 == e {
 						f = true
 						excludeFlag = true
+						keepNum--
 						break
 					}
 				}
@@ -135,8 +137,11 @@ func main() {
 			iarr = append(iarr, v)
 		}
 	}
+	if keepNum < 0 {
+		keepNum = 0
+	}
 	sort.Stable(iarr)
-	iarr = iarr[runFlag.KeepNum:]
+	iarr = iarr[keepNum:]
 	if len(iarr) == 0 {
 		fmt.Println("no delete image")
 		return