upgrade: patch spec.dataEngine for volumes, engines and replicas

Longhorn 7095 Signed-off-by: Derek Su <derek.su@suse.com>
longhorn · Dec 28, 2023 · 439a93b · 439a93b
1 parent 6208527
commit 439a93b
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/upgrade/v15xto160/upgrade.go b/upgrade/v15xto160/upgrade.go
@@ -69,6 +69,12 @@ func upgradeVolumes(namespace string, lhClient *lhclientset.Clientset, resourceM
 			v.Spec.Image = v.Spec.EngineImage
 			v.Spec.EngineImage = ""
 		}
+
+		if v.Spec.BackendStoreDriver == "" {
+			v.Spec.DataEngine = longhorn.DataEngineTypeV1
+		} else {
+			v.Spec.DataEngine = longhorn.DataEngineType(v.Spec.BackendStoreDriver)
+		}
 	}
 
 	return nil
@@ -118,6 +124,12 @@ func upgradeEngines(namespace string, lhClient *lhclientset.Clientset, resourceM
 			e.Spec.Image = e.Spec.EngineImage
 			e.Spec.EngineImage = ""
 		}
+
+		if e.Spec.BackendStoreDriver == "" {
+			e.Spec.DataEngine = longhorn.DataEngineTypeV1
+		} else {
+			e.Spec.DataEngine = longhorn.DataEngineType(e.Spec.BackendStoreDriver)
+		}
 	}
 
 	return nil
@@ -142,6 +154,12 @@ func upgradeReplicas(namespace string, lhClient *lhclientset.Clientset, resource
 			r.Spec.EngineImage = ""
 		}
 
+		if r.Spec.BackendStoreDriver == "" {
+			r.Spec.DataEngine = longhorn.DataEngineTypeV1
+		} else {
+			r.Spec.DataEngine = longhorn.DataEngineType(r.Spec.BackendStoreDriver)
+		}
+
 		r.Spec.EvictionRequested = r.Status.EvictionRequested
 	}