add insert

Author: qingyang-hu

mongo/collection.go

@@ -324,6 +324,9 @@ func (coll *Collection) insert(
 	if args.Ordered != nil {
 		op = op.Ordered(*args.Ordered)
 	}
+	if args.RawData != nil {
+		op = op.RawData(*args.RawData)
+	}
 	retry := driver.RetryNone
 	if coll.client.retryWrites {
 		retry = driver.RetryOncePerCommand
@@ -375,6 +378,9 @@ func (coll *Collection) InsertOne(ctx context.Context, document interface{},
 	if args.Comment != nil {
 		imOpts.SetComment(args.Comment)
 	}
+	if args.RawData != nil {
+		imOpts = imOpts.SetRawData(*args.RawData)
+	}
 	res, err := coll.insert(ctx, []interface{}{document}, imOpts)
 
 	rr, err := processWriteError(err)

mongo/options/insertoptions.go

@@ -13,6 +13,7 @@ package options
 type InsertOneOptions struct {
 	BypassDocumentValidation *bool
 	Comment                  interface{}
+	RawData                  *bool
 }
 
 // InsertOneOptionsBuilder represents functional options that configure an
@@ -53,6 +54,18 @@ func (ioo *InsertOneOptionsBuilder) SetComment(comment interface{}) *InsertOneOp
 	return ioo
 }
 
+// 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 (ioo *InsertOneOptionsBuilder) SetRawData(rawData bool) *InsertOneOptionsBuilder {
+	ioo.Opts = append(ioo.Opts, func(ioo *InsertOneOptions) error {
+		ioo.RawData = &rawData
+
+		return nil
+	})
+
+	return ioo
+}
+
 // InsertManyOptions represents arguments that can be used to configure an
 // InsertMany operation.
 //
@@ -61,6 +74,7 @@ type InsertManyOptions struct {
 	BypassDocumentValidation *bool
 	Comment                  interface{}
 	Ordered                  *bool
+	RawData                  *bool
 }
 
 // InsertManyOptionsBuilder contains options to configure insert operations.
@@ -121,3 +135,15 @@ func (imo *InsertManyOptionsBuilder) SetOrdered(b bool) *InsertManyOptionsBuilde
 
 	return imo
 }
+
+// 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 (imo *InsertManyOptionsBuilder) SetRawData(rawData bool) *InsertManyOptionsBuilder {
+	imo.Opts = append(imo.Opts, func(opts *InsertManyOptions) error {
+		opts.RawData = &rawData
+
+		return nil
+	})
+
+	return imo
+}

x/mongo/driver/operation/insert.go

@@ -42,6 +42,7 @@ type Insert struct {
 	result                   InsertResult
 	serverAPI                *driver.ServerAPIOptions
 	timeout                  *time.Duration
+	rawData                  *bool
 	logger                   *logger.Logger
 }
 
@@ -132,6 +133,10 @@ func (i *Insert) command(dst []byte, desc description.SelectedServer) ([]byte, e
 	if i.ordered != nil {
 		dst = bsoncore.AppendBooleanElement(dst, "ordered", *i.ordered)
 	}
+	// Set rawData for 8.2+ servers.
+	if i.rawData != nil && desc.WireVersion != nil && driverutil.VersionRangeIncludes(*desc.WireVersion, 27) {
+		dst = bsoncore.AppendBooleanElement(dst, "rawData", *i.rawData)
+	}
 	return dst, nil
 }
 
@@ -318,3 +323,13 @@ func (i *Insert) Authenticator(authenticator driver.Authenticator) *Insert {
 	i.authenticator = authenticator
 	return i
 }
+
+// RawData sets the rawData to access timeseries data in the compressed format.
+func (i *Insert) RawData(rawData bool) *Insert {
+	if i == nil {
+		i = new(Insert)
+	}
+
+	i.rawData = &rawData
+	return i
+}