KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use GetMongoDataBase method of storage Package

Best Testkube code snippet using storage.GetMongoDataBase

repository.go

Source:repository.go

1package repository2import (3	"context"4	"errors"5	"github.com/nkonev/blog-storage/logger"6	. "github.com/nkonev/blog-storage/logger"7	"github.com/nkonev/blog-storage/utils"8	"go.mongodb.org/mongo-driver/bson"9	"go.mongodb.org/mongo-driver/bson/primitive"10	"go.mongodb.org/mongo-driver/mongo"11	"go.mongodb.org/mongo-driver/mongo/options"12)13const Id = "_id"14const filename = "filename"15const published = "published"16const userId = "userid"17const collectionLimits = "limits"18const CollectionUserFiles = "userFiles"19// https://vkt.sh/go-mongodb-driver-cookbook/20type UserFileDto struct {21	Id        primitive.ObjectID `bson:"_id,omitempty"` // mongo document id equal to minio object jd22	Filename  string23	Published bool24	UserId    int6425}26type UserFileRepository struct {27	mongo *mongo.Client28}29func NewUserFileRepository(mongo *mongo.Client) *UserFileRepository {30	return &UserFileRepository{mongo: mongo}31}32type LimitsRepository struct {33	mongo *mongo.Client34}35func NewLimitsRepository(mongo *mongo.Client) *LimitsRepository {36	return &LimitsRepository{mongo: mongo}37}38func ToFileMongoDto(c *mongo.Cursor) (*UserFileDto, error) {39	var elem UserFileDto40	err := c.Decode(&elem)41	if err != nil {42		return nil, err43	}44	return &elem, nil45}46func GetIdDoc(objectId string) (*bson.D, error) {47	ids, e := primitive.ObjectIDFromHex(objectId)48	if e != nil {49		return nil, e50	}51	ds := bson.D{{Id, ids}}52	return &ds, nil53}54func GetUpdateDoc(p bson.M) bson.M {55	update := bson.M{"$set": p}56	return update57}58func (r *UserFileRepository) GetUserIdByGlobalId(objectId string) (int, error) {59	ids, e := primitive.ObjectIDFromHex(objectId)60	if e != nil {61		return 0, e62	}63	database := utils.GetMongoDatabase(r.mongo)64	ms := bson.M{Id: ids}65	one := database.Collection(CollectionUserFiles).FindOne(context.TODO(), ms)66	if one.Err() != nil {67		if one.Err() != mongo.ErrNoDocuments {68			logger.Logger.Errorf("Error during get user id by global id %v", objectId)69		} else {70			logger.Logger.Infof("No documents found by global id %v", objectId)71		}72		return 0, one.Err()73	}74	var elem UserFileDto75	if err := one.Decode(&elem); err != nil {76		return 0, err77	}78	return int(elem.UserId), nil79}80func (r *UserFileRepository) InsertMetaInfoToMongo(filename string, userId int) (*string, error) {81	database := utils.GetMongoDatabase(r.mongo)82	inserted, err := database.Collection(CollectionUserFiles).InsertOne(context.TODO(), UserFileDto{Filename: filename, Published: false, UserId: int64(userId)})83	if err != nil {84		Logger.Errorf("Error during create mongo metadata document: %v", err)85		return nil, err86	}87	idMongo := inserted.InsertedID.(primitive.ObjectID).Hex()88	return &idMongo, nil89}90func (r *UserFileRepository) GetMetainfoFromMongo(objectId string) (*UserFileDto, error) {91	database := utils.GetMongoDatabase(r.mongo)92	var userFilesCollection *mongo.Collection = database.Collection(CollectionUserFiles)93	ds, err := GetIdDoc(objectId)94	if err != nil {95		logger.Logger.Errorf("Error during creating id document %v", objectId)96		return nil, err97	}98	one := userFilesCollection.FindOne(context.TODO(), ds)99	if one == nil {100		return nil, errors.New("Unexpected nil by id " + objectId)101	}102	if one.Err() != nil {103		logger.Logger.Errorf("Error during querying record from mongo by key %v", objectId)104		return nil, one.Err()105	}106	var elem UserFileDto107	if err := one.Decode(&elem); err != nil {108		if err == mongo.ErrNoDocuments {109			logger.Logger.Errorf("No documents found by key %v", objectId)110		}111		return nil, err112	}113	return &elem, nil114}115func (r *UserFileRepository) RenameUserFile(objId string, newname string) error {116	database := utils.GetMongoDatabase(r.mongo)117	var userFilesCollection *mongo.Collection = database.Collection(CollectionUserFiles)118	findDocument, err := GetIdDoc(objId)119	if err != nil {120		return err121	}122	updateDocument := GetUpdateDoc(primitive.M{filename: newname})123	one := userFilesCollection.FindOneAndUpdate(context.TODO(), findDocument, updateDocument)124	if one == nil {125		return errors.New("Unexpected nil result during update")126	}127	if one.Err() != nil {128		return one.Err()129	}130	return nil131}132func (r *UserFileRepository) UpdatePublished(objId string, setValPublished bool) (*UserFileDto, error) {133	database := utils.GetMongoDatabase(r.mongo)134	var collection *mongo.Collection = database.Collection(CollectionUserFiles)135	findDocument, err := GetIdDoc(objId)136	if err != nil {137		return nil, err138	}139	updateDocument := GetUpdateDoc(primitive.M{published: setValPublished})140	one := collection.FindOneAndUpdate(context.TODO(), findDocument, updateDocument)141	if one == nil {142		return nil, errors.New("Unexpected nil result during update")143	}144	if one.Err() != nil {145		return nil, one.Err()146	}147	var elem UserFileDto148	if err := one.Decode(&elem); err != nil {149		return nil, err150	}151	return &elem, nil152}153func (r *UserFileRepository) FindUserFiles(userIdInt int) (*mongo.Cursor, error) {154	database := utils.GetMongoDatabase(r.mongo)155	var collection *mongo.Collection = database.Collection(CollectionUserFiles)156	return collection.Find(context.TODO(), bson.D{{userId, userIdInt}})157}158func (r *UserFileRepository) Delete(objId string) error {159	database := utils.GetMongoDatabase(r.mongo)160	var collection *mongo.Collection = database.Collection(CollectionUserFiles)161	d, e := GetIdDoc(objId)162	if e != nil {163		return e164	}165	_, e = collection.DeleteOne(context.TODO(), d)166	return e167}168func IsDocumentExists(mongoC *mongo.Client, collection string, request interface{}, opts ...*options.FindOneOptions) (bool, error) {169	database := utils.GetMongoDatabase(mongoC)170	// https://siongui.github.io/2017/03/13/go-pass-slice-or-array-as-variadic-parameter/#id12171	res := database.Collection(collection).FindOne(context.TODO(), request, opts[:]...)172	if res.Err() != nil {173		if res.Err() == mongo.ErrNoDocuments {174			return false, nil175		}176		Logger.Errorf("Error during find '%v' : %v", request, res.Err())177		return false, res.Err()178	}179	_, e := res.DecodeBytes()180	if e != nil {181		Logger.Errorf("Error during DecodeBytes '%v' : %v", request, res.Err())182		return false, e183	} else {184		return true, nil185	}186}187func (r *LimitsRepository) IsStorageUnlimitedForUser(userId int) (bool, error) {188	return IsDocumentExists(r.mongo, collectionLimits, bson.D{{Id, userId}})189}190func (r *LimitsRepository) Patch(userId int, limited bool) error {191	database := utils.GetMongoDatabase(r.mongo)192	if limited {193		_, e := database.Collection(collectionLimits).DeleteOne(context.TODO(), bson.D{{Id, userId}})194		if e != nil {195			return e196		}197	} else {198		// unlimited199		_, e := database.Collection(collectionLimits).InsertOne(context.TODO(), bson.D{{Id, userId}})200		if e != nil {201			return e202		}203	}204	return nil205}...

