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 }