#1249

update point when multi node

models/cloudbrain_spec.go

@@ -123,10 +123,14 @@ func GetResourceSpecMapByCloudbrainIDs(ids []int64) (map[int64]*Specification, e
 	return r, nil
 }
 
-func GetCloudbrainTaskUnitPrice(cloudbrainId int64) (int, error) {
-	s, err := GetCloudbrainSpecByID(cloudbrainId)
+func GetCloudbrainTaskUnitPrice(task Cloudbrain) (int, error) {
+	s, err := GetCloudbrainSpecByID(task.ID)
 	if err != nil {
 		return 0, err
 	}
-	return s.UnitPrice, nil
+	var n = 1
+	if task.WorkServerNumber > 1 {
+		n = task.WorkServerNumber
+	}
+	return s.UnitPrice * n, nil
 }

routers/repo/modelarts.go

@@ -1216,7 +1216,7 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 		ctx.RenderWithErr("Resource specification not available", tplModelArtsTrainJobNew, &form)
 		return
 	}
-	if !account.IsPointBalanceEnough(ctx.User.ID, spec.UnitPrice) {
+	if !account.IsPointBalanceEnough(ctx.User.ID, spec.UnitPrice*form.WorkServerNumber) {
 		log.Error("point balance is not enough,userId=%d specId=%d", ctx.User.ID, spec.ID)
 		cloudBrainNewDataPrepare(ctx)
 		ctx.RenderWithErr(ctx.Tr("points.insufficient_points_balance"), tplModelArtsTrainJobNew, &form)

services/reward/cloudbrain_deduct.go

@@ -29,7 +29,7 @@ func StartAndGetCloudBrainPointDeductTask(task models.Cloudbrain) (*models.Rewar
 		return nil, nil
 	}
 
-	unitPrice, err := models.GetCloudbrainTaskUnitPrice(task.ID)
+	unitPrice, err := models.GetCloudbrainTaskUnitPrice(task)
 	if err != nil {
 		return nil, err
 	}