The go-mongo-template project came to facilitate the use of the mongo driver in your go project, with self-management of sessions, simplicity in configuring the database and collection directly in the structure and much more. See below some implemented features:
- Simplicity in function calls without having to use getDatabase and getCollection.
- Database and collection configuration directly in the structure.
- Autofill id generated when inserting the document.
- Automatic rollback if an error occurs during transactions in a session. (Optional)
- Ease of opening and closing sessions, and also aborting or committing transactions.
- Pre-developed functions such as FindOneById, FindAll, FindPageable, Exists, among others.
- Easy and intuitive pagination scheme.
- Watching events has never been easier, with automatic calls to handler functions, with a context ready to work.
Use go get.
go get github.com/GabrielHCataldo/go-mongo-template
Then import the go-mongo package into your own code.
import "github.com/GabrielHCataldo/go-mongo-template/mongo"Below we will show some basic examples:
IMPORTANT: Always check the documentation in the structures and functions fields. For more details on the examples, visit All examples link.
To insert a document, it's very simple, first let's configure a structure To be able to work with go-mongo, see:
package main
import "go.mongodb.org/mongo-driver/bson/primitive"
type test struct {
Id primitive.ObjectID `json:"id,omitempty" bson:"_id,omitempty" database:"test" collection:"test"`
Random int `json:"random,omitempty" bson:"random,omitempty"`
Name string `json:"name,omitempty" bson:"name,omitempty"`
BirthDate primitive.DateTime `json:"birthDate,omitempty" bson:"birthDate,omitempty"`
Emails []string `json:"emails,omitempty" bson:"emails,omitempty"`
Balance float64 `json:"balance,omitempty" bson:"balance,omitempty"`
CreatedAt primitive.DateTime `json:"createdAt,omitempty" bson:"createdAt,omitempty"`
}As shown in the structure above, it is important to continue, the database tags with the name from your database and collection with the name of your collection. With this example structure We were able to continue with the following interactions:
Insert a document into the collection see:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"math/rand"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
testDocument := test{
Random: rand.Int(),
Name: "Foo Bar",
BirthDate: primitive.NewDateTimeFromTime(time.Date(1999, 1, 21, 0, 0, 0, 0, time.Local)),
Emails: []string{"[email protected]", "[email protected]"},
Balance: 190.12,
CreatedAt: primitive.NewDateTimeFromTime(time.Now()),
}
//new document need a pointer
err = mongoTemplate.InsertOne(ctx, &testDocument)
if helper.IsNotNil(err) {
logger.Error("error insert document:", err)
} else {
logger.Info("document inserted successfully:", testDocument)
}
}Output:
[INFO 2023/12/22 13:02:07] main.go:35: document inserted successfully: {"id":"65b8a8b753748d924631520c","random":6628457526937947134,"name":"Foo Bar","birthDate":"1999-01-21T02:00:00Z","emails":["[email protected]","[email protected]"],"balance":190.12,"createdAt":"2023-12-22T16:02:07.322Z"}
We are able to insert multiple documents, remembering that if no ID is provided, and if you have configured a field with annotation bson _id and the type is primitive.ObjectId we will fill in the documents with identifiers generated by MongoDB, see:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"github.com/GabrielHCataldo/go-mongo-template/mongo/option"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"math/rand"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
testDocuments := []*test{
{
Random: rand.Int(),
Name: "Foo Bar",
BirthDate: primitive.NewDateTimeFromTime(time.Date(1999, 1, 21, 0, 0, 0, 0, time.Local)),
Emails: []string{"[email protected]", "[email protected]"},
Balance: 190.12,
CreatedAt: primitive.NewDateTimeFromTime(time.Now()),
},
{
Random: rand.Int(),
Name: "Foo Bar 2",
BirthDate: primitive.NewDateTimeFromTime(time.Date(1999, 1, 21, 0, 0, 0, 0, time.Local)),
Emails: []string{"[email protected]", "[email protected]"},
Balance: 290.12,
CreatedAt: primitive.NewDateTimeFromTime(time.Now()),
},
}
//new document need a slice of the pointer
err = mongoTemplate.InsertMany(ctx, testDocuments)
if helper.IsNotNil(err) {
logger.Error("error insert document:", err)
} else {
logger.Info("document inserted successfully:", testDocuments)
}
}Output:
[INFO 2023/12/22 16:11:33] main.go:46: document inserted successfully: [{"balance":190.12,"birthDate":"1999-01-21T02:00:00Z","createdAt":"2023-12-22T19:11:33.206Z","emails":["[email protected]","[email protected]"],"id":"65b8a8b753748d924631520b","name":"Foo Bar","random":8094092400336225232},{"balance":290.12,"birthDate":"1999-01-21T02:00:00Z","createdAt":"2023-12-22T19:11:33.206Z","emails":["[email protected]","[email protected]"],"id":"65b8a8b753748d924631520f","name":"Foo Bar 2","random":4318945546650338065}]
For more insert examples visit link.
We will perform two simple update operations, see below using UpdateOne:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
update := bson.M{"$set": bson.M{"name": "Foo Bar Updated"}}
updateResult, err := mongoTemplate.UpdateOne(ctx, filter, update, test{})
if helper.IsNotNil(err) {
logger.Error("error update document:", err)
} else {
logger.Info("document updated successfully:", updateResult)
}
}Output:
[INFO 2023/12/22 16:29:23] main.go:29: document updated successfully: {"MatchedCount":1,"ModifiedCount":1,"UpsertedCount":0,"UpsertedID":null}
If you wanted to filter only with the document id, use the UpdateOneById function:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
objectId, _ := primitive.ObjectIDFromHex("6585f3a8bf2af8ad9bcab912")
update := bson.M{"$set": bson.M{"name": "Foo Bar Updated"}}
updateResult, err := mongoTemplate.UpdateOneById(ctx, objectId, update, test{})
if helper.IsNotNil(err) {
logger.Error("error update document:", err)
} else {
logger.Info("document updated successfully:", updateResult)
}
}Output:
[INFO 2023/12/22 17:47:00] main.go:29: document updated successfully: {"MatchedCount":1,"ModifiedCount":1,"UpsertedCount":0,"UpsertedID":null}
We can also use the UpdateMany function to update multiple documents, see in the example below:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
update := bson.M{"$set": bson.M{"name": "Foo Bar Updated"}}
updateResult, err := mongoTemplate.UpdateOne(ctx, filter, update, test{})
if helper.IsNotNil(err) {
logger.Error("error update documents:", err)
} else {
logger.Info("document updated successfully:", updateResult)
}
}Output:
[INFO 2023/12/22 16:31:41] main.go:29: document updated successfully: {"MatchedCount":13,"ModifiedCount":13,"UpsertedCount":0,"UpsertedID":null}
For more update examples visit link.
To replace a document, see the example below:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"math/rand"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
replacement := test{
Random: rand.Int(),
Name: "Foo Bar",
BirthDate: primitive.NewDateTimeFromTime(time.Date(1999, 1, 21, 0, 0, 0, 0, time.Local)),
Emails: []string{"[email protected]", "[email protected]"},
Balance: 190.12,
CreatedAt: primitive.NewDateTimeFromTime(time.Now()),
}
updateResult, err := mongoTemplate.ReplaceOne(ctx, filter, replacement, test{})
if helper.IsNotNil(err) {
logger.Error("error replace document:", err)
} else {
logger.Info("document replaced successfully:", updateResult)
}
}Output:
[INFO 2023/12/22 17:21:49] main.go:37: document replaced successfully: {"MatchedCount":1,"ModifiedCount":1,"UpsertedCount":0,"UpsertedID":null}
For more replace examples visit link.
Let's start by removing just one document using the simple filter, see below:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
deleteResult, err := mongoTemplate.DeleteOne(ctx, filter, test{})
if helper.IsNotNil(err) {
logger.Error("error delete document:", err)
} else {
logger.Info("document deleted successfully:", deleteResult)
}
}Output:
[INFO 2023/12/22 17:38:06] main.go:53: document deleted successfully: {"DeletedCount":1}
We can remove a document by passing it as a parameter as in the example below:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
objectId, _ := primitive.ObjectIDFromHex("6585f3a8bf2af8ad9bcab912")
deleteResult, err := mongoTemplate.DeleteOneById(ctx, objectId, test{})
if helper.IsNotNil(err) {
logger.Error("error delete document:", err)
} else {
logger.Info("document deleted successfully:", deleteResult)
}
}Output:
[INFO 2023/12/22 17:41:50] main.go:28: document deleted successfully: {"DeletedCount":1}
And finally, we can remove several documents using the indicated filter:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
deleteResult, err := mongoTemplate.DeleteMany(ctx, filter, test{})
if helper.IsNotNil(err) {
logger.Error("error delete documents:", err)
} else {
logger.Info("documents deleted successfully:", deleteResult)
}
}Output
[INFO 2023/12/22 17:50:01] main.go:28: documents deleted successfully: {"DeletedCount":8}
For more delete examples visit link.
Obtaining all the documents is very simple, see the example below:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson/primitive"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
//dest need a pointer
var dest []test
err = mongoTemplate.FindAll(ctx, &dest)
if helper.IsNotNil(err) {
logger.Error("error find all documents:", err)
} else {
logger.Info("find all documents successfully:", dest)
}
}Output:
[INFO 2023/12/22 17:02:22] main.go:28: find all documents successfully: [{"balance":190.12,"birthDate":"1999-01-21T02:00:00Z","createdAt":"2023-12-22T18:53:26.695Z","emails":["[email protected]","[email protected]"],"id":"65b8a8b753748d924631520g","name":"Foo Bar Updated","random":2576121145493409319},{"balance":190.12,"birthDate":"1999-01-21T02:00:00Z","createdAt":"2023-12-22T19:11:33.206Z","emails":["[email protected]","[email protected]"],"id":"65b8a8b753748d924631520j","name":"Foo Bar","random":8094092400336225232},{"balance":290.12,"birthDate":"1999-01-21T02:00:00Z","createdAt":"2023-12-22T19:11:33.206Z","emails":["[email protected]","[email protected]"],"id":"65b8a8b753748d924631520k","name":"Foo Bar 2","random":4318945546650338065}]
Be able to page in a simple and intuitive way, see:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
pageOutput, err := mongoTemplate.FindPageable(ctx, filter, mongo.PageInput{
Page: 0,
PageSize: 10,
Ref: []test{}, //need a slice of the structure
Sort: bson.M{"createdAt": mongo.SortDesc},
})
if helper.IsNotNil(err) {
logger.Error("error find pageable documents:", err)
} else {
var dest []test
_ = pageOutput.Content.Decode(&dest)
logger.Info("find pageable documents successfully:", pageOutput)
logger.Info("find pageable get first value:", dest[0])
}
}Output:
[INFO 2024/01/30 04:57:45] main.go:33: find pageable documents successfully: {"page":null,"pageSize":"10","pageTotal":"1","totalElements":"2","content":[{"balance":"190.12","birthDate":"1999-01-21T02:00:00Z","createdAt":"2024-01-30T07:57:35.865Z","emails":["[email protected]","[email protected]"],"id":"65b8abef44f575c583b91a81","name":"Foo Bar","random":"5208816839613347000"},{"balance":"290.12","birthDate":"1999-01-21T02:00:00Z","createdAt":"2024-01-30T07:57:35.865Z","emails":["[email protected]","[email protected]"],"id":"65b8abef44f575c583b91a82","name":"Foo Bar 2","random":"1873557180388720600"}],"lastQueryAt":"2024-01-30T07:57:45.444808Z"}
[INFO 2024/01/30 04:57:45] main.go:34: find pageable get first value: {"id":"65b8abef44f575c583b91a81","random":"5208816839613347000","name":"Foo Bar","birthDate":"1999-01-21T02:00:00Z","emails":["[email protected]","[email protected]"],"balance":"190.12","createdAt":"2024-01-30T07:57:35.865Z"}
Also obtain a single document by id, see below:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
objectId, _ := primitive.ObjectIDFromHex("6585db26633e225cbeadf553")
//dest need a pointer
var dest test
err = mongoTemplate.FindOneById(ctx, objectId, &dest)
if helper.IsNotNil(err) {
logger.Error("error find all documents:", err)
} else {
logger.Info("find by id document successfully:", dest)
}
}Output:
[INFO 2023/12/22 17:14:47] main.go:29: find by id document successfully: {"id":"65b8a8b753748d924631521z","random":2576121145493409319,"name":"Foo Bar Updated","birthDate":"1999-01-21T02:00:00Z","emails":["[email protected]","[email protected]"],"balance":190.12,"createdAt":"2023-12-22T18:53:26.695Z"}
For more find examples visit link.
Check if the document exists by id:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
objectId, _ := primitive.ObjectIDFromHex("6585f5b8fd8fa97d562419f7")
exists, err := mongoTemplate.ExistsById(ctx, objectId, test{})
if helper.IsNotNil(err) {
logger.Error("error check exists document:", err)
} else {
logger.Info("document exists:", exists)
}
}Output:
[INFO 2023/12/22 17:56:24] main.go:27: document exists: true
Check if the document exists using the filter:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
exists, err := mongoTemplate.Exists(ctx, filter, test{})
if helper.IsNotNil(err) {
logger.Error("error check exists document:", err)
} else {
logger.Info("document exists:", exists)
}
}Output:
[INFO 2023/12/22 17:57:46] main.go:28: document exists: true
For more examples of exists visit link.
We start with the total counter, see:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
countResult, err := mongoTemplate.EstimatedDocumentCount(ctx, test{})
if helper.IsNotNil(err) {
logger.Error("error estimated document count:", err)
} else {
logger.Info("estimated document count successfully:", countResult)
}
}Output:
[INFO 2023/12/22 18:26:44] main.go:27: estimated document count successfully: 16
You can also count the documents using the filter:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
filter := bson.M{"_id": bson.M{"$exists": true}}
countResult, err := mongoTemplate.CountDocuments(ctx, filter, test{})
if helper.IsNotNil(err) {
logger.Error("error count documents:", err)
} else {
logger.Info("count documents successfully:", countResult)
}
}Output:
[INFO 2023/12/22 18:23:47] main.go:28: count documents successfully: 10
To watch the operations we highlight the WatchWithHandler, where it will persist the searches and will convert the received data, transforming it into a EventContext where it will have all the necessary data, then After the conversion, we call the function informed in the function parameter, see a basic example:
package main
import (
"context"
"github.com/GabrielHCataldo/go-helper/helper"
"github.com/GabrielHCataldo/go-logger/logger"
"github.com/GabrielHCataldo/go-mongo-template/mongo"
"github.com/GabrielHCataldo/go-mongo-template/mongo/option"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"os"
"time"
)
func main() {
ctx, cancel := context.WithTimeout(context.TODO(), 5*time.Second)
defer cancel()
mongoTemplate, err := mongo.NewTemplate(ctx, options.Client().ApplyURI(os.Getenv("MONGODB_URL")))
if helper.IsNotNil(err) {
logger.Error("error to init mongo template:", err)
return
}
defer mongoTemplate.SimpleDisconnect(ctx)
pipeline := mongo.Pipeline{bson.D{{"$match", bson.D{
{"operationType", bson.M{"$in": []string{"insert", "update", "delete", "replace"}}},
}}}}
opt := option.NewWatchWithHandler().SetDatabaseName("test")
err = mongoTemplate.WatchWithHandler(context.TODO(), pipeline, handler, opt)
if helper.IsNotNil(err) {
logger.Error("error watch handler:", err)
} else {
logger.Info("watch handler complete successfully")
}
}
func handler(ctx *mongo.EventContext) {
logger.Info("handler called:", ctx)
}Output:
[INFO 2023/12/22 18:39:42] main.go:36: handler called: {"Event":{"ClusterTime":{"I":5,"T":1703281182},"DocumentKey":{"ID":[101,134,2,30,136,59,150,210,55,170,55,135]},"FullDocument":{"_id":[101,134,2,30,136,59,150,210,55,170,55,135],"balance":190.12,"birthDate":"1999-01-21T02:00:00Z","createdAt":"2023-12-22T21:39:42.288Z","emails":["[email protected]","[email protected]"],"name":"Foo Bar","random":2217047455415420974},"NS":{"Coll":"test","DB":"test"},"OperationType":"insert","UpdateDescription":{"RemovedFields":null,"TruncatedArrays":null,"UpdatedFields":null}}}
[INFO 2023/12/22 18:42:51] main.go:36: handler called: {"Event":{"ClusterTime":{"I":9,"T":1703281182},"DocumentKey":{"ID":[101,134,2,30,136,59,150,210,55,170,55,137]},"FullDocument":{"_id":[101,134,2,30,136,59,150,210,55,170,55,137],"balance":190.12,"birthDate":"1999-01-21T02:00:00Z","createdAt":"2023-12-22T21:39:42.627Z","emails":["[email protected]","[email protected]"],"name":"Foo Bar","random":2251712412936242638},"NS":{"Coll":"test","DB":"test"},"OperationType":"insert","UpdateDescription":{"RemovedFields":null,"TruncatedArrays":null,"UpdatedFields":null}}}
Process finished with the exit code 130 (interrupted by signal 2:SIGINT)
For more watch examples visit link.
Below are some examples that were not shown or cited:
Make a pull request, or if you find a bug, open it an Issues.
Distributed under MIT license, see the license file within the code for more details.