add client bulkWrite

Author: qingyang-hu

internal/integration/unified/client_operation_execution.go

@@ -235,6 +235,8 @@ func executeClientBulkWrite(ctx context.Context, operation *operation) (*operati
 				return nil, err
 			}
 			opts.SetWriteConcern(c)
+		case "rawData":
+			opts.SetRawData(val.Boolean())
 		default:
 			return nil, fmt.Errorf("unrecognized bulkWrite option %q", key)
 		}

mongo/client.go

@@ -956,6 +956,7 @@ func (c *Client) BulkWrite(ctx context.Context, writes []ClientBulkWrite,
 		client:                   c,
 		selector:                 selector,
 		writeConcern:             wc,
+		rawData:                  bwo.RawData,
 	}
 	if bwo.VerboseResults == nil || !(*bwo.VerboseResults) {
 		op.errorsOnly = true

mongo/client_bulk_write.go

@@ -44,6 +44,7 @@ type clientBulkWrite struct {
 	client                   *Client
 	selector                 description.ServerSelector
 	writeConcern             *writeconcern.WriteConcern
+	rawData                  *bool
 
 	result ClientBulkWriteResult
 }
@@ -143,6 +144,10 @@ func (bw *clientBulkWrite) newCommand() func([]byte, description.SelectedServer)
 			}
 			dst = bsoncore.AppendDocumentElement(dst, "let", let)
 		}
+		// Set rawData for 8.2+ servers.
+		if bw.rawData != nil && desc.WireVersion != nil && driverutil.VersionRangeIncludes(*desc.WireVersion, 27) {
+			dst = bsoncore.AppendBooleanElement(dst, "rawData", *bw.rawData)
+		}
 		return dst, nil
 	}
 }

mongo/options/clientbulkwriteoptions.go

@@ -19,6 +19,7 @@ type ClientBulkWriteOptions struct {
 	Ordered                  *bool
 	Let                      interface{}
 	WriteConcern             *writeconcern.WriteConcern
+	RawData                  *bool
 	VerboseResults           *bool
 }
 
@@ -108,6 +109,18 @@ func (b *ClientBulkWriteOptionsBuilder) SetWriteConcern(wc *writeconcern.WriteCo
 	return b
 }
 
+// SetRawData sets the value for the RawData field. If true, it allows the CRUD operations to access timeseries
+// collections on the bucket-level. This option is only valid for MongoDB versions >= 9.0. The default value is false.
+func (b *ClientBulkWriteOptionsBuilder) SetRawData(rawData bool) *ClientBulkWriteOptionsBuilder {
+	b.Opts = append(b.Opts, func(opts *ClientBulkWriteOptions) error {
+		opts.RawData = &rawData
+
+		return nil
+	})
+
+	return b
+}
+
 // SetVerboseResults sets the value for the VerboseResults field. Specifies whether detailed
 // results for each successful operation should be included in the returned BulkWriteResult.
 // The defaults value is false.