Some mongodb experiments

Signed-off-by: Rohit Nayak <[email protected]>

Author: rohit-nayak-ps

vr-mongo/connect.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"time"
+)
+
+func main() {
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	client, err := mongo.Connect(ctx, options.Client().ApplyURI(
+		"mongodb://test:[email protected]/test?retryWrites=true&w=majority",
+	))
+	if err != nil {
+		fmt.Println(err.Error())
+		return
+	}
+	//fmt.Printf("client %v:%s\n", client, err)
+
+	collection := client.Database("test").Collection("inventory")
+	count, err := collection.CountDocuments(ctx, bson.D{{}}, nil)
+	if err != nil {
+		fmt.Println(err.Error())
+		return
+	}
+	fmt.Printf("#docs %d\n", count)
+}

vr-mongo/logtail.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"fmt"
+	"github.com/gnokoheat/oplog"
+)
+
+func main() {
+	var o = &oplog.Options{
+		// "mongodb+srv://rohit:[email protected]/sample_airbnb?retryWrites=true&w=majority",
+		// (e.g. mongodb://username:[email protected]:27017,127.0.0.1:27018/local?replicaSet=rs01&authSource=admin)
+		Addrs:      []string{"127.0.0.1"}, // replicaset host and port
+		Username:   "rohit", // admin db username
+		Password:   "abc123", // admin db user password
+		ReplicaSet: "rs", // replicaset name
+		DB:         "test", // tailing target db
+		Collection: "inventory", // tailing target collection
+		Events:     []string{"insert", "update", "delete"}, // tailing target method
+	}
+
+	l := make(chan *[]oplog.Log) // Oplog Channel
+	e := make(chan error) // Error Channel
+	fmt.Printf("o is %v\n", o)
+
+	// Oplog tailing start !
+	go o.Tail(l, e)
+
+	for {
+		select {
+		case err := <-e:
+			fmt.Printf("[Error] %s\n", err)
+			return
+		case op := <-l:
+			// input oplog handling code
+			fmt.Printf("[Result] %v\n", op)
+			break
+		}
+	}
+}

vr-mongo/x.go

@@ -0,0 +1,50 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"github.com/rwynn/gtm/v2"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/bsontype"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"reflect"
+	"time"
+)
+
+func main() {
+	rb := bson.NewRegistryBuilder()
+	//rb.RegisterTypeMapEntry(bsontype.Timestamp, reflect.TypeOf(time.Time{}))
+	rb.RegisterTypeMapEntry(bsontype.DateTime, reflect.TypeOf(time.Time{}))
+	reg := rb.Build()
+	clientOptions := options.Client()
+	clientOptions.SetRegistry(reg)
+	clientOptions.ApplyURI("mongodb://localhost:27017")
+	client, err := mongo.NewClient(clientOptions)
+	if err != nil {
+		panic(err)
+	}
+	ctxm, cancel := context.WithTimeout(context.Background(), 20*time.Second)
+	defer cancel()
+	err = client.Connect(ctxm)
+	if err != nil {
+		panic(err)
+	}
+	defer client.Disconnect(context.Background())
+	ctx := gtm.Start(client, &gtm.Options{
+		DirectReadNs: []string{"test.test"},
+		ChangeStreamNs: []string{"test.test"},
+		//MaxWaitSecs: 10,
+		OpLogDisabled: true,
+	})
+	for {
+		select {
+		case err := <-ctx.ErrC:
+			fmt.Printf("got err %+v", err)
+			break
+		case op := <-ctx.OpC:
+			fmt.Printf("got op %+v", op)
+			break
+		}
+	}
+}