package models

import (
	"context"
	"fmt"
	"log"
	"vamicro/api-gateway/models"
	commonModels "vamicro/camera-common/models"

	"github.com/milvus-io/milvus-sdk-go/v2/client"
)

var dbClient *MilvusClient

// MilvusClient 封装Milvus操作的结构体
type MilvusClient struct {
	client client.Client
}

// NewMilvusClient 创建Milvus客户端
func NewMilvusClient(ctx context.Context, addr string) (*MilvusClient, error) {
	milvusClient, err := client.NewGrpcClient(ctx, addr)
	if err != nil {
		return nil, fmt.Errorf("failed to connect to Milvus: %v", err)
	}
	return &MilvusClient{client: milvusClient}, nil
}

// Close 关闭连接
func (m *MilvusClient) Close() error {
	return m.client.Close()
}

// 初始化milvus
func InitVectorDb() {
	// 连接到Milvus服务器
	ctx := context.Background()
	//milvusClient, err := NewMilvusClient(ctx, "192.168.1.232:19530")

	config := models.GetConfig()
	url := config.MilvusUrl

	milvusClient, err := NewMilvusClient(ctx, url)
	if err != nil {
		log.Fatal("Failed to connect to Milvus:", err)
	}
	//defer milvusClient.client.Close()

	fmt.Println("Successfully connected to Milvus")
	
	//初始化
	dbClient = milvusClient
	commonModels.SetClientDB(dbClient.client)
}

// GetDB ...
func GetDBCient() *MilvusClient {
	return dbClient
}

func CloseDBClient() {
	dbClient.Close()
}