|
|
|
@@ -537,3 +537,126 @@ func ServeCloudbrainPeriodStatisticsFile(ctx *context.Context) { |
|
|
|
f.WriteTo(ctx.Resp) |
|
|
|
|
|
|
|
} |
|
|
|
func DownloadCloudBrainBoard(ctx *context.Context) { |
|
|
|
|
|
|
|
page := 1 |
|
|
|
|
|
|
|
pageSize := 300 |
|
|
|
|
|
|
|
var cloudBrain = ctx.Tr("repo.cloudbrain") |
|
|
|
fileName := getCloudbrainFileName(cloudBrain) |
|
|
|
|
|
|
|
_, total, err := models.CloudbrainAll(&models.CloudbrainsOptions{ |
|
|
|
ListOptions: models.ListOptions{ |
|
|
|
Page: page, |
|
|
|
PageSize: 1, |
|
|
|
}, |
|
|
|
Type: models.TypeCloudBrainAll, |
|
|
|
NeedRepoInfo: false, |
|
|
|
}) |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
log.Warn("Can not get cloud brain info", err) |
|
|
|
ctx.Error(http.StatusBadRequest, ctx.Tr("repo.cloudbrain_query_fail")) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
totalPage := getTotalPage(total, pageSize) |
|
|
|
fmt.Printf("total:%v", total) |
|
|
|
fmt.Printf("totalPage:%v", totalPage) |
|
|
|
|
|
|
|
f := excelize.NewFile() |
|
|
|
|
|
|
|
index := f.NewSheet(cloudBrain) |
|
|
|
f.DeleteSheet("Sheet1") |
|
|
|
|
|
|
|
for k, v := range allHeader(ctx) { |
|
|
|
f.SetCellValue(cloudBrain, k, v) |
|
|
|
} |
|
|
|
|
|
|
|
var row = 2 |
|
|
|
for i := 0; i < totalPage; i++ { |
|
|
|
|
|
|
|
pageRecords, _, err := models.CloudbrainAll(&models.CloudbrainsOptions{ |
|
|
|
ListOptions: models.ListOptions{ |
|
|
|
Page: page, |
|
|
|
PageSize: pageSize, |
|
|
|
}, |
|
|
|
Type: models.TypeCloudBrainAll, |
|
|
|
NeedRepoInfo: true, |
|
|
|
}) |
|
|
|
if err != nil { |
|
|
|
log.Warn("Can not get cloud brain info", err) |
|
|
|
continue |
|
|
|
} |
|
|
|
for _, record := range pageRecords { |
|
|
|
|
|
|
|
for k, v := range allValues(row, record, ctx) { |
|
|
|
f.SetCellValue(cloudBrain, k, v) |
|
|
|
} |
|
|
|
row++ |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
page++ |
|
|
|
} |
|
|
|
f.SetActiveSheet(index) |
|
|
|
|
|
|
|
ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(fileName)) |
|
|
|
ctx.Resp.Header().Set("Content-Type", "application/octet-stream") |
|
|
|
|
|
|
|
f.WriteTo(ctx.Resp) |
|
|
|
} |
|
|
|
|
|
|
|
func getCloudbrainFileName(baseName string) string { |
|
|
|
return baseName + "_" + time.Now().Format(EXCEL_DATE_FORMAT) + ".xlsx" |
|
|
|
|
|
|
|
} |
|
|
|
func allHeader(ctx *context.Context) map[string]string { |
|
|
|
|
|
|
|
return map[string]string{"A1": ctx.Tr("repo.cloudbrain_task"), "B1": ctx.Tr("repo.cloudbrain_task_type"), "C1": ctx.Tr("repo.modelarts.status"), |
|
|
|
"D1": ctx.Tr("repo.modelarts.createtime"), "E1": ctx.Tr("repo.modelarts.train_job.dura_time"), "F1": ctx.Tr("repo.modelarts.computing_resources"), |
|
|
|
"G1": ctx.Tr("repo.cloudbrain_creator"), "H1": ctx.Tr("repo.repo_name"), "I1": ctx.Tr("repo.cloudbrain_task_name"), "J1": ctx.Tr("repo.modelarts.train_job.start_time"), |
|
|
|
"K1": ctx.Tr("repo.modelarts.train_job.end_time"), "L1": ctx.Tr("repo.modelarts.train_job.wait_time")} |
|
|
|
|
|
|
|
} |
|
|
|
func allValues(row int, rs *models.CloudbrainInfo, ctx *context.Context) map[string]string { |
|
|
|
return map[string]string{getCellName("A", row): rs.DisplayJobName, getCellName("B", row): rs.JobType, getCellName("C", row): rs.Status, |
|
|
|
getCellName("D", row): time.Unix(int64(rs.Cloudbrain.CreatedUnix), 0).Format(CREATE_TIME_FORMAT), getCellName("E", row): rs.TrainJobDuration, |
|
|
|
getCellName("F", row): rs.ComputeResource, getCellName("G", row): rs.Name, getCellName("H", row): getRepoPathName(rs), |
|
|
|
getCellName("I", row): rs.JobName, getCellName("J", row): getBrainStartTime(rs), |
|
|
|
getCellName("K", row): getBrainEndTime(rs), getCellName("L", row): getBrainWaitTime(rs), |
|
|
|
} |
|
|
|
} |
|
|
|
func getRepoPathName(rs *models.CloudbrainInfo) string { |
|
|
|
if rs.Repo != nil { |
|
|
|
return rs.Repo.OwnerName + "/" + rs.Repo.Alias |
|
|
|
} |
|
|
|
return "" |
|
|
|
} |
|
|
|
func getBrainStartTime(rs *models.CloudbrainInfo) string { |
|
|
|
timeString := time.Unix(int64(rs.Cloudbrain.StartTime), 0).Format(CREATE_TIME_FORMAT) |
|
|
|
if timeString != "1970/01/01 08:00:00" { |
|
|
|
return timeString |
|
|
|
} else { |
|
|
|
return "0" |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
func getBrainEndTime(rs *models.CloudbrainInfo) string { |
|
|
|
timeString := time.Unix(int64(rs.Cloudbrain.EndTime), 0).Format(CREATE_TIME_FORMAT) |
|
|
|
if timeString != "1970/01/01 08:00:00" { |
|
|
|
return timeString |
|
|
|
} else { |
|
|
|
return "0" |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
func getBrainWaitTime(rs *models.CloudbrainInfo) string { |
|
|
|
waitTime := rs.Cloudbrain.StartTime - rs.Cloudbrain.CreatedUnix |
|
|
|
if waitTime <= 0 { |
|
|
|
return "0" |
|
|
|
} else { |
|
|
|
return models.ConvertDurationToStr(int64(waitTime)) |
|
|
|
} |
|
|
|
} |