dtp/cmd/main.go

37 lines
685 B
Go

package main
import (
"context"
"fmt"
"strings"
"github.com/Leufolium/test/es"
"github.com/mozillazg/go-pinyin"
)
func main() {
fmt.Println("Start importing...")
client, err := es.NewElasticSearch()
if err != nil {
fmt.Printf("es client init fail : %v", err)
return
}
ctx := context.Background()
list, err := client.GetStreamerAcctList(ctx)
if err != nil {
fmt.Printf("GetStreamerAcctList err :%v", err)
}
for _, acct := range list {
acct.PinYin = strings.Join(pinyin.LazyConvert(acct.Name, nil), "")
err := client.UpdateStreamerAcct(ctx, acct)
if err != nil {
fmt.Printf("UpdateStreamerAcct err :%v", err)
}
}
fmt.Println("End importing...")
}