models/user_business_analysis.go

@@ -579,18 +579,19 @@ func refreshUserStaticTable(wikiCountMap map[string]int, tableName string, pageS
 	var CommitCodeSizeMap map[string]*git.UserKPIStats
 	var err error
 	var existCommitCodeSize map[int64]int
-	if tableName == "user_business_analysis_all" || tableName == "user_business_analysis_current_year" {
-
+	if tableName == "user_business_analysis_all" {
 		oneDayStartTime := pageEndTime.AddDate(0, 0, -1)
+		oneDayStartTime = time.Date(oneDayStartTime.Year(), oneDayStartTime.Month(), oneDayStartTime.Day(), 0, 0, 0, 1, oneDayStartTime.Location())
 		if oneDayStartTime.Format("2006-01-02") == pageStartTime.Format("2006-01-02") {
 			existCommitCodeSize = make(map[int64]int, 0)
 		} else {
 			existCommitCodeSize = queryCommitCodeSizeFromDb("public." + tableName)
 		}
+		oneDayEndTime := time.Date(oneDayStartTime.Year(), oneDayStartTime.Month(), oneDayStartTime.Day(), 23, 59, 59, 1, oneDayStartTime.Location())
 		log.Info("GetAllUserKPIStats oneDayStartTime=" + oneDayStartTime.Format("2006-01-02 15:04:05"))
-		log.Info("GetAllUserKPIStats pageEndTime=" + pageEndTime.Format("2006-01-02 15:04:05"))
+		log.Info("GetAllUserKPIStats pageEndTime=" + oneDayEndTime.Format("2006-01-02 15:04:05"))
 		log.Info("existCommitCodeSize len=" + fmt.Sprint(len(existCommitCodeSize)))
-		CommitCodeSizeMap, err = GetAllUserKPIStats(oneDayStartTime, pageEndTime)
+		CommitCodeSizeMap, err = GetAllUserKPIStats(oneDayStartTime, oneDayEndTime)
 		if err != nil {
 			log.Info("query commit code errr.")
 		} else {

models/user_year_summary.go

@@ -342,7 +342,7 @@ func queryUserYearModel(start_unix int64, end_unix int64) map[int64]map[string]i
 	var indexTotal int64
 	indexTotal = 0
 	for {
-		sess.Select("id,user_id,download_count,reference_count").Table("ai_model_manage").Where(cond).OrderBy("id asc").Limit(PAGE_SIZE, int(indexTotal))
+		sess.Select("id,name,user_id,download_count,reference_count").Table("ai_model_manage").Where(cond).OrderBy("id asc").Limit(PAGE_SIZE, int(indexTotal))
 		aiModelList := make([]*AiModelManage, 0)
 		sess.Find(&aiModelList)
 		log.Info("query user year AiModelManage size=" + fmt.Sprint(len(aiModelList)))