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
}