multitenantStack/services/companydb/companydb.go

107 lines
2.6 KiB
Go
Raw Normal View History

package companydb
2018-11-07 10:10:51 +00:00
import (
"database/sql"
"errors"
2018-11-07 10:10:51 +00:00
"fmt"
tokenTools "multitenantStack/services/tokenTools"
2018-11-07 10:10:51 +00:00
"os"
jwt "github.com/dgrijalva/jwt-go"
2018-11-07 10:10:51 +00:00
)
var dbs map[string]*sql.DB
// InitCompanyDBService Init companydb service and open system db connection
func InitCompanyDBService() {
dbs = make(map[string]*sql.DB)
2018-11-07 10:10:51 +00:00
systemDB, err := sql.Open("postgres", "host=127.0.0.1 port=5435 user=postgres password=postgre dbname=system sslmode=disable")
2018-11-07 10:10:51 +00:00
if err != nil {
fmt.Println("Fatal: could not connect to db, exiting... Error:", err)
os.Exit(1)
}
dbs["system"] = systemDB
}
// GetSystemDatabase returns system db
func GetSystemDatabase() *sql.DB {
fmt.Println(dbs)
return dbs["system"]
}
2018-11-08 10:42:31 +00:00
// GetDatabaseWithName Get orm and user information
func GetDatabaseWithName(companyName string) (*sql.DB, error) {
if dbs[companyName] != nil {
fmt.Println("DB Already open")
return dbs[companyName], nil
}
conStr := fmt.Sprintf("host=127.0.0.1 port=5435 user=postgres password=postgre dbname=%s sslmode=disable", companyName)
db, err := sql.Open("postgres", conStr)
dbs[companyName] = db
if err != nil {
return nil, err
}
return db, nil
}
// TODO: call upper function in this one to reduce code
2018-11-08 10:42:31 +00:00
2018-11-07 10:10:51 +00:00
// GetDatabase Get orm and user information
func GetDatabase(tokenString string) (jwt.MapClaims, *sql.DB, error) {
2018-11-07 10:10:51 +00:00
// validate token
valid, token := tokenTools.Validate(tokenString)
if !valid {
return nil, nil, errors.New("Token is invalid")
}
tokenMap := token.Claims.(jwt.MapClaims)
companyName := tokenMap["companyName"].(string)
if dbs[companyName] != nil {
fmt.Println("DB Already open")
return tokenMap, dbs[companyName], nil
}
conStr := fmt.Sprintf("host=127.0.0.1 port=5435 user=postgres password=postgre dbname=%s sslmode=disable", companyName)
fmt.Println(conStr)
db, err := sql.Open("postgres", conStr)
if err != nil {
return nil, nil, err
}
2018-11-07 10:10:51 +00:00
// return db with orm or error
return tokenMap, db, nil
2018-11-07 10:10:51 +00:00
}
// CreateDatabase Create a database by copying the template
func CreateDatabase(token string) {
/*
db, err = sql.Open("postgres", "host=127.0.0.1 port=5435 user=postgres password=postgre dbname=company5 sslmode=disable")
if err != nil {
log.Fatal(err)
}
or, err := orm.NewOrmWithDB("postgres", "temp", db)
*/
}
// DeleteDatabase Delete an entire database, this is very very dangerous :-)
func DeleteDatabase(token string) {
//_, db, err := GetDatabase(token)
2018-11-07 10:10:51 +00:00
/*
remove from map!
db.Close()
fmt.Println("Closed company5")
//}
res, err := o.Raw("DROP DATABASE company5;").Exec()
if err == nil {
num, _ := res.RowsAffected()
fmt.Println("mysql row affected number: ", num)
}
2018-11-07 10:10:51 +00:00
*/
}