From 03880bf61cf059f063e252ef17dfea50c932c9dc Mon Sep 17 00:00:00 2001
From: qixiaoning <jony.kee@outlook.com>
Date: 星期三, 12 十一月 2025 09:11:03 +0800
Subject: [PATCH] 清理数据接口修改
---
camera-common/models/db.go | 19 ++++++++++++++++++-
1 files changed, 18 insertions(+), 1 deletions(-)
diff --git a/camera-common/models/db.go b/camera-common/models/db.go
index 6feec2f..add8919 100644
--- a/camera-common/models/db.go
+++ b/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
+}
--
Gitblit v1.8.0