Merge pull request #16 from Zondax/add/MongoConnection

Add/mongo connection

Author: Ezequiel Raynaudo

.circleci/config.yml

@@ -21,34 +21,22 @@ jobs:
       - run: make lint
 
   integration:
-    docker:
-      - image: golang:1.16
+    machine:
+      image: ubuntu-2004:202104-01 
     steps:
       - checkout
       - run:
-          name: "Prebuild"
-          command: |
-            echo "Prebuild..."
-      - run:
-          name: "Launch background processes"
-          background: true
-          command: |
-            echo "Starting executables we need in background"
-      - run:
-          name: "Give time for processes to start"
-          command: sleep 3
-      - run:
-          name: "Run tests"
+          name: "Run database tests"
           command: |
-            make test_integration
+            make test-database-services
 
 workflows:
   version: 2
   build_all:
     jobs:
       - build
       - checks
-#      - integration:
-#          requires:
-#            - build
-#            - checks
+      - integration:
+          requires:
+            - build
+            - checks

Makefile

@@ -22,3 +22,12 @@ lint:
 
 test: build
 	go test -race ./...
+
+test-database: build
+	go test ./connections/database/... -v
+
+# Docker
+
+test-database-services:
+	docker-compose -f tests_docker/test_database.yml up --abort-on-container-exit
+

connections/database/interface.go

@@ -2,8 +2,9 @@ package database
 
 import (
 	"fmt"
-	"gorm.io/gorm"
 	"net/http"
+
+	"gorm.io/gorm"
 )
 
 type AuthHeaderTransport struct {
@@ -21,8 +22,8 @@ type DBConnection interface {
 }
 
 type IDBQueryClient interface {
-	Connect() error
 	GetDB() interface{}
+	Connect() error
 }
 
 type DBQueryClient struct {
@@ -35,6 +36,7 @@ type DBConnectionParams struct {
 	Name     string
 	Host     string
 	Port     string
+	URI      string
 }
 
 type GraphqlClientParams struct {

connections/database/mongoDB.go

@@ -0,0 +1,48 @@
+package database
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"go.mongodb.org/mongo-driver/mongo/readpref"
+)
+
+type MongoConnection struct {
+	db *mongo.Client
+}
+
+func NewMongoConnection(params *DBConnectionParams) (*MongoConnection, error) {
+	uri := params.URI
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
+	if err != nil {
+		fmt.Printf("Failed to connect to db: %v \n", err)
+		return nil, err
+	}
+
+	// Ping the primary
+	if err := client.Ping(ctx, readpref.Primary()); err != nil {
+		return nil, err
+	}
+	fmt.Println("Successfully connected and pinged.")
+
+	return &MongoConnection{db: client}, nil
+}
+
+func (c *MongoConnection) GetDB() *mongo.Client {
+	return c.db
+}
+
+func (c *DBQueryClient) Connect(params *DBConnectionParams) (*mongo.Client, error) {
+	conn, err := NewMongoConnection(params)
+	if err != nil {
+		return nil, err
+	}
+
+	return conn.GetDB(), nil
+}

connections/database/mongoDB_test.go

@@ -0,0 +1,52 @@
+package database
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"testing"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+func TestBuffer_InsertRead(t *testing.T) {
+	var c DBQueryClient
+	params := &DBConnectionParams{URI: os.Getenv("MONGO_URI")}
+	client, err := c.Connect(params)
+	if err != nil {
+		fmt.Println(err)
+		t.Fail()
+	}
+
+	coll := client.Database("test").Collection("sample")
+	res, err := coll.InsertOne(context.TODO(), bson.D{{Key: "name", Value: "Alice"}})
+	if err != nil {
+		fmt.Println(err)
+		t.Fail()
+	}
+	fmt.Printf("inserted document with ID %v\n", res.InsertedID)
+
+	// Find the document for which the _id field matches id.
+	// Specify the Sort option to sort the documents by age.
+	// The first document in the sorted order will be returned.
+	opts := options.FindOne().SetSort(bson.D{{Key: "age", Value: 1}})
+	var result bson.M
+	readErr := coll.FindOne(
+		context.TODO(),
+		bson.D{{Key: "_id", Value: res.InsertedID}},
+		opts,
+	).Decode(&result)
+	if readErr != nil {
+		// ErrNoDocuments means that the filter did not match any documents in
+		// the collection.
+		if err == mongo.ErrNoDocuments {
+			fmt.Println("No document found")
+		}
+		fmt.Println(err)
+		t.Fail()
+	}
+
+	fmt.Printf("found document %v", result)
+}

go.mod

@@ -12,6 +12,7 @@ require (
 	github.com/prometheus/client_golang v1.11.0
 	github.com/spf13/cobra v1.1.1
 	github.com/spf13/viper v1.7.1
+	go.mongodb.org/mongo-driver v1.7.1
 	go.uber.org/zap v1.16.0
 	gorm.io/driver/postgres v1.0.6
 	gorm.io/gorm v1.21.12

go.sum

@@ -0,0 +1,28 @@
+version: "3.9"
+networks:
+  dbNetwork:
+    driver: bridge
+
+services:
+  mongodb:
+    image: mongo:latest
+    logging:
+      driver: none
+    networks:
+      - dbNetwork
+    environment:
+      MONGO_INITDB_ROOT_USERNAME: mongo
+      MONGO_INITDB_ROOT_PASSWORD: password
+
+  goapp:
+    image: golang:latest
+    depends_on:
+      - mongodb
+    networks:
+      - dbNetwork
+    volumes:
+      - ./../:/app
+    working_dir: /app
+    environment:
+      MONGO_URI: mongodb://mongo:password@mongodb:27017/
+    entrypoint: ["make", "test-database"]