Update main.go

FloatTech · Sep 3, 2023 · 8f29c2a · 8f29c2a
1 parent ac26687
commit 8f29c2a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/plugin/mcfish/main.go b/plugin/mcfish/main.go
@@ -343,6 +343,10 @@ func (sql *fishdb) pickFishFor(uid int64, number int) (fishNames map[string]int,
 	}
 	for i := number; i > 0; i-- {
 		randNumber := rand.Intn(len(fishTypes))
+		if fishTypes[randNumber].Number <= 0 {
+			i++
+			continue
+		}
 		fishTypes[randNumber].Number--
 		err = sql.db.Insert(name, &fishTypes[randNumber])
 		if err != nil {