mongo_lock.go

Source:mongo_lock.go

1package mongo_lock2import (3	"context"4	. "github.com/nkonev/blog-storage/logger"5	"github.com/nkonev/blog-storage/utils"6	"go.mongodb.org/mongo-driver/bson"7	"go.mongodb.org/mongo-driver/mongo"8	"go.mongodb.org/mongo-driver/mongo/options"9	"time"10)11type mongoLock struct {12	mongoClient    *mongo.Client13	lockCollection string14	idDoc          bson.D15}16func GetIdDoc() bson.D {17	return bson.D{{"_id", 42}}18}19func NewMongoLock(mongoClient *mongo.Client, lockCollection string) *mongoLock {20	return &mongoLock{mongoClient: mongoClient, lockCollection: lockCollection, idDoc: GetIdDoc()}21}22func createBson(data string) bson.D {23	bdoc := bson.D{}24	err := bson.UnmarshalExtJSON([]byte(data), true, &bdoc)25	if err != nil {26		Logger.Panicf("Error during creating bson from \"%v\": %v", data, err)27	}28	return bdoc29}30func ensureIndex(client *mongo.Client, lockCollection string) {31	database := utils.GetMongoDatabase(client)32	commandResult := database.RunCommand(context.TODO(), createBson(`{33        "createIndexes": "`+lockCollection+`",34        "indexes": [35            {36                "key": {37                    "id": 138                },39                "name": "unique_id",40	        	"unique": true41            }42        ]43}`))44	if commandResult.Err() != nil {45		Logger.Panicf("Error during creating unique index: %v", commandResult.Err())46	}47}48func getUpdateDoc(p bson.M) bson.M {49	update := bson.M{"$set": p}50	return update51}52func (ml *mongoLock) AcquireLock() {53	ensureIndex(ml.mongoClient, ml.lockCollection)54	database := utils.GetMongoDatabase(ml.mongoClient)55	var upsert = true56	duration, _ := time.ParseDuration("1s")57	for {58		result, err := database.Collection(ml.lockCollection).UpdateOne(context.TODO(), ml.idDoc, getUpdateDoc(bson.M{"lastAcquired": time.Now()}), &options.UpdateOptions{Upsert: &upsert})59		if err != nil {60			Logger.Panicf("Error during acquiring lock: %v", err)61		} else {62			if result.UpsertedID != nil {63				Logger.Infof("Lock has been acquired")64				break65			} else {66				Logger.Infof("Lock has n' t been acquired - waiting %v", duration)67				time.Sleep(duration)68				continue69			}70		}71	}72}73func (ml *mongoLock) ReleaseLock() {74	database := utils.GetMongoDatabase(ml.mongoClient)75	_, err := database.Collection(ml.lockCollection).DeleteOne(context.TODO(), ml.idDoc)76	if err != nil {77		Logger.Panicf("Error during releasing lock: %v", err)78	}79	Logger.Infof("Lock successfully released")80}...

