qixiaoning
2025-07-22 550492cb85ea7e01831939f2866259193378f2f7
camera-common/models/db.go
@@ -1,8 +1,17 @@
package models
import "github.com/jinzhu/gorm"
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
@@ -11,3 +20,11 @@
func GetDB() *gorm.DB {
   return db
}
func SetClientDB(h client.Client) {
   dbClient = &MilvusClient{client: h}
}
func GetClientDB() *MilvusClient {
   return dbClient
}