diff --git a/internal/logic/admin/console/queryServerTotalDataLogic.go b/internal/logic/admin/console/queryServerTotalDataLogic.go index 6670d62..c21ad57 100644 --- a/internal/logic/admin/console/queryServerTotalDataLogic.go +++ b/internal/logic/admin/console/queryServerTotalDataLogic.go @@ -104,19 +104,19 @@ func (l *QueryServerTotalDataLogic) QueryServerTotalData() (resp *types.ServerTo return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), " Query server traffic failed: %v", err.Error()) } - todayServerRanking := make([]types.ServerTrafficData, len(todayTop10Server)) - for i, item := range todayTop10Server { + var todayServerRanking []types.ServerTrafficData + for _, item := range todayTop10Server { info, err := l.svcCtx.NodeModel.FindOneServer(l.ctx, item.ServerId) if err != nil { l.Errorw("[QueryServerTotalDataLogic] FindOneServer error", logger.Field("error", err.Error()), logger.Field("server_id", item.ServerId)) continue } - todayServerRanking[i] = types.ServerTrafficData{ + todayServerRanking = append(todayServerRanking, types.ServerTrafficData{ ServerId: item.ServerId, Name: info.Name, Upload: item.Upload, Download: item.Download, - } + }) } // query server traffic rank yesterday @@ -133,19 +133,18 @@ func (l *QueryServerTotalDataLogic) QueryServerTotalData() (resp *types.ServerTo if err != nil { l.Errorw("[QueryServerTotalDataLogic] Unmarshal yesterday server traffic rank log error", logger.Field("error", err.Error())) } - yesterdayTop10Server = make([]types.ServerTrafficData, len(rank.Rank)) - for i, v := range rank.Rank { + for _, v := range rank.Rank { info, err := l.svcCtx.NodeModel.FindOneServer(l.ctx, v.ServerId) if err != nil { l.Errorw("[QueryServerTotalDataLogic] FindOneServer error", logger.Field("error", err.Error()), logger.Field("server_id", v.ServerId)) continue } - yesterdayTop10Server[i] = types.ServerTrafficData{ + yesterdayTop10Server = append(yesterdayTop10Server, types.ServerTrafficData{ ServerId: v.ServerId, Name: info.Name, Upload: v.Upload, Download: v.Download, - } + }) } }