提交代码。

Signed-off-by: zouap <zouap@pcl.ac.cn>

routers/repo/user_data_analysis.go

@@ -186,6 +186,75 @@ func writeExcel(row int, xlsx *excelize.File, sheetName string, userRecord *mode
 	formatTime = userRecord.DataDate
 	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, formatTime)
 }
+
+func writeExcelPage(row int, xlsx *excelize.File, sheetName string, userRecord *models.UserBusinessAnalysis) {
+	rows := fmt.Sprint(row)
+	var tmp byte
+	tmp = 0
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.ID)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.Name)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", userRecord.UserIndex))
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", userRecord.UserIndexPrimitive))
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CodeMergeCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.IssueCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommentCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.FocusRepoCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.StarRepoCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.LoginCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.WatchedCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitCodeSize)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.SolveIssueCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.EncyclopediasCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CreateRepoCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", userRecord.OpenIIndex))
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CloudBrainTaskNum)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, fmt.Sprintf("%.2f", float64(userRecord.CloudBrainRunTime)/3600))
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitDatasetNum)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CommitModelCount)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.FocusOtherUser)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectDataset)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectedDataset)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.RecommendDataset)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectImage)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.CollectedImage)
+	tmp = tmp + 1
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, userRecord.RecommendImage)
+	tmp = tmp + 1
+	formatTime := userRecord.RegistDate.Format("2006-01-02 15:04:05")
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, formatTime[0:len(formatTime)-3])
+	tmp = tmp + 1
+
+	formatTime = userRecord.DataDate
+	xlsx.SetCellValue(sheetName, getColumn(tmp)+rows, formatTime)
+}
+
 func getColumn(tmp byte) string {
 	var tmpA byte
 	tmpA = 'A'
@@ -481,36 +550,13 @@ func QueryUserStaticDataPage(ctx *context.Context) {
 	}
 
 	if IsReturnFile {
-		re, count := models.QueryUserStaticDataAll(pageOpts)
-		log.Info("return count=" + fmt.Sprint(count))
-		//writer exec file.
-		xlsx := excelize.NewFile()
+		//re, count := models.QueryUserStaticDataAll(pageOpts)
+		re, count := models.QueryUserStaticDataPage(pageOpts)
 		sheetName := ctx.Tr("user.static.sheetname")
-		index := xlsx.NewSheet(sheetName)
-		xlsx.DeleteSheet("Sheet1")
-
-		dataHeader := getExcelHeader(ctx)
-		for k, v := range dataHeader {
-			//设置单元格的值
-			xlsx.SetCellValue(sheetName, k, v)
-		}
-
-		for i, userRecord := range re {
-			row := i + 2
-			writeExcel(row, xlsx, sheetName, userRecord)
-		}
-
-		//设置默认打开的表单
-		xlsx.SetActiveSheet(index)
-
-		filename := sheetName + "_" + ctx.Tr("user.static.all") + ".xlsx"
-
-		ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(filename))
-		ctx.Resp.Header().Set("Content-Type", "application/octet-stream")
-		if _, err := xlsx.WriteTo(ctx.Resp); err != nil {
-			log.Info("writer exel error." + err.Error())
-		}
+		filename := sheetName + "_" + startDate + "_" + endDate + ".xlsx"
 
+		go writeFileToDisk(ctx, count, re, filename)
+		ctx.JSON(http.StatusOK, filename)
 	} else {
 		mapInterface := make(map[string]interface{})
 		re, count := models.QueryUserStaticDataPage(pageOpts)
@@ -520,6 +566,38 @@ func QueryUserStaticDataPage(ctx *context.Context) {
 	}
 }
 
+func writeFileToDisk(ctx *context.Context, count int64, re []*models.UserBusinessAnalysis, filename string) {
+	log.Info("return count=" + fmt.Sprint(count))
+	//writer exec file.
+	xlsx := excelize.NewFile()
+	sheetName := ctx.Tr("user.static.sheetname")
+	index := xlsx.NewSheet(sheetName)
+	xlsx.DeleteSheet("Sheet1")
+
+	dataHeader := getExcelHeader(ctx)
+	for k, v := range dataHeader {
+		//设置单元格的值
+		xlsx.SetCellValue(sheetName, k, v)
+	}
+
+	for i, userRecord := range re {
+		row := i + 2
+		writeExcelPage(row, xlsx, sheetName, userRecord)
+	}
+
+	//设置默认打开的表单
+	xlsx.SetActiveSheet(index)
+
+	//ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+url.QueryEscape(filename))
+	//ctx.Resp.Header().Set("Content-Type", "application/octet-stream")
+	filename = setting.AppDataPath + "/useranalysis/" + filename
+	if err := xlsx.SaveAs(filename); err != nil {
+		log.Info("writer exel error." + err.Error())
+	} else {
+		log.Info("write to file succeed, filepath=" + filename)
+	}
+}
+
 func TimingCountDataByDateAndReCount(date string, isReCount bool) {
 
 	t, _ := time.Parse("2006-01-02", date)