1
0
mirror of https://github.com/stevenferrer/multi-select-facet.git synced 2025-11-23 21:54:45 +02:00

Upgrade solr-go to version 0.2

This commit is contained in:
Steven Ferrer
2021-02-10 09:52:07 +08:00
parent fe3169f05f
commit 973d09768b
12 changed files with 193 additions and 218 deletions

52
suggest_handler.go Normal file
View File

@@ -0,0 +1,52 @@
package main
import (
"encoding/json"
"net/http"
"github.com/sf9v/solr-go"
)
type suggestHandler struct {
collection string
solrClient solr.Client
}
type suggestion struct {
Term string `json:"term"`
}
func (h *suggestHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
q := r.URL.Query().Get("q")
if len(q) == 0 {
return
}
dict := "default"
suggestParams := solr.NewSuggesterParams("suggest").
Build().Query(q).Dictionaries(dict)
suggestResp, err := h.solrClient.Suggest(r.Context(), h.collection, suggestParams)
if err != nil {
http.Error(w, err.Error(), 500)
return
}
suggest := *suggestResp.Suggest
termBody := suggest[dict][q]
suggestions := []suggestion{}
for _, suggest := range termBody.Suggestions {
suggestions = append(suggestions, suggestion{
Term: suggest.Term,
})
}
err = json.NewEncoder(w).Encode(solr.M{
"numFound": termBody.NumFound,
"suggestions": suggestions,
})
if err != nil {
http.Error(w, err.Error(), 500)
}
}