fix-2357“

models/cloudbrain.go

@@ -137,6 +137,8 @@ type Cloudbrain struct {
 	Type                 int
 	BenchmarkTypeID      int
 	BenchmarkChildTypeID int
+	CardType             string
+	Cluster              string
 
 	VersionID       int64  //版本id
 	VersionName     string `xorm:"INDEX"` //当前版本

routers/admin/cloudbrains.go

@@ -10,6 +10,7 @@ import (
 	"github.com/360EntSecGroup-Skylar/excelize/v2"
 
 	"code.gitea.io/gitea/modules/modelarts"
+	"code.gitea.io/gitea/routers/repo"
 
 	"code.gitea.io/gitea/models"
 	"code.gitea.io/gitea/modules/base"
@@ -89,6 +90,10 @@ func CloudBrains(ctx *context.Context) {
 		ciTasks[i].CanDebug = true
 		ciTasks[i].CanDel = true
 		ciTasks[i].Cloudbrain.ComputeResource = task.ComputeResource
+		ciTasks[i].Cloudbrain.AiCenter = repo.GetCloudbrainAiCenter(task.Cloudbrain, ctx)
+		_, cardType, _ := models.GetCloudbrainCardNumAndType(task.Cloudbrain)
+		ciTasks[i].Cloudbrain.CardType = cardType
+		ciTasks[i].Cloudbrain.Cluster = repo.GetCloudbrainCluster(task.Cloudbrain, ctx)
 	}
 
 	pager := context.NewPagination(int(count), setting.UI.IssuePagingNum, page, getTotalPage(count, setting.UI.IssuePagingNum))

routers/api/v1/repo/cloudbrain_dashboard.go

@@ -10,6 +10,7 @@ import (
 	"code.gitea.io/gitea/models"
 	"code.gitea.io/gitea/modules/context"
 	"code.gitea.io/gitea/modules/log"
+	"code.gitea.io/gitea/routers/repo"
 	"github.com/360EntSecGroup-Skylar/excelize/v2"
 )
 
@@ -754,7 +755,7 @@ func GetCloudbrainsDetailData(ctx *context.Context) {
 		taskDetail.CardNum, taskDetail.CardType, _ = models.GetCloudbrainCardNumAndType(ciTasks[i].Cloudbrain)
 
 		taskDetail.CardDuration = models.ConvertDurationToStr(int64(taskDetail.CardNum) * ciTasks[i].Cloudbrain.Duration)
-		taskDetail.AiCenter = getCloudbrainAiCenter(ciTasks[i].Cloudbrain, ctx)
+		taskDetail.AiCenter = repo.GetCloudbrainAiCenter(ciTasks[i].Cloudbrain, ctx)
 		taskDetail.FlavorName, _ = models.GetCloudbrainFlavorName(ciTasks[i].Cloudbrain)
 
 		taskDetail.WaitTime = getCloudbrainWaitTime(ciTasks[i].Cloudbrain)

routers/repo/cloudbrain.go

@@ -2,7 +2,6 @@ package repo
 
 import (
 	"bufio"
-	"code.gitea.io/gitea/modules/grampus"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -16,6 +15,8 @@ import (
 	"time"
 	"unicode/utf8"
 
+	"code.gitea.io/gitea/modules/grampus"
+
 	"code.gitea.io/gitea/modules/timeutil"
 	"github.com/unknwon/i18n"
 
@@ -2309,3 +2310,21 @@ func GetBenchmarkTypes(ctx *context.Context) *models.BenchmarkTypes {
 	}
 	return benchmarkTypesMap[lang]
 }
+func GetCloudbrainAiCenter(task models.Cloudbrain, ctx *context.Context) string {
+	if task.Type == models.TypeCloudBrainOne {
+		return ctx.Tr("repo.cloudbrain1")
+	} else if task.Type == models.TypeCloudBrainTwo {
+		return ctx.Tr("repo.cloudbrain2")
+	} else if task.Type == models.TypeC2Net {
+		return task.AiCenter
+	}
+	return ""
+}
+func GetCloudbrainCluster(task models.Cloudbrain, ctx *context.Context) string {
+	if task.Type == models.TypeCloudBrainOne || task.Type == models.TypeCloudBrainTwo {
+		return ctx.Tr("cloudbrain.resource_cluster_openi")
+	} else if task.Type == models.TypeC2Net {
+		return ctx.Tr("cloudbrain.resource_cluster_c2net")
+	}
+	return ""
+}

routers/user/home.go

@@ -834,6 +834,11 @@ func Cloudbrains(ctx *context.Context) {
 		ciTasks[i].CanDebug = true
 		ciTasks[i].CanDel = true
 		ciTasks[i].Cloudbrain.ComputeResource = task.ComputeResource
+		ciTasks[i].Cloudbrain.AiCenter = GetCloudbrainAiCenter(task.Cloudbrain, ctx)
+		_, cardType, _ := models.GetCloudbrainCardNumAndType(task.Cloudbrain)
+		ciTasks[i].Cloudbrain.CardType = cardType
+		ciTasks[i].Cloudbrain.Cluster = GetCloudbrainCluster(task.Cloudbrain, ctx)
+
 	}
 
 	pager := context.NewPagination(int(count), setting.UI.IssuePagingNum, page, getTotalPage(count, setting.UI.IssuePagingNum))
@@ -857,3 +862,22 @@ func getTotalPage(total int64, pageSize int) int {
 	return int(total)/pageSize + another
 
 }
+
+func GetCloudbrainAiCenter(task models.Cloudbrain, ctx *context.Context) string {
+	if task.Type == models.TypeCloudBrainOne {
+		return ctx.Tr("repo.cloudbrain1")
+	} else if task.Type == models.TypeCloudBrainTwo {
+		return ctx.Tr("repo.cloudbrain2")
+	} else if task.Type == models.TypeC2Net {
+		return task.AiCenter
+	}
+	return ""
+}
+func GetCloudbrainCluster(task models.Cloudbrain, ctx *context.Context) string {
+	if task.Type == models.TypeCloudBrainOne || task.Type == models.TypeCloudBrainTwo {
+		return ctx.Tr("cloudbrain.resource_cluster_openi")
+	} else if task.Type == models.TypeC2Net {
+		return ctx.Tr("cloudbrain.resource_cluster_c2net")
+	}
+	return ""
+}