Refactor and add database test

Author: m-lukas

.env.example

@@ -0,0 +1,5 @@
+DATABASE_TYPE=<mongo|dynamo>
+MONGO_USER=user
+MONGO_PASS=<password>
+MONGO_HOST=<host>:<port>
+MONGO_NAME=admin

.gitignore

@@ -1,3 +1,4 @@
 .realize.yaml
 data/
-github-analyser
+github-analyser
+.env

app/app.go

@@ -0,0 +1,37 @@
+package app
+
+import (
+	"net/http"
+
+	"github.com/go-chi/chi"
+
+	"github.com/go-chi/chi/middleware"
+	"github.com/rs/cors"
+)
+
+func InitRouter(apiPath string) *chi.Mux {
+
+	router := chi.NewRouter()
+
+	router.Use(middleware.RequestID)
+	router.Use(middleware.RealIP)
+	router.Use(middleware.Logger)
+	router.Use(middleware.Recoverer)
+
+	router.HandleFunc(apiPath, checkAPI)
+
+	return router
+}
+
+func HandleCors(h http.Handler) http.Handler {
+
+	cors := cors.New(cors.Options{
+		AllowedOrigins:   []string{"*"},
+		AllowCredentials: true,
+		AllowedHeaders:   []string{"Content-Type", "*"},
+		Debug:            false,
+	})
+	handler := cors.Handler(h)
+	return handler
+
+}

app/routes.go

@@ -0,0 +1,40 @@
+package app
+
+import (
+	"context"
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/globalsign/mgo/bson"
+	"github.com/m-lukas/github-analyser/db"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+)
+
+func checkAPI(w http.ResponseWriter, r *http.Request) {
+	db, err := db.Get()
+	if err != nil {
+		log.Fatal("Unable to establish database connection!")
+	}
+	collection := db.Collection("testing")
+
+	result := struct {
+		ID    primitive.ObjectID `bson:"_id,omitempty"`
+		Value string             `bson:"Value"`
+	}{}
+
+	objectID, err := primitive.ObjectIDFromHex("5cab24edf3f1a30a16927aeb")
+	if err != nil {
+		log.Fatal("Could not convert test object id!")
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	data := collection.FindOne(ctx, bson.M{"_id": objectID})
+
+	err = data.Decode(&result)
+	if err != nil {
+		log.Fatal("Error while decoding mongo document data!")
+	}
+	w.Write([]byte(result.Value))
+}

db/config.go

@@ -0,0 +1,105 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"os"
+	"time"
+
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"go.mongodb.org/mongo-driver/mongo/readpref"
+)
+
+var dbInstance *Database
+
+type Database struct {
+	MongoClient *mongo.Client
+	Config      *DatabaseConfig
+}
+
+type DatabaseConfig struct {
+	MongoDatabaseName string
+}
+
+func defaultDatabaseConfig() *DatabaseConfig {
+	return &DatabaseConfig{
+		MongoDatabaseName: "core",
+	}
+}
+
+func Init() error {
+	dbInstance = &Database{
+		Config: defaultDatabaseConfig(),
+	}
+
+	mongoClient, err := initMongoClient()
+	if err != nil {
+		return err
+	}
+
+	dbInstance.MongoClient = mongoClient
+	return nil
+}
+
+/*
+	Returns default mongo database.
+*/
+func Get() (*mongo.Database, error) {
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	err := dbInstance.MongoClient.Ping(ctx, readpref.Primary())
+
+	if err == nil {
+		client := dbInstance.MongoClient
+		config := dbInstance.Config
+		return client.Database(config.MongoDatabaseName), nil
+	}
+
+	mongoClient, err := initMongoClient()
+	if err != nil {
+		return nil, err
+	}
+
+	dbInstance.MongoClient = mongoClient
+
+	client := dbInstance.MongoClient
+	config := dbInstance.Config
+	return client.Database(config.MongoDatabaseName), nil
+}
+
+/*
+	Initializes the mongoDB Client to access databases and collections.
+*/
+func initMongoClient() (*mongo.Client, error) {
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	client, err := mongo.Connect(ctx, options.Client().ApplyURI(getMongoDBConfig()))
+	if err != nil {
+		return nil, err
+	}
+
+	err = client.Ping(ctx, readpref.Primary())
+	if err != nil {
+		return nil, err
+	}
+
+	log.Println("Initialized mongo client!")
+	return client, err
+
+}
+
+/*
+	Returns configurated URI string for MongoDB.
+*/
+func getMongoDBConfig() (uri string) {
+	dbHost := os.Getenv("MONGO_HOST")
+	dbName := os.Getenv("MONGO_NAME")
+	dbUser := os.Getenv("MONGO_USER")
+	dbPass := os.Getenv("MONGO_PASS")
+
+	return fmt.Sprintf(`mongodb://%s:%s@%s/%s`, dbUser, dbPass, dbHost, dbName)
+}

db/db.go

@@ -23,12 +23,6 @@ server:
 schema:
 - name: github-analyser
   path: /Users/m-lukas/Documents/Projects/go/src/github.com/m-lukas/github-analyser
-  env:
-    DATABASE_TYPE: <mongo/dynamo>
-    MONGO_USER: <user>
-    MONGO_PASS: <password>
-    MONGO_HOST: <host&port>
-    MONGO_NAME: admin
   commands:
     fmt:
       status: true