185 lines
6.4 KiB
Go
185 lines
6.4 KiB
Go
// Copyright (C) MongoDB, Inc. 2017-present.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License. You may obtain
|
|
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
package bsoncore
|
|
|
|
// DocumentBuilder builds a bson document
|
|
type DocumentBuilder struct {
|
|
doc []byte
|
|
indexes []int32
|
|
}
|
|
|
|
// startDocument reserves the document's length and set the index to where the length begins
|
|
func (db *DocumentBuilder) startDocument() *DocumentBuilder {
|
|
var index int32
|
|
index, db.doc = AppendDocumentStart(db.doc)
|
|
db.indexes = append(db.indexes, index)
|
|
return db
|
|
}
|
|
|
|
// NewDocumentBuilder creates a new DocumentBuilder
|
|
func NewDocumentBuilder() *DocumentBuilder {
|
|
return (&DocumentBuilder{}).startDocument()
|
|
}
|
|
|
|
// Build updates the length of the document and index to the beginning of the documents length
|
|
// bytes, then returns the document (bson bytes)
|
|
func (db *DocumentBuilder) Build() Document {
|
|
last := len(db.indexes) - 1
|
|
db.doc, _ = AppendDocumentEnd(db.doc, db.indexes[last])
|
|
db.indexes = db.indexes[:last]
|
|
return db.doc
|
|
}
|
|
|
|
// AppendInt32 will append an int32 element using key and i32 to DocumentBuilder.doc
|
|
func (db *DocumentBuilder) AppendInt32(key string, i32 int32) *DocumentBuilder {
|
|
db.doc = AppendInt32Element(db.doc, key, i32)
|
|
return db
|
|
}
|
|
|
|
// AppendDocument will append a bson embedded document element using key
|
|
// and doc to DocumentBuilder.doc
|
|
func (db *DocumentBuilder) AppendDocument(key string, doc []byte) *DocumentBuilder {
|
|
db.doc = AppendDocumentElement(db.doc, key, doc)
|
|
return db
|
|
}
|
|
|
|
// AppendArray will append a bson array using key and arr to DocumentBuilder.doc
|
|
func (db *DocumentBuilder) AppendArray(key string, arr []byte) *DocumentBuilder {
|
|
db.doc = AppendHeader(db.doc, TypeArray, key)
|
|
db.doc = AppendArray(db.doc, arr)
|
|
return db
|
|
}
|
|
|
|
// AppendDouble will append a double element using key and f to DocumentBuilder.doc
|
|
func (db *DocumentBuilder) AppendDouble(key string, f float64) *DocumentBuilder {
|
|
db.doc = AppendDoubleElement(db.doc, key, f)
|
|
return db
|
|
}
|
|
|
|
// AppendString will append str to DocumentBuilder.doc with the given key
|
|
func (db *DocumentBuilder) AppendString(key string, str string) *DocumentBuilder {
|
|
db.doc = AppendStringElement(db.doc, key, str)
|
|
return db
|
|
}
|
|
|
|
// AppendObjectID will append oid to DocumentBuilder.doc with the given key
|
|
func (db *DocumentBuilder) AppendObjectID(key string, oid objectID) *DocumentBuilder {
|
|
db.doc = AppendObjectIDElement(db.doc, key, oid)
|
|
return db
|
|
}
|
|
|
|
// AppendBinary will append a BSON binary element using key, subtype, and
|
|
// b to db.doc
|
|
func (db *DocumentBuilder) AppendBinary(key string, subtype byte, b []byte) *DocumentBuilder {
|
|
db.doc = AppendBinaryElement(db.doc, key, subtype, b)
|
|
return db
|
|
}
|
|
|
|
// AppendUndefined will append a BSON undefined element using key to db.doc
|
|
func (db *DocumentBuilder) AppendUndefined(key string) *DocumentBuilder {
|
|
db.doc = AppendUndefinedElement(db.doc, key)
|
|
return db
|
|
}
|
|
|
|
// AppendBoolean will append a boolean element using key and b to db.doc
|
|
func (db *DocumentBuilder) AppendBoolean(key string, b bool) *DocumentBuilder {
|
|
db.doc = AppendBooleanElement(db.doc, key, b)
|
|
return db
|
|
}
|
|
|
|
// AppendDateTime will append a datetime element using key and dt to db.doc
|
|
func (db *DocumentBuilder) AppendDateTime(key string, dt int64) *DocumentBuilder {
|
|
db.doc = AppendDateTimeElement(db.doc, key, dt)
|
|
return db
|
|
}
|
|
|
|
// AppendNull will append a null element using key to db.doc
|
|
func (db *DocumentBuilder) AppendNull(key string) *DocumentBuilder {
|
|
db.doc = AppendNullElement(db.doc, key)
|
|
return db
|
|
}
|
|
|
|
// AppendRegex will append pattern and options using key to db.doc
|
|
func (db *DocumentBuilder) AppendRegex(key, pattern, options string) *DocumentBuilder {
|
|
db.doc = AppendRegexElement(db.doc, key, pattern, options)
|
|
return db
|
|
}
|
|
|
|
// AppendDBPointer will append ns and oid to using key to db.doc
|
|
func (db *DocumentBuilder) AppendDBPointer(key string, ns string, oid objectID) *DocumentBuilder {
|
|
db.doc = AppendDBPointerElement(db.doc, key, ns, oid)
|
|
return db
|
|
}
|
|
|
|
// AppendJavaScript will append js using the provided key to db.doc
|
|
func (db *DocumentBuilder) AppendJavaScript(key, js string) *DocumentBuilder {
|
|
db.doc = AppendJavaScriptElement(db.doc, key, js)
|
|
return db
|
|
}
|
|
|
|
// AppendSymbol will append a BSON symbol element using key and symbol db.doc
|
|
func (db *DocumentBuilder) AppendSymbol(key, symbol string) *DocumentBuilder {
|
|
db.doc = AppendSymbolElement(db.doc, key, symbol)
|
|
return db
|
|
}
|
|
|
|
// AppendCodeWithScope will append code and scope using key to db.doc
|
|
func (db *DocumentBuilder) AppendCodeWithScope(key string, code string, scope Document) *DocumentBuilder {
|
|
db.doc = AppendCodeWithScopeElement(db.doc, key, code, scope)
|
|
return db
|
|
}
|
|
|
|
// AppendTimestamp will append t and i to db.doc using provided key
|
|
func (db *DocumentBuilder) AppendTimestamp(key string, t, i uint32) *DocumentBuilder {
|
|
db.doc = AppendTimestampElement(db.doc, key, t, i)
|
|
return db
|
|
}
|
|
|
|
// AppendInt64 will append i64 to dst using key to db.doc
|
|
func (db *DocumentBuilder) AppendInt64(key string, i64 int64) *DocumentBuilder {
|
|
db.doc = AppendInt64Element(db.doc, key, i64)
|
|
return db
|
|
}
|
|
|
|
// AppendDecimal128 will append d128 to db.doc using provided key
|
|
func (db *DocumentBuilder) AppendDecimal128(key string, high, low uint64) *DocumentBuilder {
|
|
db.doc = AppendDecimal128Element(db.doc, key, high, low)
|
|
return db
|
|
}
|
|
|
|
// AppendMaxKey will append a max key element using key to db.doc
|
|
func (db *DocumentBuilder) AppendMaxKey(key string) *DocumentBuilder {
|
|
db.doc = AppendMaxKeyElement(db.doc, key)
|
|
return db
|
|
}
|
|
|
|
// AppendMinKey will append a min key element using key to db.doc
|
|
func (db *DocumentBuilder) AppendMinKey(key string) *DocumentBuilder {
|
|
db.doc = AppendMinKeyElement(db.doc, key)
|
|
return db
|
|
}
|
|
|
|
// AppendValue will append a BSON element with the provided key and value to the document.
|
|
func (db *DocumentBuilder) AppendValue(key string, val Value) *DocumentBuilder {
|
|
db.doc = AppendValueElement(db.doc, key, val)
|
|
return db
|
|
}
|
|
|
|
// StartDocument starts building an inline document element with the provided key
|
|
// After this document is completed, the user must call finishDocument
|
|
func (db *DocumentBuilder) StartDocument(key string) *DocumentBuilder {
|
|
db.doc = AppendHeader(db.doc, TypeEmbeddedDocument, key)
|
|
db = db.startDocument()
|
|
return db
|
|
}
|
|
|
|
// FinishDocument builds the most recent document created
|
|
func (db *DocumentBuilder) FinishDocument() *DocumentBuilder {
|
|
db.doc = db.Build()
|
|
return db
|
|
}
|