Store: Set start/end prefix even when one argument is given.

sarifsystems · May 21, 2017 · 9f20005 · 9f20005
1 parent ac5ab6c
commit 9f20005
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/services/store/service.go b/services/store/service.go
@@ -265,10 +265,13 @@ type DocsPayload struct {
 }
 
 func (s *Service) doScan(collection string, p scanMessage) (interface{}, error) {
-	if p.Start == "" && p.End == "" {
-		if p.Prefix != "" {
-			p.Start = p.Prefix
-			p.End = p.Prefix + "~~~~~" // oh god, what a hack
+	if p.Prefix != "" {
+		start, end := clampPrefix(p.Prefix)
+		if p.Start == "" {
+			p.Start = start
+		}
+		if p.End == "" {
+			p.End = end
 		}
 	}
 	if p.Limit == 0 {
@@ -312,3 +315,7 @@ func (s *Service) doScan(collection string, p scanMessage) (interface{}, error)
 		return DocsPayload{keys, values}, nil
 	}
 }
+
+func clampPrefix(prefix string) (start, end string) {
+	return prefix, prefix + " ~~~~~"
+}