models/cloudbrain.go

@@ -2162,7 +2162,7 @@ type GetGrampusResourceSpecsResult struct {
 
 type GetGrampusResourceSpecPoolsResult struct {
 	GrampusResult
-	Infos []GrampusSpecPools `json:"resourceSpecs"`
+	Infos []GrampusSpecPools `json:"resourceSpecPools"`
 }
 
 type GetGrampusAiCentersResult struct {

models/resource_queue.go

@@ -196,6 +196,21 @@ func InsertResourceQueue(queue ResourceQueue) (int64, error) {
 func UpdateResourceQueueById(queueId int64, queue ResourceQueue) (int64, error) {
 	return x.ID(queueId).Update(&queue)
 }
+
+func UpdateResourceQueueCardNum(queue ResourceQueue) error {
+	// 直接更新 cards_total_num 字段
+	_, err := x.Cols("cards_total_num").Where("queue_code = ? and ai_center_code = ? and acc_card_type ILIKE ?", queue.QueueCode, queue.AiCenterCode, queue.AccCardType).Update(&ResourceQueue{
+		CardsTotalNum: queue.CardsTotalNum,
+	})
+	if err != nil {
+		return err
+	} else {
+		log.Info("QueueCode is %v, AiCenterCode is: %v, AccCardType is: %v, CardsTotalNum is: %v\n", queue.QueueCode, queue.AiCenterCode, queue.AccCardType, queue.CardsTotalNum)
+	}
+
+	return nil
+}
+
 func UpdateResourceCardsTotalNumAndInternetStatus(queueId int64, queue ResourceQueue, isAvailable int) (int64, error) {
 	sess := x.NewSession()
 	if err := sess.Begin(); err != nil {

modules/grampus/resty.go

@@ -29,7 +29,7 @@ const (
 	urlGetToken             = urlOpenApiV1 + "token"
 	urlTrainJob             = urlOpenApiV1 + "trainjob"
 	urlGetResourceSpecs     = urlOpenApiV1 + "resourcespec"
-	urlGetResourceSpecPools = urlOpenApiV1 + "resourceSpecPools"
+	urlGetResourceSpecPools = urlOpenApiV1 + "resourceSpecPool"
 	urlGetAiCenter          = urlOpenApiV1 + "sharescreen/aicenter"
 	urlGetImages            = urlOpenApiV1 + "image"
 	urlDelImages            = urlOpenApiV1 + "delimage"

services/cloudbrain/resource/resource_specification.go

@@ -140,10 +140,13 @@ func SyncGrampusSpecPools() error {
 			fmt.Printf("strconv.ParseInt error: %v\n", err)
 		}
 		resourceQueue := models.ResourceQueue{
-			ID:            queue_id,
+			AiCenterCode:  info.CenterId,
+			QueueCode:     info.ID,
+			AccCardType:   info.CardType,
 			CardsTotalNum: info.CardNum,
 		}
-		_, err = models.UpdateResourceQueueById(queue_id, resourceQueue)
+
+		err = models.UpdateResourceQueueCardNum(resourceQueue)
 		if err != nil {
 			fmt.Printf("UpdateResourceQueueById error: %v, queueId is: %v\n", err, queue_id)
 		}