1
0
Fork 0

Add Anagrams function

This commit is contained in:
Andinus 2020-04-06 23:52:41 +05:30
parent 1e6075d11c
commit 30f7ea8ddf
Signed by: andinus
GPG Key ID: B67D55D482A799FD
2 changed files with 38 additions and 5 deletions

35
search/anagrams.go Normal file
View File

@ -0,0 +1,35 @@
package search
import "tildegit.org/andinus/grus/storage"
// Anagrams will search for unjumbled words in database, given sorted
// word along with all the anagrams.
func Anagrams(sorted string, db *storage.DB) (anagrams []string, err error) {
db.Mu.RLock()
defer db.Mu.RUnlock()
stmt, err := db.Conn.Prepare("SELECT word FROM words WHERE sorted = ?")
if err != nil {
return
}
defer stmt.Close()
rows, err := stmt.Query(sorted)
if err != nil {
return
}
defer rows.Close()
for rows.Next() {
var word string
err = rows.Scan(&word)
if err != nil {
return
}
anagrams = append(anagrams, word)
}
err = rows.Err()
if err != nil {
return
}
return
}

View File

@ -1,11 +1,9 @@
package search
import (
"tildegit.org/andinus/grus/storage"
)
import "tildegit.org/andinus/grus/storage"
// Word will search for unjumbled words in database, given sorted word.
func Word(sorted string, db *storage.DB) (out string, err error) {
func Word(sorted string, db *storage.DB) (word string, err error) {
db.Mu.RLock()
defer db.Mu.RUnlock()
@ -15,7 +13,7 @@ func Word(sorted string, db *storage.DB) (out string, err error) {
}
defer stmt.Close()
err = stmt.QueryRow(sorted).Scan(&out)
err = stmt.QueryRow(sorted).Scan(&word)
if err != nil {
return
}