added greedy flag.

main.go

@@ -15,6 +15,7 @@ const envDataPath = "NEXTWORD_DATA_PATH"
 var dataPath = flag.String("data", os.Getenv(envDataPath), "path to data directory")
 var maxCandidatesNum = flag.Int("candidates-num", 100, "max candidates num (positive int)")
 var helpFlag = flag.Bool("h", false, "show this message")
+var greedyFlag = flag.Bool("greedy", false, "show as many result as possible")
 
 func main() {
 	os.Exit(run())
@@ -46,7 +47,7 @@ environment value "$NEXTWORD_DATA_PATH". `
 }
 
 func serve() error {
-	sg := NewSuggester(*dataPath, *maxCandidatesNum)
+	sg := NewSuggester(*dataPath, *maxCandidatesNum, *greedyFlag)
 
 	sc := bufio.NewScanner(os.Stdin)
 	for sc.Scan() {

suggester.go

@@ -12,10 +12,15 @@ import (
 type Suggester struct {
 	dataPath     string
 	candidateLen int
+	greedy       bool
 }
 
-func NewSuggester(dataPath string, candidateLen int) *Suggester {
-	return &Suggester{dataPath: dataPath, candidateLen: candidateLen}
+func NewSuggester(dataPath string, candidateLen int, greedy bool) *Suggester {
+	return &Suggester{
+		dataPath:     dataPath,
+		candidateLen: candidateLen,
+		greedy:       greedy,
+	}
 }
 
 func (*Suggester) fileName(n int, prefix string) string {
@@ -44,6 +49,17 @@ func (sg *Suggester) Suggest(query string) (candidates []string, err error) {
 			return
 		}
 		candidates = append(candidates, cand...)
+
+		// return when non greedy with non empty candidates
+		if !sg.greedy {
+			candidates = sg.filterCandidates(candidates, prefix)
+			if len(candidates) > sg.candidateLen {
+				candidates = candidates[:sg.candidateLen]
+			}
+			if len(candidates) > 0 {
+				return
+			}
+		}
 	}
 
 	// search 1gram