fix: can't recover after a upgrade failed

pkg/manager/member/tikv_upgrader.go

@@ -84,11 +84,16 @@ func (tku *tikvUpgrader) Upgrade(tc *v1alpha1.TidbCluster, oldSet *apps.Stateful
 			return controller.RequeueErrorf("tidbcluster: [%s/%s]'s tikv pod: [%s] have not label: %s", ns, tcName, podName, apps.ControllerRevisionHashLabelKey)
 		}
 
+		if store == nil {
+			setUpgradePartition(newSet, i)
+			continue
+		}
 		if revision == tc.Status.TiKV.StatefulSet.UpdateRevision {
+
 			if pod.Status.Phase != corev1.PodRunning {
 				return controller.RequeueErrorf("tidbcluster: [%s/%s]'s upgraded tikv pods are not all running", ns, tcName)
 			}
-			if store == nil || store.State != v1alpha1.TiKVStateUp {
+			if store.State != v1alpha1.TiKVStateUp {
 				return controller.RequeueErrorf("tidbcluster: [%s/%s]'s upgraded tikv are not all ready", ns, tcName)
 			}
 			err := tku.endEvictLeader(tc, i)
@@ -199,11 +204,6 @@ func (tku *tikvUpgrader) getStoreByOrdinal(tc *v1alpha1.TidbCluster, ordinal int
 			return &store
 		}
 	}
-	for _, store := range tc.Status.TiKV.TombstoneStores {
-		if store.PodName == podName {
-			return &store
-		}
-	}
 	return nil
 }