From 64caf6daff734a05ac46b266750a9a920eacd5b7 Mon Sep 17 00:00:00 2001
From: Bryce Soghigian <bsoghigian@gmail.com>
Date: Tue, 3 Jun 2025 14:37:16 -0700
Subject: [PATCH] feat: nvme disk size added

---
 const.go | 2 ++
 sku.go   | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/const.go b/const.go
index 1b6f8fb..d57a0b3 100644
--- a/const.go
+++ b/const.go
@@ -35,6 +35,8 @@ const (
 	// UltraSSDAvailable identifies the capability for ultra ssd
 	// enablement.
 	UltraSSDAvailable = "UltraSSDAvailable"
+	// NvmeDiskSizeInMiB declares the disk size of the Nvme drive
+	NvmeDiskSizeInMiB = "NvmeDiskSizeInMiB"
 	// CachedDiskBytes identifies the maximum size of the cache disk for
 	// a vm.
 	CachedDiskBytes = "CachedDiskBytes"
diff --git a/sku.go b/sku.go
index 436ad93..8f8d8cc 100644
--- a/sku.go
+++ b/sku.go
@@ -59,6 +59,11 @@ func (s *SKU) Memory() (float64, error) {
 	return s.GetCapabilityFloatQuantity(MemoryGB)
 }
 
+// NvmeDiskSizeInMiB returns the integer value for the disk size
+func (s *SKU) NvmeDiskSizeInMiB() (int64, error) {
+	return s.GetCapabilityIntegerQuantity(NvmeDiskSizeInMiB)
+}
+
 // MaxCachedDiskBytes returns the number of bytes available for the
 // cache if it exists on this VM size.
 func (s *SKU) MaxCachedDiskBytes() (int64, error) {