Roc 2 gadi atpakaļ
vecāks
revīzija
26a5a1bf7b
1 mainītis faili ar 7 papildinājumiem un 0 dzēšanām
  1. 7 0
      services/data/predict_edb_info.go

+ 7 - 0
services/data/predict_edb_info.go

@@ -83,6 +83,13 @@ func getPredictEdbDayList(startDate, endDate time.Time, frequency string) (dayLi
 			}
 			currDate = currDate.AddDate(0, 0, 1)
 		}
+	case "年度":
+		for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); {
+			currDate = time.Date(currDate.Year()+1, 12, 31, 0, 0, 0, 0, time.Now().Location())
+			if !currDate.After(endDate) && !currDate.Equal(startDate) {
+				dayList = append(dayList, currDate)
+			}
+		}
 	}
 	return
 }