main.go

package sqac

import (
	"fmt"
	"log"
	"strings"

	"github.com/jmoiron/sqlx/reflectx"

	//_ "github.com/SAP/go-hdb/driver" // needed
	"github.com/jmoiron/sqlx"
)

// Sqac is the main access structure for the
// sqac library.
type Sqac struct {
	Hndl PublicDB
}

func init() {
	log.Println("sqac init...")
}

// Open a sqlx connection to the specified database
func Open(flavor string, args ...interface{}) (db *sqlx.DB, err error) {

	if len(args) != 1 {
		return nil, fmt.Errorf("incorrect number of args detected in Open()")
	}
	db, err = sqlx.Connect(flavor, args[0].(string))
	if err != nil {
		return nil, err
	}
	return db, nil
}

// Create establishes a connection with the db based on the connectionString.  A handle
// conforming to the sqac.PublicDB interface is passed back to the caller.  The type of
// the underlying handle object is that of the DBFlavor corresponding to the flavor var
// in the function definition.
func Create(flavor string, logFlag bool, dbLogFlag bool, connectionString string) (handle PublicDB) {

	switch flavor {
	case "postgres":
		pgh := new(PostgresFlavor)
		handle = pgh
		// db, err := Open("postgres", "host=127.0.0.1 user=godev dbname=sqactst sslmode=disable password=gogogo123")
		db, err := Open("postgres", connectionString)
		if err != nil {
			log.Fatalf("%s\n", err.Error())
		}
		handle.SetDB(db)

	case "mysql":
		myh := new(MySQLFlavor)
		handle = myh
		// db, err := Open("mysql", "godev:gogogo123@tcp(192.168.1.50:3306)/sqactst?charset=utf8&parseTime=True&loc=Local")
		db, err := Open("mysql", connectionString)
		if err != nil {
			log.Fatalf("%s\n", err.Error())
			panic(err)
		}
		handle.SetDB(db)
		// defer db.Close()

	case "sqlite":
		sqh := new(SQLiteFlavor)
		handle = sqh
		// db, err := Open("sqlite3", "testdb.sqlite")
		db, err := Open("sqlite3", connectionString)
		if err != nil {
			log.Fatalf("%s\n", err.Error())
			panic(err)
		}
		handle.SetDB(db)
		// defer db.Close()

	case "mssql":
		msh := new(MSSQLFlavor)
		handle = msh
		// db, err := Open("mssql", "sqlserver://SA:gogogo123@localhost:1401?database=sqactst")
		db, err := Open("mssql", connectionString)
		if err != nil {
			log.Fatalf("%s\n", err.Error())
			panic(err)
		}
		handle.SetDB(db)
		err = db.Ping()
		if err != nil {
			log.Fatalf("%s\n", err.Error())
			panic(err)
		}
		// defer db.Close()

	case "db2":

	case "hdb":
		hdh := new(HDBFlavor)
		handle = hdh
		// db, err := Open("hdb", "hdb://godev:gogogo123@your.hanadb.com:30047")
		db, err := Open("hdb", connectionString)
		if err != nil {
			log.Fatalf("%s\n", err.Error())
			panic(err)
		}

		// hdb shifts defs to upper-case if the DDL omits parentheses around the column /
		// table / view names. this is the SAP recommended approach to DDL definition,
		// so set the mapper to perform the shift in order to allow mapping of query results.
		db.Mapper = reflectx.NewMapperTagFunc("db", nil, func(s string) string {
			return strings.ToUpper(s)
		})
		handle.SetDB(db)
		err = db.Ping()
		if err != nil {
			log.Fatalf("%s\n", err.Error())
			panic(err)
		}

	default:

	}

	// detailed logging?
	if logFlag {
		handle.Log(true)
	} else {
		handle.Log(false)
	}

	if dbLogFlag {
		handle.DBLog(true)
	} else {
		handle.DBLog(false)
	}
	return handle
}