From 95c43cf2717aed6ed67f24a02ada6d70519be8e0 Mon Sep 17 00:00:00 2001
From: tzwang <wtz1003@sina.com>
Date: Tue, 30 Apr 2024 17:35:56 +0800
Subject: [PATCH] updated ai db functions

Former-commit-id: c1a9ee19503fbedfe980ec175ec0fc846dd26e87
---
 .../logic/ai/getcenteroverviewlogic.go        |  2 +-
 api/internal/scheduler/database/aiStorage.go  | 28 +++++++++++++++++++
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/api/internal/logic/ai/getcenteroverviewlogic.go b/api/internal/logic/ai/getcenteroverviewlogic.go
index 9f76979d..eb93684b 100644
--- a/api/internal/logic/ai/getcenteroverviewlogic.go
+++ b/api/internal/logic/ai/getcenteroverviewlogic.go
@@ -61,7 +61,7 @@ func (l *GetCenterOverviewLogic) GetCenterOverview() (resp *types.CenterOverview
 		}
 	}
 
-	resp.CardNum = centerNum
+	resp.CardNum = cardNum
 	resp.PowerInTops = totalTops
 
 	return resp, nil
diff --git a/api/internal/scheduler/database/aiStorage.go b/api/internal/scheduler/database/aiStorage.go
index c458c622..018f58b8 100644
--- a/api/internal/scheduler/database/aiStorage.go
+++ b/api/internal/scheduler/database/aiStorage.go
@@ -169,6 +169,34 @@ func (s *AiStorage) GetClusterResourcesById(clusterId string) (*models.TClusterR
 	return &clusterResource, nil
 }
 
+func (s *AiStorage) SaveClusterResources(clusterId string, clusterName string, clusterType int64, cpuAvail float64, cpuTotal float64,
+	memAvail float64, memTotal float64, diskAvail float64, diskTotal float64, gpuAvail float64, gpuTotal float64, cardTotal int64, topsTotal float64) error {
+	cId, err := strconv.ParseInt(clusterId, 10, 64)
+	if err != nil {
+		return err
+	}
+	clusterResource := models.TClusterResource{
+		ClusterId:     cId,
+		ClusterName:   clusterName,
+		ClusterType:   clusterType,
+		CpuAvail:      cpuAvail,
+		CpuTotal:      cpuTotal,
+		MemAvail:      memAvail,
+		MemTotal:      memTotal,
+		DiskAvail:     diskAvail,
+		DiskTotal:     diskTotal,
+		GpuAvail:      gpuAvail,
+		GpuTotal:      gpuTotal,
+		CardTotal:     cardTotal,
+		CardTopsTotal: topsTotal,
+	}
+	tx := s.DbEngin.Create(&clusterResource)
+	if tx.Error != nil {
+		return tx.Error
+	}
+	return nil
+}
+
 func (s *AiStorage) UpdateTask() error {
 	return nil
 }