migration: add migrations for NVIDIA MIG

Release.toml

@@ -1,4 +1,4 @@
-version = "1.31.0"
+version = "1.32.0"
 
 [migrations]
 "(0.3.1, 0.3.2)" = ["migrate_v0.3.2_admin-container-v0-5-0.lz4"]
@@ -393,4 +393,8 @@ version = "1.31.0"
     "migrate_v1.31.0_public-admin-container-v0-11-16.lz4",
     "migrate_v1.31.0_aws-control-container-v0-7-20.lz4",
     "migrate_v1.31.0_public-control-container-v0-7-20.lz4",
+    "migrate_v1.31.0_kubelet-device-plugins-mig-settings.lz4",
+]
+"(1.31.0, 1.32.0)" = [
+    "migrate_v1.31.0_kubelet-device-plugins-mig-settings.lz4",
 ]

sources/Cargo.toml

@@ -83,6 +83,7 @@ members = [
     "settings-migrations/v1.31.0/public-admin-container-v0-11-16",
     "settings-migrations/v1.31.0/aws-control-container-v0-7-20",
     "settings-migrations/v1.31.0/public-control-container-v0-7-20",
+    "settings-migrations/v1.32.0/kubelet-device-plugins-mig-settings",
 
     "settings-plugins/aws-dev",
     "settings-plugins/aws-ecs-1",
@@ -158,23 +159,23 @@ tag = "bottlerocket-defaults-helper-v0.1.0"
 version = "0.1.0"
 
 [workspace.dependencies.bottlerocket-modeled-types]
-git = "https://github.com/bottlerocket-os/bottlerocket-settings-sdk"
-tag = "bottlerocket-settings-models-v0.7.0"
-version = "0.7.0"
+git = "https://github.com/piyush-jena/bottlerocket-settings-sdk"
+tag = "bottlerocket-settings-models-v0.8.0"
+version = "0.8.0"
 
 [workspace.dependencies.bottlerocket-settings-models]
-git = "https://github.com/bottlerocket-os/bottlerocket-settings-sdk"
-tag = "bottlerocket-settings-models-v0.7.0"
-version = "0.7.0"
+git = "https://github.com/piyush-jena/bottlerocket-settings-sdk"
+tag = "bottlerocket-settings-models-v0.8.0"
+version = "0.8.0"
 
 [workspace.dependencies.bottlerocket-settings-plugin]
 git = "https://github.com/bottlerocket-os/bottlerocket-settings-sdk"
-tag = "bottlerocket-settings-models-v0.7.0"
+tag = "bottlerocket-settings-plugin-v0.1.0"
 version = "0.1.0"
 
 [workspace.dependencies.settings-extension-oci-defaults]
-git = "https://github.com/bottlerocket-os/bottlerocket-settings-sdk"
-tag = "bottlerocket-settings-models-v0.7.0"
+git = "https://github.com/piyush-jena/bottlerocket-settings-sdk"
+tag = "bottlerocket-settings-models-v0.8.0"
 version = "0.1.0"
 
 [profile.release]

sources/settings-migrations/v1.32.0/kubelet-device-plugins-mig-settings/Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "kubelet-device-plugins-mig-settings"
+version = "0.1.0"
+authors = ["Piyush Jena <[email protected]>"]
+license = "Apache-2.0 OR MIT"
+edition = "2021"
+publish = false
+# Don't rebuild crate just because of changes to README.
+exclude = ["README.md"]
+
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+migration-helpers.workspace = true

sources/settings-migrations/v1.32.0/kubelet-device-plugins-mig-settings/src/main.rs

@@ -0,0 +1,21 @@
+use migration_helpers::common_migrations::AddSettingsMigration;
+use migration_helpers::{migrate, Result};
+use std::process;
+
+/// We added new settings for configuring the NVIDIA k8s device plugin.
+fn run() -> Result<()> {
+    migrate(AddSettingsMigration(&[
+        "settings.kubelet-device-plugins.nvidia.device-partitioning-strategy",
+        "settings.kubelet-device-plugins.nvidia.mig",
+    ]))
+}
+
+// Returning a Result from main makes it print a Debug representation of the error, but with Snafu
+// we have nice Display representations of the error, so we wrap "main" (run) and print any error.
+// https://github.com/shepmaster/snafu/issues/110
+fn main() {
+    if let Err(e) = run() {
+        eprintln!("{}", e);
+        process::exit(1);
+    }
+}