database support for PendingMgsUpdate for SP updates (#8291)

Author: davepacheco

nexus/db-model/src/deployment.rs

@@ -5,12 +5,13 @@
 //! Types for representing the deployed software and configuration in the
 //! database
 
-use crate::inventory::ZoneType;
+use crate::inventory::{SpMgsSlot, SpType, ZoneType};
 use crate::omicron_zone_config::{self, OmicronZoneNic};
 use crate::typed_uuid::DbTypedUuid;
 use crate::{
-    ArtifactHash, ByteCount, DbOximeterReadMode, Generation, MacAddr, Name,
-    SledState, SqlU8, SqlU16, SqlU32, TufArtifact, impl_enum_type, ipv6,
+    ArtifactHash, ByteCount, DbArtifactVersion, DbOximeterReadMode, Generation,
+    MacAddr, Name, SledState, SqlU8, SqlU16, SqlU32, TufArtifact,
+    impl_enum_type, ipv6,
 };
 use anyhow::{Context, Result, anyhow, bail};
 use chrono::{DateTime, Utc};
@@ -21,10 +22,10 @@ use nexus_db_schema::schema::{
     bp_clickhouse_keeper_zone_id_to_node_id,
     bp_clickhouse_server_zone_id_to_node_id, bp_omicron_dataset,
     bp_omicron_physical_disk, bp_omicron_zone, bp_omicron_zone_nic,
-    bp_oximeter_read_policy, bp_sled_metadata, bp_target,
+    bp_oximeter_read_policy, bp_pending_mgs_update_sp, bp_sled_metadata,
+    bp_target,
 };
 use nexus_sled_agent_shared::inventory::OmicronZoneDataset;
-use nexus_types::deployment::BlueprintDatasetDisposition;
 use nexus_types::deployment::BlueprintPhysicalDiskConfig;
 use nexus_types::deployment::BlueprintPhysicalDiskDisposition;
 use nexus_types::deployment::BlueprintTarget;
@@ -33,14 +34,18 @@ use nexus_types::deployment::BlueprintZoneDisposition;
 use nexus_types::deployment::BlueprintZoneType;
 use nexus_types::deployment::ClickhouseClusterConfig;
 use nexus_types::deployment::CockroachDbPreserveDowngrade;
+use nexus_types::deployment::PendingMgsUpdate;
+use nexus_types::deployment::PendingMgsUpdateDetails;
 use nexus_types::deployment::{
     BlueprintDatasetConfig, BlueprintZoneImageVersion, OximeterReadMode,
 };
+use nexus_types::deployment::{BlueprintDatasetDisposition, ExpectedVersion};
 use nexus_types::deployment::{BlueprintZoneImageSource, blueprint_zone_type};
 use nexus_types::deployment::{
     OmicronZoneExternalFloatingAddr, OmicronZoneExternalFloatingIp,
     OmicronZoneExternalSnatIp,
 };
+use nexus_types::inventory::BaseboardId;
 use omicron_common::api::internal::shared::NetworkInterface;
 use omicron_common::disk::DiskIdentity;
 use omicron_common::zpool_name::ZpoolName;
@@ -50,6 +55,7 @@ use omicron_uuid_kinds::{
     PhysicalDiskKind, SledKind, SledUuid, ZpoolKind, ZpoolUuid,
 };
 use std::net::{IpAddr, SocketAddrV6};
+use std::sync::Arc;
 use uuid::Uuid;
 
 /// See [`nexus_types::deployment::Blueprint`].
@@ -1245,3 +1251,40 @@ impl BpOximeterReadPolicy {
         }
     }
 }
+
+#[derive(Queryable, Clone, Debug, Selectable, Insertable)]
+#[diesel(table_name = bp_pending_mgs_update_sp)]
+pub struct BpPendingMgsUpdateSp {
+    pub blueprint_id: DbTypedUuid<BlueprintKind>,
+    pub hw_baseboard_id: Uuid,
+    pub sp_type: SpType,
+    pub sp_slot: SpMgsSlot,
+    pub artifact_sha256: ArtifactHash,
+    pub artifact_version: DbArtifactVersion,
+    pub expected_active_version: DbArtifactVersion,
+    pub expected_inactive_version: Option<DbArtifactVersion>,
+}
+
+impl BpPendingMgsUpdateSp {
+    pub fn into_generic(
+        self,
+        baseboard_id: Arc<BaseboardId>,
+    ) -> PendingMgsUpdate {
+        PendingMgsUpdate {
+            baseboard_id,
+            sp_type: self.sp_type.into(),
+            slot_id: u32::from(**self.sp_slot),
+            artifact_hash: self.artifact_sha256.into(),
+            artifact_version: (*self.artifact_version).clone(),
+            details: PendingMgsUpdateDetails::Sp {
+                expected_active_version: (*self.expected_active_version)
+                    .clone(),
+                expected_inactive_version: match self.expected_inactive_version
+                {
+                    Some(v) => ExpectedVersion::Version((*v).clone()),
+                    None => ExpectedVersion::NoValidVersion,
+                },
+            },
+        }
+    }
+}

nexus/db-model/src/lib.rs

@@ -237,6 +237,7 @@ pub use switch_interface::*;
 pub use switch_port::*;
 pub use target_release::*;
 pub use tuf_repo::*;
+pub use typed_uuid::DbTypedUuid;
 pub use typed_uuid::to_db_typed_uuid;
 pub use upstairs_repair::*;
 pub use user_builtin::*;

nexus/db-model/src/schema_versions.rs

@@ -16,7 +16,7 @@ use std::{collections::BTreeMap, sync::LazyLock};
 ///
 /// This must be updated when you change the database schema.  Refer to
 /// schema/crdb/README.adoc in the root of this repository for details.
-pub const SCHEMA_VERSION: Version = Version::new(153, 0, 0);
+pub const SCHEMA_VERSION: Version = Version::new(154, 0, 0);
 
 /// List of all past database schema versions, in *reverse* order
 ///
@@ -28,6 +28,7 @@ static KNOWN_VERSIONS: LazyLock<Vec<KnownVersion>> = LazyLock::new(|| {
         // |  leaving the first copy as an example for the next person.
         // v
         // KnownVersion::new(next_int, "unique-dirname-with-the-sql-files"),
+        KnownVersion::new(154, "add-pending-mgs-updates"),
         KnownVersion::new(153, "chicken-switches"),
         KnownVersion::new(152, "ereports"),
         KnownVersion::new(151, "zone-image-resolver-inventory"),