Check if the result limit has been hit
Signed-off-by: Hadi <112569860+anotherhadi@users.noreply.github.com>
This commit is contained in:
@@ -10,8 +10,6 @@ import (
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// TODO: We need to know when we hit the LIMIT
|
||||
|
||||
func routes(s *server.Server, cache *map[string]*search.Result) {
|
||||
s.Router.Use(
|
||||
func(c *gin.Context) {
|
||||
|
||||
@@ -15,6 +15,7 @@ type LeakResult struct {
|
||||
Duration time.Duration
|
||||
Rows []map[string]string
|
||||
Error string
|
||||
LimitHit bool // Whether the search hit the limit
|
||||
}
|
||||
|
||||
func Search(s *server.Server, queryText, column string, exactMatch bool) LeakResult {
|
||||
@@ -76,6 +77,10 @@ func Search(s *server.Server, queryText, column string, exactMatch bool) LeakRes
|
||||
return result
|
||||
}
|
||||
|
||||
if len(result.Rows) >= s.Settings.Limit {
|
||||
result.LimitHit = true
|
||||
}
|
||||
|
||||
result.Rows = removeDuplicateMaps(result.Rows)
|
||||
|
||||
result.Duration = time.Since(now)
|
||||
|
||||
Reference in New Issue
Block a user