Merge pull request #183 from microsoft/user/vishalk/add-osversion

Add OsVersion to Tags
microsoft · Oct 18, 2023 · f07e232 · f07e232
2 parents bb185cc + 72327f9
commit f07e232
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pkg/constant/constant.go b/pkg/constant/constant.go
@@ -12,4 +12,5 @@ const (
 	CertificateValidityThreshold float64 = (30.0 / 100.0)
 	RenewalBackoff               float64 = (2.0 / 100.0)
 	OsRegistrationStatus         string  = "osRegistrationStatus"
+	OsVersion                    string  = "osVersion"
 )
diff --git a/services/cloud/node/node.go b/services/cloud/node/node.go
@@ -82,6 +82,7 @@ func getNodeStatuses(node *wssdcloud.Node) map[string]*string {
 func generateNodeTags(node *wssdcloud.Node) map[string]*string {
 	tags := make(map[string]*string)
 	populateOsRegistrationStatusTag(tags, node)
+	populateOsVersionTag(tags, node)
 	if len(tags) > 0 {
 		return tags
 	}
@@ -94,3 +95,9 @@ func populateOsRegistrationStatusTag(tags map[string]*string, node *wssdcloud.No
 		tags[constant.OsRegistrationStatus] = &osRegistrationStatus
 	}
 }
+
+func populateOsVersionTag(tags map[string]*string, node *wssdcloud.Node) {
+	if node.Info != nil && node.Info.OsInfo != nil {
+		tags[constant.OsVersion] = &node.Info.OsInfo.Osversion
+	}
+}