Database.go

package main

import (
	"encoding/gob"
	"fmt"
	"github.com/LiamDotPro/Go-Multitenancy/tenants"
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/postgres"
	"github.com/wader/gormstore"
	"os"
	"time"
)

var Connection *gorm.DB
var Store *gormstore.Store

func startDatabaseServices() {

	// Database Connection string
	db, err := gorm.Open(os.Getenv("dialect"), os.Getenv("connectionString"))

	if err != nil {
		fmt.Println(err)
		panic("failed to connect database")
	}

	// Turn logging for the database on.
	db.LogMode(true)

	// Make Master connection available globally.
	Connection = db

	// Now Setup store - Tenant Store
	// Password is passed as byte key method
	Store = gormstore.NewOptions(db, gormstore.Options{
		TableName:       "sessions",
		SkipCreateTable: false,
	}, []byte(os.Getenv("sessionsPassword")))

	// Register session types for consuming in sessions
	gob.Register(HostProfile{})
	gob.Register(ClientProfile{})

	// Always attempt to migrate changes to the master tenant schema
	if err := migrateMasterTenantDatabase(); err != nil {
		fmt.Print("There was an error while trying to migrate the tenant tables..")
		os.Exit(1)
	}

	// attempt to migrate any tenant table changes to all clients.
	AutoMigrateTenantTableChanges()

	// Makes quit Available
	quit := make(chan struct{})

	// Every hour remove dead sessions.
	go Store.PeriodicCleanup(1*time.Hour, quit)
}

// Simply migrates all of the tenant tables
func AutoMigrateTenantTableChanges() {

	var TenantInformation [] tenants.TenantConnectionInformation

	Connection.Find(&tenants.TenantConnectionInformation{})

	for _, element := range TenantInformation {

		conn, _ := element.GetConnection()

		if err := migrateTenantTables(conn); err != nil {
			fmt.Print("An error occurred while attempting to migrate tenant tables")
			os.Exit(1)
		}
	}
}