获取算力接口实现

qixiaoning

2025-08-21 e38654fe9eff4562da4f18f8f018aed7879d493c

 camera-service/models/milvus.go

@@ -1,63 +1,63 @@
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()
}
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()

}