feat: mock training.
This commit is contained in:
@@ -351,6 +351,252 @@ func (sc *StatisticsController) StatisticsByRegion(c *gin.Context) {
|
||||
})
|
||||
}
|
||||
|
||||
type trainingSessionRegionStatisticsItem struct {
|
||||
RegionID uint32 `json:"regionId"`
|
||||
KindergartenName string `json:"kindergartenName"`
|
||||
Count int64 `json:"count"`
|
||||
StartedCount int64 `json:"startedCount"`
|
||||
EndedCount int64 `json:"endedCount"`
|
||||
CompletedCount int64 `json:"completedCount"`
|
||||
InProgressCount int64 `json:"inProgressCount"`
|
||||
TotalDurationMs int64 `json:"totalDurationMs"`
|
||||
AvgDurationMs float64 `json:"avgDurationMs"`
|
||||
EventTypeCounts map[string]int64 `json:"eventTypeCounts"`
|
||||
AppNameCounts map[string]int64 `json:"appNameCounts"`
|
||||
FlavorTypeCounts map[string]int64 `json:"flavorTypeCounts"`
|
||||
FirstPublishedAt *time.Time `json:"firstPublishedAt"`
|
||||
LastPublishedAt *time.Time `json:"lastPublishedAt"`
|
||||
}
|
||||
|
||||
func (sc *StatisticsController) TrainingSessionStatisticsByRegion(c *gin.Context) {
|
||||
regionIDStr := c.Query("regionId")
|
||||
flavorType := strings.TrimSpace(c.Query("flavorType"))
|
||||
startTimeStr := c.Query("startTime")
|
||||
endTimeStr := c.Query("endTime")
|
||||
|
||||
query := sc.DB.Model(&models.MqttTrainingSessionRecord{})
|
||||
if regionIDStr != "" {
|
||||
if regionID, err := strconv.ParseUint(regionIDStr, 10, 32); err == nil {
|
||||
query = query.Where("region_id = ?", uint32(regionID))
|
||||
}
|
||||
}
|
||||
if flavorType != "" {
|
||||
query = query.Where("flavor_type = ?", flavorType)
|
||||
}
|
||||
if startTimeStr != "" {
|
||||
if startTime, err := strconv.ParseInt(startTimeStr, 10, 64); err == nil {
|
||||
query = query.Where("published_at >= ?", startTime)
|
||||
}
|
||||
}
|
||||
if endTimeStr != "" {
|
||||
if endTime, err := strconv.ParseInt(endTimeStr, 10, 64); err == nil {
|
||||
query = query.Where("published_at <= ?", endTime)
|
||||
}
|
||||
}
|
||||
|
||||
type rawTrainingStats struct {
|
||||
RegionID *uint32
|
||||
Count int64
|
||||
StartedCount int64
|
||||
EndedCount int64
|
||||
CompletedCount int64
|
||||
InProgressCount int64
|
||||
TotalDurationMs int64
|
||||
FirstPublishedAt *int64
|
||||
LastPublishedAt *int64
|
||||
}
|
||||
|
||||
var rawResults []rawTrainingStats
|
||||
err := query.Select(`
|
||||
region_id,
|
||||
COUNT(*) as count,
|
||||
COALESCE(SUM(CASE WHEN started_at IS NOT NULL THEN 1 ELSE 0 END), 0) as started_count,
|
||||
COALESCE(SUM(CASE WHEN ended_at IS NOT NULL THEN 1 ELSE 0 END), 0) as ended_count,
|
||||
COALESCE(SUM(CASE WHEN started_at IS NOT NULL AND ended_at IS NOT NULL THEN 1 ELSE 0 END), 0) as completed_count,
|
||||
COALESCE(SUM(CASE WHEN started_at IS NOT NULL AND ended_at IS NULL THEN 1 ELSE 0 END), 0) as in_progress_count,
|
||||
COALESCE(SUM(CASE WHEN started_at IS NOT NULL AND ended_at IS NOT NULL AND ended_at >= started_at THEN ended_at - started_at ELSE 0 END), 0) as total_duration_ms,
|
||||
MIN(published_at) as first_published_at,
|
||||
MAX(published_at) as last_published_at
|
||||
`).Group("region_id").Scan(&rawResults).Error
|
||||
if err != nil {
|
||||
writeError(c, http.StatusInternalServerError, "failed to query training session statistics")
|
||||
return
|
||||
}
|
||||
|
||||
type trainingEventTypeCount struct {
|
||||
RegionID *uint32
|
||||
EventType string
|
||||
Count int64
|
||||
}
|
||||
var eventTypeResults []trainingEventTypeCount
|
||||
if err := query.Select("region_id, event_type, COUNT(*) as count").Group("region_id, event_type").Scan(&eventTypeResults).Error; err != nil {
|
||||
writeError(c, http.StatusInternalServerError, "failed to query training session event type statistics")
|
||||
return
|
||||
}
|
||||
|
||||
type trainingAppNameCount struct {
|
||||
RegionID *uint32
|
||||
AppName string
|
||||
Count int64
|
||||
}
|
||||
var appNameResults []trainingAppNameCount
|
||||
if err := query.Select("region_id, app_name, COUNT(*) as count").Group("region_id, app_name").Scan(&appNameResults).Error; err != nil {
|
||||
writeError(c, http.StatusInternalServerError, "failed to query training session app name statistics")
|
||||
return
|
||||
}
|
||||
|
||||
type trainingFlavorTypeCount struct {
|
||||
RegionID *uint32
|
||||
FlavorType string
|
||||
Count int64
|
||||
}
|
||||
var flavorTypeResults []trainingFlavorTypeCount
|
||||
if err := query.Select("region_id, flavor_type, COUNT(*) as count").Group("region_id, flavor_type").Scan(&flavorTypeResults).Error; err != nil {
|
||||
writeError(c, http.StatusInternalServerError, "failed to query training session flavor type statistics")
|
||||
return
|
||||
}
|
||||
|
||||
eventTypeMap := make(map[uint32]map[string]int64)
|
||||
for _, r := range eventTypeResults {
|
||||
regionID := uint32(0)
|
||||
if r.RegionID != nil {
|
||||
regionID = *r.RegionID
|
||||
}
|
||||
if eventTypeMap[regionID] == nil {
|
||||
eventTypeMap[regionID] = make(map[string]int64)
|
||||
}
|
||||
eventTypeMap[regionID][r.EventType] = r.Count
|
||||
}
|
||||
|
||||
appNameMap := make(map[uint32]map[string]int64)
|
||||
for _, r := range appNameResults {
|
||||
regionID := uint32(0)
|
||||
if r.RegionID != nil {
|
||||
regionID = *r.RegionID
|
||||
}
|
||||
if appNameMap[regionID] == nil {
|
||||
appNameMap[regionID] = make(map[string]int64)
|
||||
}
|
||||
appNameMap[regionID][r.AppName] = r.Count
|
||||
}
|
||||
|
||||
flavorTypeMap := make(map[uint32]map[string]int64)
|
||||
for _, r := range flavorTypeResults {
|
||||
regionID := uint32(0)
|
||||
if r.RegionID != nil {
|
||||
regionID = *r.RegionID
|
||||
}
|
||||
if flavorTypeMap[regionID] == nil {
|
||||
flavorTypeMap[regionID] = make(map[string]int64)
|
||||
}
|
||||
flavorTypeMap[regionID][r.FlavorType] = r.Count
|
||||
}
|
||||
|
||||
regionIDs := make([]uint32, 0, len(rawResults))
|
||||
for _, r := range rawResults {
|
||||
if r.RegionID != nil && *r.RegionID > 0 {
|
||||
regionIDs = append(regionIDs, *r.RegionID)
|
||||
}
|
||||
}
|
||||
kindergartenMap := make(map[uint32]string)
|
||||
if len(regionIDs) > 0 {
|
||||
var kindergartens []models.Kindergarten
|
||||
if err := sc.DB.Where("region_id IN ?", regionIDs).Find(&kindergartens).Error; err == nil {
|
||||
for _, k := range kindergartens {
|
||||
kindergartenMap[k.RegionID] = k.Name
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
overall := trainingSessionRegionStatisticsItem{
|
||||
EventTypeCounts: make(map[string]int64),
|
||||
AppNameCounts: make(map[string]int64),
|
||||
FlavorTypeCounts: make(map[string]int64),
|
||||
}
|
||||
regions := make(map[string]trainingSessionRegionStatisticsItem, len(rawResults))
|
||||
|
||||
for _, r := range rawResults {
|
||||
regionID := uint32(0)
|
||||
if r.RegionID != nil {
|
||||
regionID = *r.RegionID
|
||||
}
|
||||
avgDuration := float64(0)
|
||||
if r.CompletedCount > 0 {
|
||||
avgDuration = float64(r.TotalDurationMs) / float64(r.CompletedCount)
|
||||
}
|
||||
kgName := ""
|
||||
if regionID > 0 {
|
||||
kgName = kindergartenMap[regionID]
|
||||
}
|
||||
|
||||
var firstPublishedAt, lastPublishedAt *time.Time
|
||||
if r.FirstPublishedAt != nil {
|
||||
t := time.UnixMilli(*r.FirstPublishedAt)
|
||||
firstPublishedAt = &t
|
||||
}
|
||||
if r.LastPublishedAt != nil {
|
||||
t := time.UnixMilli(*r.LastPublishedAt)
|
||||
lastPublishedAt = &t
|
||||
}
|
||||
|
||||
item := trainingSessionRegionStatisticsItem{
|
||||
RegionID: regionID,
|
||||
KindergartenName: kgName,
|
||||
Count: r.Count,
|
||||
StartedCount: r.StartedCount,
|
||||
EndedCount: r.EndedCount,
|
||||
CompletedCount: r.CompletedCount,
|
||||
InProgressCount: r.InProgressCount,
|
||||
TotalDurationMs: r.TotalDurationMs,
|
||||
AvgDurationMs: avgDuration,
|
||||
EventTypeCounts: eventTypeMap[regionID],
|
||||
AppNameCounts: appNameMap[regionID],
|
||||
FlavorTypeCounts: flavorTypeMap[regionID],
|
||||
FirstPublishedAt: firstPublishedAt,
|
||||
LastPublishedAt: lastPublishedAt,
|
||||
}
|
||||
|
||||
regions[strconv.FormatUint(uint64(regionID), 10)] = item
|
||||
|
||||
overall.Count += r.Count
|
||||
overall.StartedCount += r.StartedCount
|
||||
overall.EndedCount += r.EndedCount
|
||||
overall.CompletedCount += r.CompletedCount
|
||||
overall.InProgressCount += r.InProgressCount
|
||||
overall.TotalDurationMs += r.TotalDurationMs
|
||||
|
||||
if firstPublishedAt != nil {
|
||||
if overall.FirstPublishedAt == nil || firstPublishedAt.Before(*overall.FirstPublishedAt) {
|
||||
overall.FirstPublishedAt = firstPublishedAt
|
||||
}
|
||||
}
|
||||
if lastPublishedAt != nil {
|
||||
if overall.LastPublishedAt == nil || lastPublishedAt.After(*overall.LastPublishedAt) {
|
||||
overall.LastPublishedAt = lastPublishedAt
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for _, r := range eventTypeResults {
|
||||
overall.EventTypeCounts[r.EventType] += r.Count
|
||||
}
|
||||
for _, r := range appNameResults {
|
||||
overall.AppNameCounts[r.AppName] += r.Count
|
||||
}
|
||||
for _, r := range flavorTypeResults {
|
||||
overall.FlavorTypeCounts[r.FlavorType] += r.Count
|
||||
}
|
||||
|
||||
if overall.CompletedCount > 0 {
|
||||
overall.AvgDurationMs = float64(overall.TotalDurationMs) / float64(overall.CompletedCount)
|
||||
}
|
||||
|
||||
writeSuccess(c, http.StatusOK, "query success", gin.H{
|
||||
"overall": overall,
|
||||
"regions": regions,
|
||||
})
|
||||
}
|
||||
|
||||
func (sc *StatisticsController) TimelineStatistics(c *gin.Context) {
|
||||
regionIDStr := c.Query("regionId")
|
||||
startTimeStr := c.Query("startTime")
|
||||
|
||||
Reference in New Issue
Block a user