|
|
@@ -0,0 +1,162 @@
|
|
|
+package storage
|
|
|
+
|
|
|
+import (
|
|
|
+ "atreugo/src/templates"
|
|
|
+ "context"
|
|
|
+ "fmt"
|
|
|
+ "log"
|
|
|
+ "math/rand"
|
|
|
+ "time"
|
|
|
+
|
|
|
+ "go.mongodb.org/mongo-driver/mongo"
|
|
|
+ "go.mongodb.org/mongo-driver/mongo/options"
|
|
|
+ "go.mongodb.org/mongo-driver/mongo/readpref"
|
|
|
+ "gopkg.in/mgo.v2/bson"
|
|
|
+)
|
|
|
+
|
|
|
+// Mongo struct
|
|
|
+type Mongo struct {
|
|
|
+ db *mongo.Client
|
|
|
+ database *mongo.Database
|
|
|
+ // mongodb collections
|
|
|
+ worlds *mongo.Collection
|
|
|
+ fortunes *mongo.Collection
|
|
|
+}
|
|
|
+
|
|
|
+// Connect create connection and ping db
|
|
|
+func (m *Mongo) Connect(dbConnectionString string, maxConnectionsInPool int) error {
|
|
|
+ var err error
|
|
|
+
|
|
|
+ opts := options.Client()
|
|
|
+ // opts.SetMaxPoolSize(uint16(maxConnectionsInPool))
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
|
|
+ defer cancel()
|
|
|
+ m.db, err = mongo.Connect(ctx, opts.ApplyURI(dbConnectionString))
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ ctx, cancel = context.WithTimeout(context.Background(), 2*time.Second)
|
|
|
+ defer cancel()
|
|
|
+ err = m.db.Ping(ctx, readpref.Primary())
|
|
|
+ if err != nil {
|
|
|
+ return err
|
|
|
+ }
|
|
|
+
|
|
|
+ m.database = m.db.Database("hello_world")
|
|
|
+ m.worlds = m.database.Collection("world")
|
|
|
+ m.fortunes = m.database.Collection("fortune")
|
|
|
+
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+// Close connect to db
|
|
|
+func (m *Mongo) Close() {
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
|
|
+ defer cancel()
|
|
|
+ m.db.Disconnect(ctx)
|
|
|
+}
|
|
|
+
|
|
|
+// GetOneRandomWorld return one random World struct
|
|
|
+func (m Mongo) GetOneRandomWorld(w *World) error {
|
|
|
+ var err error
|
|
|
+ queryID := rand.Intn(worldsCount) + 1
|
|
|
+
|
|
|
+ filter := bson.M{"_id": queryID}
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
|
+ defer cancel()
|
|
|
+
|
|
|
+ if err = m.worlds.FindOne(ctx, filter).Decode(w); err != nil {
|
|
|
+ err = fmt.Errorf("error scanning world row with ID %d: %s", queryID, err)
|
|
|
+ }
|
|
|
+
|
|
|
+ return err
|
|
|
+}
|
|
|
+
|
|
|
+// UpdateWorlds updates some number of worlds entries, passed as arg
|
|
|
+func (m Mongo) UpdateWorlds(selectedWorlds []World) error {
|
|
|
+ for _, selectedWorld := range selectedWorlds {
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
|
+ defer cancel()
|
|
|
+
|
|
|
+ selectedWorld.RandomNumber = rand.Intn(worldsCount) + 1
|
|
|
+ if _, err := m.worlds.UpdateOne(
|
|
|
+ ctx,
|
|
|
+ bson.M{"_id": selectedWorld.ID},
|
|
|
+ bson.M{"$set": bson.M{"randomNumber": selectedWorld.RandomNumber}},
|
|
|
+ ); err != nil {
|
|
|
+ log.Fatalf("Error updating world with id: %s", err.Error())
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+// GetFortunes selects all fortunes from table
|
|
|
+func (m Mongo) GetFortunes() ([]templates.Fortune, error) {
|
|
|
+ fortunes := make([]templates.Fortune, 0, 16)
|
|
|
+
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
|
|
+ defer cancel()
|
|
|
+
|
|
|
+ cur, err := m.fortunes.Find(ctx, bson.M{})
|
|
|
+ if err != nil {
|
|
|
+ return fortunes, err
|
|
|
+ }
|
|
|
+ defer cur.Close(ctx)
|
|
|
+
|
|
|
+ var fortune templates.Fortune
|
|
|
+ for cur.Next(context.Background()) {
|
|
|
+ err = cur.Decode(&fortune)
|
|
|
+ log.Println(fortune)
|
|
|
+ if err != nil {
|
|
|
+ return fortunes, err
|
|
|
+ }
|
|
|
+ fortunes = append(fortunes, fortune)
|
|
|
+ }
|
|
|
+
|
|
|
+ log.Println(fortunes)
|
|
|
+
|
|
|
+ if err := cur.Err(); err != nil {
|
|
|
+ return fortunes, err
|
|
|
+ }
|
|
|
+
|
|
|
+ return fortunes, nil
|
|
|
+}
|
|
|
+
|
|
|
+// GetFortunesPool selects all fortunes from table
|
|
|
+func (m Mongo) GetFortunesPool() ([]templates.Fortune, error) {
|
|
|
+ fortunes := templates.FortunesPool.Get().([]templates.Fortune)
|
|
|
+
|
|
|
+ ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
|
+ defer cancel()
|
|
|
+
|
|
|
+ cur, err := m.fortunes.Find(ctx, bson.M{})
|
|
|
+ if err != nil {
|
|
|
+ return fortunes, err
|
|
|
+ }
|
|
|
+ defer cur.Close(ctx)
|
|
|
+
|
|
|
+ var fortune templates.Fortune
|
|
|
+ for cur.Next(context.Background()) {
|
|
|
+ err = cur.Decode(&fortune)
|
|
|
+ if err != nil {
|
|
|
+ return fortunes, err
|
|
|
+ }
|
|
|
+ fortunes = append(fortunes, fortune)
|
|
|
+ }
|
|
|
+ if err := cur.Err(); err != nil {
|
|
|
+ return fortunes, err
|
|
|
+ }
|
|
|
+
|
|
|
+ return fortunes, nil
|
|
|
+}
|
|
|
+
|
|
|
+// NewMongoDB creates new connection to postgres db with official mongo driver
|
|
|
+func NewMongoDB(dbConnectionString string, maxConnectionsInPool int) (DB, error) {
|
|
|
+ var m Mongo
|
|
|
+ if err := m.Connect(dbConnectionString, maxConnectionsInPool); err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ return &m, nil
|
|
|
+}
|
chunariov