db.go

Source:db.go

1package db2import (3	"github.com/flavioribeiro/gonfig"4	"github.com/snickers/snickers/types"5)6//go:generate counterfeiter . Storage7// Storage defines functions for accessing data8type Storage interface {9	// Preset methods10	StorePreset(types.Preset) (types.Preset, error)11	RetrievePreset(string) (types.Preset, error)12	UpdatePreset(string, types.Preset) (types.Preset, error)13	GetPresets() ([]types.Preset, error)14	DeletePreset(string) (types.Preset, error)15	// Job methods16	DeleteJob(string) (types.Job, error)17	StoreJob(types.Job) (types.Job, error)18	RetrieveJob(string) (types.Job, error)19	UpdateJob(string, types.Job) (types.Job, error)20	GetJobs() ([]types.Job, error)21	ClearDatabase() error22}23// GetDatabase selects the right driver based on config24func GetDatabase(config gonfig.Gonfig) (Storage, error) {25	driver, err := config.GetString("DATABASE_DRIVER", "memory")26	if err != nil {27		return nil, err28	}29	if driver == "mongo" || driver == "mongodb" {30		return getMongoDatabase(config)31	}32	return getMemoryDatabase()33}...

GetMongoDataBase

Using AI Code Generation

1import (2func main() {3	s := storage.Storage{}4	db := s.GetMongoDataBase()5	fmt.Println("db: ", db)6}7db:  &{0xc0000a61e0 0xc0000a61e0 0xc0000a61e0 0xc0000a61e0}8How to use import in Go9How to use import in Go

GetMongoDataBase

Using AI Code Generation

1import (2func main() {3	storageObject := storage.Storage{}4	mongoDataBase := storageObject.GetMongoDataBase()5	fmt.Println("mongoDataBase: ", mongoDataBase)6}7Go: How to import packages in Go - September 3, 2020

GetMongoDataBase

Using AI Code Generation

1import (2func main() {3	storage := storage.NewStorage()4	db := storage.GetMongoDataBase()5	fmt.Println(db)6}7import (8func main() {9	storage := storage.NewStorage()10	db := storage.GetMongoDataBase()11	fmt.Println(db)12}13import (14func main() {15	storage := storage.NewStorage()16	db := storage.GetMongoDataBase()17	fmt.Println(db)18}19import (20func main() {21	storage := storage.NewStorage()22	db := storage.GetMongoDataBase()23	fmt.Println(db)24}25import (26func main() {27	storage := storage.NewStorage()28	db := storage.GetMongoDataBase()29	fmt.Println(db)30}31import (32func main() {33	storage := storage.NewStorage()34	db := storage.GetMongoDataBase()35	fmt.Println(db)36}37import (38func main() {39	storage := storage.NewStorage()40	db := storage.GetMongoDataBase()41	fmt.Println(db)42}

GetMongoDataBase

Using AI Code Generation

1import (2func main() {3    db := storage.GetMongoDataBase()4    fmt.Println(db)5}6import (7func GetMongoDataBase() *mgo.Database {8    session, err := mgo.Dial("localhost")9    if err != nil {10        panic(err)11    }12    defer session.Close()13    session.SetSafe(&mgo.Safe{})14    c := session.DB("test").C("people")15    err = c.Insert(&Person{"Ale", "+55 53 8116 9639"},16        &Person{"Cla", "+55 53 8402 8510"})17    if err != nil {18        panic(err)19    }20    result := Person{}21    err = c.Find(bson.M{"name": "Ale"}).One(&result)22    if err != nil {23        panic(err)24    }25    fmt.Println("Phone:", result.Phone)26    return session.DB("test")27}28import "gopkg.in/mgo.v2/bson"29type Person struct {30}31func GetPerson() Person {32    return Person{"Ale", "+55 53 8116 9639"}33}34import (35func main() {

GetMongoDataBase

Using AI Code Generation

1import (2func main() {3    db := storage.GetMongoDataBase()4    collection := db.C("users")5    fmt.Println("Collection name:", collection.Name)6}7import (8func main() {9    db := storage.GetMongoDataBase()10    collection := db.C("users")11    fmt.Println("Collection name:", collection.Name)12}13import (14func main() {15    db := storage.GetMongoDataBase()16    collection := db.C("users")17    fmt.Println("Collection name:", collection.Name)18}19import (20func main() {21    db := storage.GetMongoDataBase()22    collection := db.C("users")23    fmt.Println("Collection name:", collection.Name)24}25import (26func main() {27    db := storage.GetMongoDataBase()28    collection := db.C("users")29    fmt.Println("Collection name:", collection.Name)30}31import (32func main() {

GetMongoDataBase

Using AI Code Generation

1import (2func main() {3	db := mongo.GetMongoDataBase()4	collection := db.C("users")5	cursor := collection.Find(nil)6	err := cursor.All(&result)7	if err != nil {8		log.Fatal(err)9	}10	cursor.Close()11	fmt.Println(result)12}13import (14func main() {15	db := mongo.GetMongoDataBase()16	collection := db.C("users")17	cursor := collection.Find(nil)18	err := cursor.All(&result)19	if err != nil {20		log.Fatal(err)21	}22	cursor.Close()23	fmt.Println(result)24}25import (26func main() {

GetMongoDataBase

Using AI Code Generation

1import (2type Person struct {3}4func main() {5	mongoDataBase := storage.GetMongoDataBase()6	collection := mongoDataBase.GetCollection("person")7	result := collection.Find(bson.M{})8	var documents []interface{}9	err := result.All(&documents)10	if err != nil {11		panic(err)12	}13	for _, document := range documents {14		person := document.(Person)15		fmt.Println(person.Name, person.Age)16	}17}18import (19type Person struct {20}21func main() {22	mongoDataBase := storage.GetMongoDataBase()

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.