x

Author: coderbradlee

action/protocol/staking/protocol.go

@@ -474,8 +474,11 @@ func (p *Protocol) ReadState(ctx context.Context, sr protocol.StateReader, metho
 	case iotexapi.ReadStakingDataMethod_CANDIDATE_BY_ADDRESS:
 		resp, height, err = readStateCandidateByAddress(ctx, csr, r.GetCandidateByAddress())
 	case iotexapi.ReadStakingDataMethod_TOTAL_STAKING_AMOUNT:
-		if p.archiveMode && p.hu.IsPre(config.Greenland, height) {
+		if p.archiveMode && p.hu.IsPre(config.Greenland, inputHeight) {
 			resp, height, err = readStateTotalStakingAmountFromIndexer(sr, r.GetTotalStakingAmount(), inputHeight)
+			if err != nil {
+				resp, height, err = readStateTotalStakingAmount(ctx, csr, r.GetTotalStakingAmount())
+			}
 		} else {
 			resp, height, err = readStateTotalStakingAmount(ctx, csr, r.GetTotalStakingAmount())
 		}