From 19ea73917a6d2d298206067452bd3e1decf22c14 Mon Sep 17 00:00:00 2001
From: Tanmay Satam <tsatam@redhat.com>
Date: Wed, 20 Mar 2024 18:21:05 -0400
Subject: [PATCH] Include generated client

---
 .../cosmosdb/zz_generated_collection.go       | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/pkg/database/cosmosdb/zz_generated_collection.go b/pkg/database/cosmosdb/zz_generated_collection.go
index 97c72b5d060..b4a6bef1196 100644
--- a/pkg/database/cosmosdb/zz_generated_collection.go
+++ b/pkg/database/cosmosdb/zz_generated_collection.go
@@ -170,6 +170,8 @@ type PartitionKeyRanges struct {
 
 // PartitionKeyRange represents a partition key range
 type PartitionKeyRange struct {
+	MissingFields
+
 	ID                 string                  `json:"id,omitempty"`
 	ResourceID         string                  `json:"_rid,omitempty"`
 	Timestamp          int                     `json:"_ts,omitempty"`
@@ -192,6 +194,27 @@ const (
 	PartitionKeyRangeStatusOnline PartitionKeyRangeStatus = "online"
 )
 
+// MissingFields retains values that do not map to struct fields during JSON
+// marshalling/unmarshalling.  MissingFields implements
+// github.com/ugorji/go/codec.MissingFielder.
+type MissingFields struct {
+	m map[string]interface{}
+}
+
+// CodecMissingField is called to set a missing field and value pair
+func (mf *MissingFields) CodecMissingField(field []byte, value interface{}) bool {
+	if mf.m == nil {
+		mf.m = map[string]interface{}{}
+	}
+	(mf.m)[string(field)] = value
+	return true
+}
+
+// CodecMissingFields returns the set of fields which are not struct fields
+func (mf *MissingFields) CodecMissingFields() map[string]interface{} {
+	return mf.m
+}
+
 type collectionClient struct {
 	*databaseClient
 	path string