108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/meilisearch/meilisearch-go"
|
|
)
|
|
|
|
func main() {
|
|
var hostname, masterKey string
|
|
var list bool
|
|
flag.StringVar(&hostname, "host", "meili.zerops", "hostname to connect to elastic")
|
|
flag.StringVar(&masterKey, "key", "nope", "master key")
|
|
flag.BoolVar(&list, "list", false, "whether to list the whole index")
|
|
flag.Parse()
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
client := meilisearch.New(fmt.Sprintf("http://%s:7700", hostname), meilisearch.WithAPIKey(masterKey))
|
|
|
|
const indexName = "xxx"
|
|
|
|
taskInfo, err := client.CreateIndexWithContext(ctx, &meilisearch.IndexConfig{
|
|
Uid: indexName,
|
|
PrimaryKey: "id",
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
status := taskInfo.Status
|
|
var taskErr any
|
|
for status != "succeeded" {
|
|
if status == "failed" || status == "canceled" {
|
|
fmt.Println("task failed:", taskErr)
|
|
break
|
|
}
|
|
task, err := client.GetTaskWithContext(ctx, taskInfo.TaskUID)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
status = task.Status
|
|
taskErr = task.Error
|
|
}
|
|
|
|
indices, err := client.ListIndexesWithContext(ctx, &meilisearch.IndexesQuery{})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for _, index := range indices.Results {
|
|
fmt.Println(index.UID, index.PrimaryKey)
|
|
}
|
|
|
|
index := client.Index(indexName)
|
|
|
|
if list {
|
|
resp, err := index.SearchWithContext(ctx, "*", &meilisearch.SearchRequest{Limit: 1024 * 1024})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for _, hit := range resp.Hits {
|
|
fmt.Printf("LIST: ==> %s\n", hit.(map[string]any)["message"])
|
|
}
|
|
}
|
|
|
|
for {
|
|
if ctx.Err() != nil {
|
|
fmt.Println("context canceled")
|
|
return
|
|
}
|
|
|
|
now := time.Now()
|
|
message := now.Format(time.TimeOnly)
|
|
document := map[string]any{
|
|
"id": now.UnixNano(),
|
|
"name": "meili",
|
|
"message": message,
|
|
}
|
|
taskInfo, err = index.AddDocumentsWithContext(ctx, []map[string]any{document})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
status = taskInfo.Status
|
|
for status != "succeeded" {
|
|
if status == "failed" || status == "canceled" {
|
|
fmt.Println("task failed:", taskErr)
|
|
break
|
|
}
|
|
task, err := index.GetTaskWithContext(ctx, taskInfo.TaskUID)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
status = task.Status
|
|
taskErr = task.Error
|
|
}
|
|
fmt.Printf("==> %s\n", document["message"])
|
|
|
|
time.Sleep(time.Second)
|
|
}
|
|
}
|