package models

import (
	"github.com/jinzhu/gorm"
	"github.com/milvus-io/milvus-sdk-go/v2/client"
)

var db *gorm.DB
var dbClient *MilvusClient

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

func SetDB(h *gorm.DB) {
	db = h
}

func GetDB() *gorm.DB {
	return db
}

func SetClientDB(h client.Client) {
	dbClient = &MilvusClient{client: h}
}

func GetClientDB() *MilvusClient {
	return dbClient
}