30 lines
720 B
Go
30 lines
720 B
Go
|
package milvus
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/milvus-io/milvus-sdk-go/v2/client"
|
||
|
|
"leafdev.top/Leaf/leaf-library/internal/base/conf"
|
||
|
|
"leafdev.top/Leaf/leaf-library/internal/base/logger"
|
||
|
|
|
||
|
|
"strconv"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewMilvus(config *conf.Config, logger *logger.Logger) client.Client {
|
||
|
|
var address = config.Milvus.Host + ":" + strconv.Itoa(config.Milvus.Port)
|
||
|
|
|
||
|
|
logger.Sugar.Infof("Waiting for milvus, address=%s, dbname=%s", address, config.Milvus.DBName)
|
||
|
|
|
||
|
|
c, err := client.NewClient(context.Background(), client.Config{
|
||
|
|
Address: address,
|
||
|
|
DBName: config.Milvus.DBName,
|
||
|
|
})
|
||
|
|
|
||
|
|
logger.Sugar.Infof("Connected to milvus, address=%s, dbname=%s", address, config.Milvus.DBName)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return c
|
||
|
|
}
|