Merge "mecm-mepm uninstall playbook added"

[ealt-edge.git] / mecm / mepm / applcm / broker / pkg / handlers / adapter / dbAdapter / dbAdapter.go

diff --git a/mecm/mepm/applcm/broker/pkg/handlers/adapter/dbAdapter/dbAdapter.go b/mecm/mepm/applcm/broker/pkg/handlers/adapter/dbAdapter/dbAdapter.go
index 8af5e51..23fb49e 100644 (file)
--- a/mecm/mepm/applcm/broker/pkg/handlers/adapter/dbAdapter/dbAdapter.go
+++ b/mecm/mepm/applcm/broker/pkg/handlers/adapter/dbAdapter/dbAdapter.go
@@ -17,13 +17,20 @@ package dbAdapter
 
 import (
        "broker/pkg/handlers/model"
+       "fmt"
        "github.com/jinzhu/gorm"
+       _ "github.com/jinzhu/gorm/dialects/postgres"
        "github.com/sirupsen/logrus"
        "os"
 )
 
-// DB name
-const DbName = "applcmDB"
+var (
+       username = os.Getenv("POSTGRES_USER")
+       password = os.Getenv("POSTGRES_PASSWORD")
+       dbName = os.Getenv("POSTGRES_DATABASE")
+       dbHost = os.Getenv("DBHOST")
+
+)
 
 // Database adapter
 type DbAdapter struct {
@@ -37,21 +44,16 @@ func NewDbAdapter(logger *logrus.Logger) *DbAdapter {
 
 // Creates database
 func (adapter *DbAdapter) CreateDatabase() {
-       adapter.logger.Info("creating Database...")
+       adapter.logger.Infof("creating Database...")
 
-       usrpswd := os.Getenv("MYSQL_USER") + ":" + os.Getenv("MYSQL_PASSWORD")
-       host := "@tcp(" + "dbhost" + ":3306)/"
+       dbUri := fmt.Sprintf("host=%s user=%s dbname=%s sslmode=disable password=%s", dbHost, username, dbName, password) //Build connection string
 
-       db, err := gorm.Open("mysql", usrpswd + host)
+       conn, err := gorm.Open("postgres", dbUri)
        if err != nil {
-               adapter.logger.Fatalf("Database connect error", err.Error())
+               fmt.Print(err)
        }
 
-       db.Exec("CREATE DATABASE  " + DbName)
-       db.Exec("USE applcmDB")
-       gorm.DefaultCallback.Create().Remove("mysql:set_identity_insert")
-
-       adapter.logger.Info("Migrating models...")
+       db := conn
        db.AutoMigrate(&model.AppPackageInfo{})
        db.AutoMigrate(&model.AppInstanceInfo{})
        adapter.db = db
@@ -69,7 +71,7 @@ func (adapter *DbAdapter) GetAppInstanceInfo(key string) (appInstInfo model.AppI
        adapter.logger.Infof("Get App Instance Info %s", key)
        var appInstanceInfo model.AppInstanceInfo
        returnVal := adapter.db.First(&appInstanceInfo, "id=?", key).Error
-       if returnVal !=  nil {
+       if returnVal != nil {
                return
        }
 
@@ -114,7 +116,7 @@ func (adapter *DbAdapter) GetAppPackageInfo(key string) (appPackageInfo model.Ap
        adapter.logger.Infof("Get App Package Info: %s", key)
        var appPkgInfo model.AppPackageInfo
        err := adapter.db.First(&appPkgInfo, "id=?", key).Error
-       if err !=  nil {
+       if err != nil {
                return
        }
        return appPkgInfo