1 /* 2 * Copyright (C) 2019 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 package com.android.tradefed.retry; 17 18 import com.android.tradefed.testtype.IRemoteTest; 19 20 import java.util.HashMap; 21 import java.util.List; 22 import java.util.Map; 23 24 /** 25 * Structure holding the statistics for a retry session of one {@link IRemoteTest}. Not all fields 26 * might be populated depending of the {@link com.android.tradefed.retry.RetryStrategy}. 27 */ 28 public class RetryStatistics { 29 // The time spent in retry. Always populated if retries or iterations occurred 30 public long mRetryTime = 0L; 31 public Map<Integer, Long> mAttemptIsolationCost = new HashMap<>(); 32 33 // Success and failure counts. Populated for RETRY_ANY_FAILURE. 34 public long mRetrySuccess = 0L; 35 public long mRetryFailure = 0L; 36 37 /** Helper method to aggregate the statistics of several retries. */ aggregateStatistics(List<RetryStatistics> stats)38 public static final RetryStatistics aggregateStatistics(List<RetryStatistics> stats) { 39 RetryStatistics aggregatedStats = new RetryStatistics(); 40 for (RetryStatistics s : stats) { 41 aggregatedStats.mRetryTime += s.mRetryTime; 42 aggregatedStats.mRetrySuccess += s.mRetrySuccess; 43 aggregatedStats.mRetryFailure += s.mRetryFailure; 44 } 45 return aggregatedStats; 46 } 47 isolationCostPerAttempt(int attempt, List<RetryStatistics> stats)48 public static long isolationCostPerAttempt(int attempt, List<RetryStatistics> stats) { 49 long isolationCost = 0L; 50 for (RetryStatistics s : stats) { 51 Long attemptCost = s.mAttemptIsolationCost.get(attempt); 52 if (attemptCost != null) { 53 isolationCost += attemptCost; 54 } 55 } 56 return isolationCost; 57 } 58 } 59